unit-tests: follow tested package structure

Author: rnovatorov

tests/unit/test_async.py

@@ -0,0 +1,18 @@
+import pytest
+
+import enapter
+
+
+async def test_aclose() -> None:
+    @enapter.async_.generator
+    async def agen():
+        yield 1
+        yield 2
+        yield 3  # pragma: no cover
+
+    async with agen() as g:
+        assert await g.__anext__() == 1
+        assert await g.__anext__() == 2
+
+    with pytest.raises(StopAsyncIteration):
+        await g.__anext__()

tests/unit/test_async/__init__.py

@@ -0,0 +1,74 @@
+import asyncio
+
+import pytest
+
+import enapter
+
+
+async def test_run_not_implemented() -> None:
+    class R(enapter.async_.Routine):
+        pass
+
+    with pytest.raises(TypeError):
+        R()  # type: ignore
+
+
+async def test_start_twice() -> None:
+    started = asyncio.Event()
+
+    class R(enapter.async_.Routine):
+        async def _run(self) -> None:
+            started.set()
+            await asyncio.Event().wait()
+
+    async with R() as r:
+        await started.wait()
+        with pytest.raises(RuntimeError):
+            await r.start()
+
+
+async def test_stop_without_start() -> None:
+    class R(enapter.async_.Routine):
+        async def _run(self) -> None:
+            await asyncio.Event().wait()  # pragma: no cover
+
+    r = R()
+    with pytest.raises(RuntimeError):
+        await r.stop()
+
+
+async def test_context_manager() -> None:
+    started = asyncio.Event()
+    done = asyncio.Event()
+
+    class R(enapter.async_.Routine):
+        async def _run(self) -> None:
+            started.set()
+            try:
+                await asyncio.Event().wait()
+            finally:
+                done.set()
+
+    async with R():
+        await started.wait()
+
+    assert done.is_set()
+
+
+async def test_task_group() -> None:
+    started = asyncio.Event()
+    done = asyncio.Event()
+
+    class R(enapter.async_.Routine):
+        async def _run(self) -> None:
+            started.set()
+            try:
+                await asyncio.Event().wait()
+            finally:
+                done.set()
+
+    async with asyncio.TaskGroup() as tg:
+        async with R(task_group=tg):
+            await started.wait()
+
+    assert done.is_set()

tests/unit/test_async/test_generator.py

@@ -0,0 +1,70 @@
+import contextlib
+import io
+import json
+import re
+from typing import Generator
+
+import enapter
+
+
+def test_record_ends_with_newline() -> None:
+    buf = io.StringIO()
+
+    with configure_logging(level="info", stream=buf):
+        enapter.log.LOGGER.info("hello")
+
+    assert buf.getvalue().endswith("\n")
+
+
+def test_record_fields() -> None:
+    buf = io.StringIO()
+
+    with configure_logging(level="info", stream=buf):
+        enapter.log.LOGGER.info("hello")
+
+    record = json.loads(buf.getvalue())
+
+    time = record.pop("time")
+    assert re.match(r"\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\.\d+", time) is not None
+
+    assert record["level"] == "INFO"
+    assert record["name"] == "enapter"
+    assert record["message"] == "hello"
+
+
+def test_exc_info() -> None:
+    buf = io.StringIO()
+
+    with configure_logging(level="info", stream=buf):
+        try:
+            raise RuntimeError("oops")
+        except RuntimeError:
+            enapter.log.LOGGER.exception("boom")
+
+    record = json.loads(buf.getvalue())
+
+    assert record["message"] == "boom"
+    assert "Traceback (most recent call last)" in record["exc_info"]
+    assert 'RuntimeError("oops")' in record["exc_info"]
+
+
+def test_stack_info() -> None:
+    buf = io.StringIO()
+
+    with configure_logging(level="info", stream=buf):
+        enapter.log.LOGGER.info("hello", stack_info=True)
+
+    record = json.loads(buf.getvalue())
+
+    assert record["message"] == "hello"
+    assert "Stack (most recent call last)" in record["stack_info"]
+    assert "test_stack_info" in record["stack_info"]
+
+
+@contextlib.contextmanager
+def configure_logging(*args, **kwargs) -> Generator[None, None, None]:
+    enapter.log.configure(*args, **kwargs)
+    try:
+        yield
+    finally:
+        enapter.log.configure(level=None)