Shore up test suite.

1 files changed, 22 insertions, 0 deletions

diff --git a/test/test_log.py b/test/test_log.py
new file mode 100644
index 00000000..5473236e
--- /dev/null
+++ b/test/test_log.py
@@ -0,0 +1,22 @@
+import StringIO
+from libpathod import log
+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()
+    try:
+        with log.Log(outf, False, rw, rw) as lg:
+            raise netlib.tcp.NetLibDisconnect()
+    except netlib.tcp.NetLibDisconnect:
+        pass
+    assert "Disconnected" in outf.getvalue()