feat: basefiles

tests/conftest.py

@@ -0,0 +1,16 @@
+import logging
+
+import pytest
+
+
+@pytest.fixture()
+def anyio_backend():
+    return 'trio'
+
+
+@pytest.fixture()
+def logger():
+    logging.basicConfig()
+    logger = logging.getLogger("test_logger")
+    logger.setLevel(logging.INFO)
+    return logger

tests/test_muzzle.py

@@ -0,0 +1,26 @@
+import pytest
+
+import shush
+
+
+def test_muzzle(caplog, logger):
+    @shush.muzzle("foo")
+    def muzzled_func():
+        logger.info("this contains `foo` so it should be muzzled out")
+        logger.info("this doesn't contain it so it should be showing")
+
+    muzzled_func()
+    assert caplog.text == ("INFO     test_logger:test_muzzle.py:10 "
+                           "this doesn't contain it so it should be showing\n")
+
+
+@pytest.mark.anyio()
+async def test_muzzle_async(caplog, logger):
+    @shush.muzzle("foo")
+    async def muzzled_func():
+        logger.info("this contains `foo` so it should be muzzled out")
+        logger.info("this doesn't contain it so it should be showing")
+
+    await muzzled_func()
+    assert caplog.text == ("INFO     test_logger:test_muzzle.py:22 "
+                           "this doesn't contain it so it should be showing\n")

tests/test_supress.py

@@ -0,0 +1,36 @@
+import pytest
+
+import shush
+
+
+def test_supress(caplog, logger):
+    @shush.suppress
+    def suppressed_func():
+        logger.info("This log should not appear")
+
+    def normal_func():
+        logger.info("This log should appear")
+
+    suppressed_func()
+    assert caplog.records == []
+
+    normal_func()
+    assert caplog.text == ('INFO     test_logger:test_supress.py:12 '
+                           'This log should appear\n')
+
+
+@pytest.mark.anyio()
+async def test_supress_async(caplog, logger):
+    @shush.suppress
+    async def suppressed_func_async():
+        logger.info("This log should not appear")
+
+    async def normal_func_async():
+        logger.info("This log should appear")
+
+    await suppressed_func_async()
+    assert caplog.records == []
+
+    await normal_func_async()
+    assert caplog.text == ('INFO     test_logger:test_supress.py:29 '
+                           'This log should appear\n')