mirror of
https://github.com/evilsocket/audit
synced 2026-06-06 15:44:27 +00:00
Simone Margaritelli
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
"""Tests for audit.json_utils.extract_json."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from audit.json_utils import extract_json
|
|
|
|
|
|
def test_plain_json() -> None:
|
|
assert extract_json('{"a": 1}') == {"a": 1}
|
|
|
|
|
|
def test_json_array() -> None:
|
|
assert extract_json("[1, 2, 3]") == [1, 2, 3]
|
|
|
|
|
|
def test_fenced_json() -> None:
|
|
text = 'Sure!\n```json\n{"a": 1}\n```\nDone.'
|
|
assert extract_json(text) == {"a": 1}
|
|
|
|
|
|
def test_fenced_no_language() -> None:
|
|
text = '```\n{"a": 1}\n```'
|
|
assert extract_json(text) == {"a": 1}
|
|
|
|
|
|
def test_prose_with_embedded_json() -> None:
|
|
text = 'Here is my answer: {"a": 1, "b": [1,2]} please use it.'
|
|
assert extract_json(text) == {"a": 1, "b": [1, 2]}
|
|
|
|
|
|
def test_embedded_json_with_braces_in_strings() -> None:
|
|
text = 'preamble {"k": "value with } brace"} epilogue'
|
|
assert extract_json(text) == {"k": "value with } brace"}
|
|
|
|
|
|
def test_empty_raises() -> None:
|
|
with pytest.raises(ValueError):
|
|
extract_json("")
|
|
|
|
|
|
def test_no_json_raises() -> None:
|
|
with pytest.raises(ValueError):
|
|
extract_json("just text, no json here")
|