Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(206)

Unified Diff: tools/telemetry/telemetry/internal/platform/network_controller_backend_unittest.py

Issue 1647513002: Delete tools/telemetry. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 4 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: tools/telemetry/telemetry/internal/platform/network_controller_backend_unittest.py
diff --git a/tools/telemetry/telemetry/internal/platform/network_controller_backend_unittest.py b/tools/telemetry/telemetry/internal/platform/network_controller_backend_unittest.py
deleted file mode 100644
index ac43dc05e66b2c06622a0cfe151c42aa2e0866ee..0000000000000000000000000000000000000000
--- a/tools/telemetry/telemetry/internal/platform/network_controller_backend_unittest.py
+++ /dev/null
@@ -1,263 +0,0 @@
-# Copyright 2014 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 tempfile
-import unittest
-
-from telemetry.internal.platform import network_controller_backend
-from telemetry.internal import forwarders
-from telemetry.util import wpr_modes
-
-
-class FakePlatformBackend(object):
- @property
- def wpr_ca_cert_path(self):
- return None
-
- @property
- def forwarder_factory(self):
- return FakeForwarderFactory()
-
-
-class FakeForwarderFactory(object):
- def __init__(self):
- self.host_ip = '123.321.123.321'
- self.port_pairs = None
-
- def Create(self, port_pairs):
- return forwarders.Forwarder(port_pairs)
-
-
-class FakeBrowserBackend(object):
- def __init__(self, http_ports, https_ports, dns_ports):
- self.wpr_port_pairs = forwarders.PortPairs(
- http=forwarders.PortPair(*http_ports),
- https=forwarders.PortPair(*https_ports),
- dns=forwarders.PortPair(*dns_ports) if dns_ports else None)
-
-
-class FakeReplayServer(object):
- def __init__(self, archive_path, host_ip, http_port, https_port, dns_port,
- replay_args):
- self.archive_path = archive_path
- self.host_ip = host_ip
- self.http_port = http_port
- self.https_port = https_port
- self.dns_port = dns_port
- self.replay_args = replay_args
- self.is_stopped = False
-
- def StartServer(self):
- return self.http_port, self.https_port, self.dns_port
-
- def StopServer(self):
- assert not self.is_stopped
- self.is_stopped = True
-
-
-class TestNetworkControllerBackend(
- network_controller_backend.NetworkControllerBackend):
- """NetworkControllerBackend with a fake ReplayServer."""
-
- def __init__(self, platform_backend, fake_started_http_port,
- fake_started_https_port, fake_started_dns_port):
- super(TestNetworkControllerBackend, self).__init__(platform_backend)
- self.fake_started_http_port = fake_started_http_port
- self.fake_started_https_port = fake_started_https_port
- self.fake_started_dns_port = fake_started_dns_port
- self.fake_replay_server = None
-
- def _ReplayServer(self, archive_path, host_ip, http_port, https_port,
- dns_port, replay_args):
- http_port = http_port or self.fake_started_http_port
- https_port = https_port or self.fake_started_https_port
- dns_port = (None if dns_port is None else
- (dns_port or self.fake_started_dns_port))
- self.fake_replay_server = FakeReplayServer(
- archive_path, host_ip, http_port, https_port, dns_port, replay_args)
- return self.fake_replay_server
-
-
-class NetworkControllerBackendTest(unittest.TestCase):
-
- def setUp(self):
- self.browser_backend = FakeBrowserBackend(
- http_ports=(0, 0),
- https_ports=(0, 0),
- dns_ports=None)
- self.network_controller_backend = TestNetworkControllerBackend(
- FakePlatformBackend(),
- fake_started_http_port=222,
- fake_started_https_port=444,
- fake_started_dns_port=None)
-
- def testSameArgsReuseServer(self):
- b = self.network_controller_backend
- with tempfile.NamedTemporaryFile() as temp_file:
- archive_path = temp_file.name
-
- # Create Replay server.
- b.SetReplayArgs(archive_path, wpr_modes.WPR_REPLAY, '3g', ['--some-arg'])
- b.UpdateReplay(self.browser_backend)
-
- self.assertEqual(archive_path, b.fake_replay_server.archive_path)
- self.assertEqual('123.321.123.321', b.fake_replay_server.host_ip)
- self.assertEqual(
- ['--some-arg', '--net=3g', '--inject_scripts='],
- b.fake_replay_server.replay_args)
- self.assertEqual(222, b.wpr_http_device_port)
- self.assertEqual(444, b.wpr_https_device_port)
-
- # Reuse Replay server.
- fake_replay_server = b.fake_replay_server
- b.SetReplayArgs(archive_path, wpr_modes.WPR_REPLAY, '3g', ['--some-arg'])
- b.UpdateReplay(self.browser_backend)
-
- self.assertIs(fake_replay_server, b.fake_replay_server)
- b.StopReplay()
- self.assertTrue(b.fake_replay_server.is_stopped)
-
- def testDifferentArgsUseDifferentServer(self):
- b = self.network_controller_backend
- with tempfile.NamedTemporaryFile() as temp_file:
- archive_file = temp_file.name
-
- # Create Replay server.
- b.SetReplayArgs(archive_file, wpr_modes.WPR_REPLAY, '3g', ['--some-arg'])
- b.UpdateReplay(self.browser_backend)
-
- self.assertEqual(
- ['--some-arg', '--net=3g', '--inject_scripts='],
- b.fake_replay_server.replay_args)
- self.assertEqual(222, b.wpr_http_device_port)
- self.assertEqual(444, b.wpr_https_device_port)
-
- # If Replay restarts, it uses these ports when passed "0" for ports.
- b.fake_started_http_port = 212
- b.fake_started_https_port = 323
- b.fake_started_dns_port = None
-
- # Create different Replay server (set netsim to None instead of 3g).
- fake_replay_server = b.fake_replay_server
- b.SetReplayArgs(archive_file, wpr_modes.WPR_REPLAY, None, ['--some-arg'])
- b.UpdateReplay(self.browser_backend)
-
- self.assertIsNot(fake_replay_server, b.fake_replay_server)
- self.assertTrue(fake_replay_server.is_stopped)
- self.assertFalse(b.fake_replay_server.is_stopped)
-
- self.assertEqual(
- ['--some-arg', '--inject_scripts='],
- b.fake_replay_server.replay_args)
- self.assertEqual(212, b.wpr_http_device_port)
- self.assertEqual(323, b.wpr_https_device_port)
- b.StopReplay()
- self.assertTrue(b.fake_replay_server.is_stopped)
-
- def testUpdateReplayWithoutArchivePathDoesNotStopReplay(self):
- b = TestNetworkControllerBackend(
- FakePlatformBackend(),
- fake_started_http_port=222,
- fake_started_https_port=444,
- fake_started_dns_port=None)
- with tempfile.NamedTemporaryFile() as temp_file:
- archive_file = temp_file.name
- # Create Replay server.
- b.SetReplayArgs(archive_file, wpr_modes.WPR_REPLAY, '3g', ['--some-arg'])
- browser_backend = FakeBrowserBackend(
- http_ports=(0, 0), https_ports=(0, 0), dns_ports=None)
- b.UpdateReplay(browser_backend)
- self.assertFalse(b.fake_replay_server.is_stopped)
- b.SetReplayArgs(None, wpr_modes.WPR_REPLAY, '3g', ['--some-arg'])
- b.UpdateReplay()
- self.assertFalse(b.fake_replay_server.is_stopped)
-
- def testUpdateReplayWithoutArgsIsOkay(self):
- b = self.network_controller_backend
- b.UpdateReplay(self.browser_backend) # does not raise
-
- def testBadArchivePathRaises(self):
- b = self.network_controller_backend
- b.SetReplayArgs('/tmp/nonexistant', wpr_modes.WPR_REPLAY, '3g', [])
- with self.assertRaises(network_controller_backend.ArchiveDoesNotExistError):
- b.UpdateReplay(self.browser_backend)
-
- def testBadArchivePathOnRecordIsOkay(self):
- """No ArchiveDoesNotExistError for record mode."""
- b = self.network_controller_backend
- b.SetReplayArgs('/tmp/nonexistant', wpr_modes.WPR_RECORD, '3g', [])
- b.UpdateReplay(self.browser_backend) # does not raise
-
- def testModeOffDoesNotCreateReplayServer(self):
- b = self.network_controller_backend
- b.SetReplayArgs('/tmp/nonexistant', wpr_modes.WPR_OFF, '3g', [])
- b.UpdateReplay(self.browser_backend)
- self.assertIsNone(b.fake_replay_server)
-
- def testSameBrowserUsesSamePorts(self):
- b = self.network_controller_backend
- with tempfile.NamedTemporaryFile() as temp_file:
- archive_path = temp_file.name
-
- # Create Replay server.
- b.SetReplayArgs(archive_path, wpr_modes.WPR_REPLAY, '3g', ['--some-arg'])
- b.UpdateReplay(self.browser_backend)
-
- self.assertEqual(archive_path, b.fake_replay_server.archive_path)
- self.assertEqual('123.321.123.321', b.fake_replay_server.host_ip)
- self.assertEqual(
- ['--some-arg', '--net=3g', '--inject_scripts='],
- b.fake_replay_server.replay_args)
- self.assertEqual(222, b.wpr_http_device_port)
- self.assertEqual(444, b.wpr_https_device_port)
-
- # If Replay restarts, it uses these ports when passed "0" for ports.
- b.fake_started_http_port = 212
- b.fake_started_https_port = 434
- b.fake_started_dns_port = None
-
- # Reuse Replay server.
- fake_replay_server = b.fake_replay_server
- b.SetReplayArgs(archive_path, wpr_modes.WPR_REPLAY, None, ['--NEW-ARG'])
- b.UpdateReplay() # no browser backend means use the previous one
-
- # Even though WPR is restarted, it uses the same ports because
- # the browser was configured to a particular port set.
- self.assertIsNot(fake_replay_server, b.fake_replay_server)
- self.assertEqual(222, b.wpr_http_device_port)
- self.assertEqual(444, b.wpr_https_device_port)
- b.StopReplay()
- self.assertTrue(b.fake_replay_server.is_stopped)
-
-# pylint: disable=protected-access
-class ForwarderPortPairsTest(unittest.TestCase):
- def testZeroIsOkayForRemotePorts(self):
- started_ports = (8080, 8443, None)
- wpr_port_pairs = forwarders.PortPairs(
- http=forwarders.PortPair(0, 0),
- https=forwarders.PortPair(0, 0),
- dns=None)
- expected_port_pairs = forwarders.PortPairs(
- http=forwarders.PortPair(8080, 0),
- https=forwarders.PortPair(8443, 0),
- dns=None)
- self.assertEqual(
- expected_port_pairs,
- network_controller_backend._ForwarderPortPairs(started_ports,
- wpr_port_pairs))
-
- def testCombineStartedAndRemotePorts(self):
- started_ports = (8888, 4343, 5353)
- wpr_port_pairs = forwarders.PortPairs(
- http=forwarders.PortPair(0, 80),
- https=forwarders.PortPair(0, 443),
- dns=forwarders.PortPair(0, 53))
- expected_port_pairs = forwarders.PortPairs(
- http=forwarders.PortPair(8888, 80),
- https=forwarders.PortPair(4343, 443),
- dns=forwarders.PortPair(5353, 53))
- self.assertEqual(
- expected_port_pairs,
- network_controller_backend._ForwarderPortPairs(started_ports,
- wpr_port_pairs))

Powered by Google App Engine
This is Rietveld 408576698