1 files changed, 49 insertions, 0 deletions
diff --git a/catapult/telemetry/telemetry/internal/forwarders/do_nothing_forwarder_unittest.py b/catapult/telemetry/telemetry/internal/forwarders/do_nothing_forwarder_unittest.py
new file mode 100644
index 00000000..7a17c870
--- /dev/null
+++ b/catapult/telemetry/telemetry/internal/forwarders/do_nothing_forwarder_unittest.py
@@ -0,0 +1,49 @@
+# Copyright 2012 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import unittest
+
+from telemetry.internal import forwarders
+from telemetry.internal.forwarders import do_nothing_forwarder
+
+import py_utils
+
+
+class TestDoNothingForwarder(do_nothing_forwarder.DoNothingForwarder):
+  """Override _WaitForConnect to avoid actual socket connection."""
+
+  def __init__(self, port_pairs):
+    self.connected_addresses = []
+    super(TestDoNothingForwarder, self).__init__(port_pairs)
+
+  def _WaitForConnectionEstablished(self, address, timeout):
+    self.connected_addresses.append(address)
+
+
+class TestErrorDoNothingForwarder(do_nothing_forwarder.DoNothingForwarder):
+  """Simulate a connection error."""
+
+  def _WaitForConnectionEstablished(self, address, timeout):
+    raise py_utils.TimeoutException
+
+
+class CheckPortPairsTest(unittest.TestCase):
+  def testBasicCheck(self):
+    port_pair = forwarders.PortPair(80, 80)
+    f = TestDoNothingForwarder(port_pair)
+    expected_connected_addresses = [
+        ('127.0.0.1', 80),
+        ]
+    self.assertEqual(expected_connected_addresses, f.connected_addresses)
+
+  def testPortMismatchRaisesPortsMismatchError(self):
+    # The do_nothing_forward cannot forward from one port to another.
+    port_pair = forwarders.PortPair(80, 81)
+    with self.assertRaises(do_nothing_forwarder.PortsMismatchError):
+      TestDoNothingForwarder(port_pair)
+
+  def testConnectionTimeoutRaisesConnectionError(self):
+    port_pair = forwarders.PortPair(80, 80)
+    with self.assertRaises(do_nothing_forwarder.ConnectionError):
+      TestErrorDoNothingForwarder(port_pair)