move tests into shared folder

1 files changed, 25 insertions, 0 deletions

diff --git a/test/pathod/test_log.py b/test/pathod/test_log.py
new file mode 100644
index 00000000..8f38c040
--- /dev/null
+++ b/test/pathod/test_log.py
@@ -0,0 +1,25 @@
+import StringIO
+from libpathod import log
+from netlib.exceptions import TcpDisconnect
+import netlib.tcp
+
+
+class DummyIO(StringIO.StringIO):
+
+    def start_log(self, *args, **kwargs):
+        pass
+
+    def get_log(self, *args, **kwargs):
+        return ""
+
+
+def test_disconnect():
+    outf = DummyIO()
+    rw = DummyIO()
+    l = log.ConnectionLogger(outf, False, rw, rw)
+    try:
+        with l.ctx() as lg:
+            lg("Test")
+    except TcpDisconnect:
+        pass
+    assert "Test" in outf.getvalue()