LibBotUniversal/tests/test_config_async.py

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