Index: tools/telemetry/telemetry/internal/forwarders/forwarders_unittest.py |
diff --git a/tools/telemetry/telemetry/internal/forwarders/forwarders_unittest.py b/tools/telemetry/telemetry/internal/forwarders/forwarders_unittest.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..173b97c9ccb5b24145d148e81c4b50b962d96252 |
--- /dev/null |
+++ b/tools/telemetry/telemetry/internal/forwarders/forwarders_unittest.py |
@@ -0,0 +1,38 @@ |
+# Copyright 2016 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 |
+ |
+ |
+class PortPairsUnittest(unittest.TestCase): |
+ |
+ def testPortPairs(self): |
+ port_pairs = forwarders.PortPairs( |
+ http=forwarders.PortPair(local_port=1, remote_port=2), |
+ https=forwarders.PortPair(local_port=3, remote_port=4), |
+ dns=forwarders.PortPair(local_port=5, remote_port=6), |
+ ) |
+ |
+ self.assertEquals(port_pairs, |
+ forwarders.PortPairs.Make( |
+ http=(1, 2), https=(3, 4), dns=(5, 6))) |
+ self.assertEquals(port_pairs.local_ports, |
+ forwarders.PortSet(1, 3, 5)) |
+ self.assertEquals(port_pairs.remote_ports, |
+ forwarders.PortSet(2, 4, 6)) |
+ |
+ def testPortPairs_noDns(self): |
+ port_pairs = forwarders.PortPairs( |
+ http=forwarders.PortPair(local_port=1, remote_port=2), |
+ https=forwarders.PortPair(local_port=3, remote_port=4), |
+ dns=None |
+ ) |
+ |
+ self.assertEquals(port_pairs, |
+ forwarders.PortPairs.Make(http=(1, 2), https=(3, 4))) |
+ self.assertEquals(port_pairs.local_ports, |
+ forwarders.PortSet(1, 3, None)) |
+ self.assertEquals(port_pairs.remote_ports, |
+ forwarders.PortSet(2, 4, None)) |