61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
from json import JSONDecodeError, dumps
|
|
from pathlib import Path
|
|
from typing import Any, Union
|
|
|
|
import pytest
|
|
|
|
from libbot import json_read, json_write
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"path, expected",
|
|
[
|
|
(
|
|
"data/test.json",
|
|
{
|
|
"foo": "bar",
|
|
"abcdefg": ["higklmnop", {"lol": {"kek": [1.0000035, 0.2542, 1337]}}],
|
|
},
|
|
),
|
|
("data/empty.json", {}),
|
|
],
|
|
)
|
|
async def test_json_read_valid(path: Union[str, Path], expected: Any):
|
|
assert await json_read(path) == expected
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"path, expected",
|
|
[
|
|
("data/invalid.json", pytest.raises(JSONDecodeError)),
|
|
("data/nonexistent.json", pytest.raises(FileNotFoundError)),
|
|
],
|
|
)
|
|
async def test_json_read_invalid(path: Union[str, Path], expected: Any):
|
|
with expected:
|
|
assert await json_read(path) == expected
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"data, path",
|
|
[
|
|
(
|
|
{
|
|
"foo": "bar",
|
|
"abcdefg": ["higklmnop", {"lol": {"kek": [1.0000035, 0.2542, 1337]}}],
|
|
},
|
|
"data/test.json",
|
|
),
|
|
({}, "data/empty.json"),
|
|
],
|
|
)
|
|
async def test_json_write(data: Any, path: Union[str, Path]):
|
|
await json_write(data, path)
|
|
|
|
assert Path(path).is_file()
|
|
with open(path, "r", encoding="utf-8") as f:
|
|
assert f.read() == dumps(data, ensure_ascii=False, indent=4)
|