50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
from pathlib import Path
|
|
from typing import Any, List
|
|
|
|
import pytest
|
|
|
|
from libbot import config_get, config_set
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"args, expected",
|
|
[
|
|
(["locale"], "en"),
|
|
(["bot_token", "bot"], "sample_token"),
|
|
],
|
|
)
|
|
async def test_config_get_valid(args: List[str], expected: str):
|
|
assert (
|
|
await config_get(args[0], *args[1:], config_file=Path("tests/config.json"))
|
|
== expected
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"args, expected",
|
|
[
|
|
(["bot_stonks", "bot"], pytest.raises(KeyError)),
|
|
],
|
|
)
|
|
async def test_config_get_invalid(args: List[str], expected: Any):
|
|
with expected:
|
|
assert (
|
|
await config_get(args[0], *args[1:], config_file=Path("tests/config.json"))
|
|
== expected
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"key, path, value",
|
|
[
|
|
("locale", [], "en"),
|
|
("bot_token", ["bot"], "sample_token"),
|
|
],
|
|
)
|
|
async def test_config_set(key: str, path: List[str], value: Any):
|
|
await config_set(key, value, *path, config_file=Path("tests/config.json"))
|
|
assert await config_get(key, *path, config_file=Path("tests/config.json")) == value
|