From 06c06a2838b6984a659f42875342153be58927ec Mon Sep 17 00:00:00 2001
From: Dave Hall <dave@etianen.com>
Date: Sun, 4 Feb 2024 15:31:44 +0000
Subject: [PATCH] wip

---
 tests/test_logot.py | 23 +----------------------
 tests/test_wait.py  | 27 +++++++++++++++++++++++++++
 2 files changed, 28 insertions(+), 22 deletions(-)
 create mode 100644 tests/test_wait.py

diff --git a/tests/test_logot.py b/tests/test_logot.py
index 2feb007e..23100c47 100644
--- a/tests/test_logot.py
+++ b/tests/test_logot.py
@@ -5,7 +5,7 @@
 import pytest
 
 from logot import Captured, Logot, logged
-from tests import capture_soon, lines, logger
+from tests import lines, logger
 
 
 def test_capturing() -> None:
@@ -59,27 +59,6 @@ def test_assert_not_logged_fail(logot: Logot) -> None:
     )
 
 
-def test_wait_for_pass_immediate(logot: Logot) -> None:
-    logot.capture(Captured("INFO", "foo bar"))
-    logot.wait_for(logged.info("foo bar"))
-
-
-def test_wait_for_pass_soon(logot: Logot) -> None:
-    with capture_soon(logot, Captured("INFO", "foo bar")):
-        logot.wait_for(logged.info("foo bar"))
-
-
-def test_wait_for_fail(logot: Logot) -> None:
-    logot.capture(Captured("INFO", "boom!"))
-    with pytest.raises(AssertionError) as ex:
-        logot.wait_for(logged.info("foo bar"), timeout=0.1)
-    assert str(ex.value) == lines(
-        "Not logged:",
-        "",
-        "[INFO] foo bar",
-    )
-
-
 def test_clear(logot: Logot) -> None:
     logot.capture(Captured("INFO", "foo bar"))
     logot.clear()
diff --git a/tests/test_wait.py b/tests/test_wait.py
new file mode 100644
index 00000000..7c5ae3d2
--- /dev/null
+++ b/tests/test_wait.py
@@ -0,0 +1,27 @@
+from __future__ import annotations
+
+import pytest
+
+from logot import Captured, Logot, logged
+from tests import capture_soon, lines
+
+
+def test_wait_for_pass_immediate(logot: Logot) -> None:
+    logot.capture(Captured("INFO", "foo bar"))
+    logot.wait_for(logged.info("foo bar"))
+
+
+def test_wait_for_pass_soon(logot: Logot) -> None:
+    with capture_soon(logot, Captured("INFO", "foo bar")):
+        logot.wait_for(logged.info("foo bar"))
+
+
+def test_wait_for_fail(logot: Logot) -> None:
+    logot.capture(Captured("INFO", "boom!"))
+    with pytest.raises(AssertionError) as ex:
+        logot.wait_for(logged.info("foo bar"), timeout=0.1)
+    assert str(ex.value) == lines(
+        "Not logged:",
+        "",
+        "[INFO] foo bar",
+    )