test close

con · May 14, 2024 · d32a0ff · d32a0ff
1 parent 50c96f0
commit d32a0ff
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/duct.py b/src/duct.py
@@ -220,6 +220,9 @@ class TailPipe:
     def __init__(self, file_path, buffer):
         self.file_path = file_path
         self.buffer = buffer
+        self.stop_event = None
+        self.infile = None
+        self.thread = None
 
     def start(self):
         Path(self.file_path).touch()

diff --git a/test/test_tailpipe.py b/test/test_tailpipe.py
@@ -43,5 +43,14 @@ def test_high_throughput(mock_stdout, fixture_path):
     assert mock_stdout.getvalue() == expected
 
 
+@patch("sys.stdout", new_callable=lambda: MockStream())
+def test_close(mock_stdout):
+    with tempfile.NamedTemporaryFile(mode="wb") as tmpfile:
+        stream = TailPipe(tmpfile.name, mock_stdout.buffer)
+        stream.start()
+        stream.close()
+        assert stream.infile.closed
+
+
 if __name__ == "__main__":
     test_high_throughput()