49 lines
1.2 KiB
Python
49 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, location_config: Path):
|
|
assert await config_get(args[0], *args[1:], config_file=location_config) == 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, location_config: Path
|
|
):
|
|
with expected:
|
|
assert (
|
|
await config_get(args[0], *args[1:], config_file=location_config)
|
|
== 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, location_config: Path):
|
|
await config_set(key, value, *path, config_file=location_config)
|
|
assert await config_get(key, *path, config_file=location_config) == value
|