From 37ace81e6e17438f41eb26f45c2fffbc5c519a46 Mon Sep 17 00:00:00 2001
From: Brian Warner <warner@lothar.com>
Date: Fri, 3 Nov 2017 16:13:47 -0700
Subject: [PATCH] WIP new tests

---
 .../test/test_connect.py                      | 28 +++++++++++++
 .../test/test_logging.py                      | 41 +++++++++++++++++++
 2 files changed, 69 insertions(+)
 create mode 100644 src/wormhole_transit_relay/test/test_connect.py
 create mode 100644 src/wormhole_transit_relay/test/test_logging.py

diff --git a/src/wormhole_transit_relay/test/test_connect.py b/src/wormhole_transit_relay/test/test_connect.py
new file mode 100644
index 0000000..e30f824
--- /dev/null
+++ b/src/wormhole_transit_relay/test/test_connect.py
@@ -0,0 +1,28 @@
+from __future__ import print_function, unicode_literals
+import mock
+from twisted.trial import unittest
+from ..transit_server import Transit
+
+class Connection(unittest.TestCase):
+    def test_connection(self):
+        ts = Transit(blur_usage=60, usage_logfile=None, stats_file=None)
+        c1 = mock.Mock()
+        c2 = mock.Mock()
+        ts.connection_got_token("token1", "side1", c1)
+        self.assertEqual(c1.mock_calls, [])
+
+        ts.connection_got_token("token1", "side2", c2)
+        self.assertEqual(c1.mock_calls, [mock.call.buddy_connected(c2)])
+        self.assertEqual(c2.mock_calls, [mock.call.buddy_connected(c1)])
+
+        ts.transitFinished(c1, "token1", "side1", "desc1")
+        ts.transitFinished(c2, "token1", "side2", "desc2")
+        
+
+    def test_lonely(self):
+        ts = Transit(blur_usage=60, usage_logfile=None, stats_file=None)
+        c1 = mock.Mock()
+        ts.connection_got_token("token1", "side1", c1)
+        self.assertEqual(c1.mock_calls, [])
+
+        ts.transitFinished(c1, "token1", "side1", "desc1")
diff --git a/src/wormhole_transit_relay/test/test_logging.py b/src/wormhole_transit_relay/test/test_logging.py
new file mode 100644
index 0000000..227d1dc
--- /dev/null
+++ b/src/wormhole_transit_relay/test/test_logging.py
@@ -0,0 +1,41 @@
+from __future__ import print_function, unicode_literals
+import mock
+from twisted.trial import unittest
+from ..transit_server import Transit
+
+class FakeConnection(object):
+    def __init__(self, token):
+        self._token = token
+    def describeToken(self):
+        return self._token
+    def buddy_connected(self, other):
+        pass
+
+class Logging(unittest.TestCase):
+    def test_connection_yeslog(self):
+        ts = Transit(blur_usage=None, usage_logfile=None, stats_file=None)
+        c1 = FakeConnection("c1")
+        c2 = FakeConnection("c2")
+        expected = []
+        with mock.patch("twisted.python.log.msg") as m:
+            ts.connection_got_token("token1", "side1", c1)
+            expected.append(mock.call("transit relay 1: c1"))
+            self.assertEqual(m.mock_calls, expected)
+
+            ts.connection_got_token("token1", "side2", c2)
+            expected.append(mock.call("transit relay 2: c2"))
+            self.assertEqual(m.mock_calls, expected)
+
+            ts.transitFinished(c1, "token1", "side1", "desc1")
+            expected.append(mock.call("transitFinished desc1"))
+            self.assertEqual(m.mock_calls, expected)
+
+    def test_connection_nolog(self):
+        ts = Transit(blur_usage=60, usage_logfile=None, stats_file=None)
+        c1 = FakeConnection("c1")
+        with mock.patch("twisted.python.log.msg") as m:
+            ts.connection_got_token("token1", "side1", c1)
+            self.assertEqual(m.mock_calls, [])
+            ts.connection_got_token("token1", "side2", c1)
+            self.assertEqual(m.mock_calls, [])
+