mirror of
https://github.com/rio-labs/rio.git
synced 2026-02-10 23:59:10 -06:00
75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
import typing_extensions as te
|
|
from pytest import MonkeyPatch
|
|
from uniserde import JsonDoc
|
|
|
|
import rio.testing
|
|
|
|
|
|
class FakeFile:
|
|
def __init__(self, content: str) -> None:
|
|
self._content = content
|
|
|
|
async def __aenter__(self) -> te.Self:
|
|
return self
|
|
|
|
async def __aexit__(self, *args) -> None:
|
|
pass
|
|
|
|
async def read(self) -> str:
|
|
return self._content
|
|
|
|
|
|
class RootSettings(rio.UserSettings):
|
|
foo: str = "foo"
|
|
|
|
|
|
class FooSettings(rio.UserSettings):
|
|
section_name = "foo"
|
|
|
|
bar: str = "bar"
|
|
|
|
|
|
async def test_load_settings() -> None:
|
|
user_settings: JsonDoc = {
|
|
":foo": "bar",
|
|
"foo:bar": "baz",
|
|
}
|
|
|
|
async with rio.testing.TestClient(
|
|
running_in_window=False,
|
|
default_attachments=(RootSettings(), FooSettings()),
|
|
user_settings=user_settings,
|
|
) as test_client:
|
|
root_settings = test_client.session[RootSettings]
|
|
foo_settings = test_client.session[FooSettings]
|
|
|
|
assert root_settings.foo == "bar"
|
|
assert foo_settings.bar == "baz"
|
|
|
|
|
|
async def test_load_settings_file(monkeypatch: MonkeyPatch) -> None:
|
|
try:
|
|
import aiofiles
|
|
except ImportError:
|
|
raise RuntimeError(
|
|
"This test requires the window extra to run, specifically `aiofiles`."
|
|
)
|
|
|
|
monkeypatch.setattr(
|
|
aiofiles,
|
|
"open",
|
|
lambda _: FakeFile('{"foo": "bar", "section:foo": {"bar": "baz"} }'),
|
|
)
|
|
|
|
async with rio.testing.TestClient(
|
|
running_in_window=True,
|
|
default_attachments=(RootSettings(), FooSettings()),
|
|
) as test_client:
|
|
root_settings = test_client.session[RootSettings]
|
|
foo_settings = test_client.session[FooSettings]
|
|
|
|
assert root_settings.foo == "bar"
|
|
assert foo_settings.bar == "baz"
|