Index: telemetry/telemetry/core/local_server.py |
diff --git a/telemetry/telemetry/core/local_server.py b/telemetry/telemetry/core/local_server.py |
index 5c20c455f425a3fec67bd9aeacb2dbc9d992e8c4..c8557d969afa1245ddaaefb3168be61fde3065ab 100644 |
--- a/telemetry/telemetry/core/local_server.py |
+++ b/telemetry/telemetry/core/local_server.py |
@@ -82,13 +82,15 @@ |
stdout=subprocess.PIPE) |
named_ports = self._GetNamedPortsFromBackend() |
- named_port_pair_map = {'http': None, 'https': None, 'dns': None} |
- for name, port in named_ports: |
- assert name in named_port_pair_map, '%s forwarding is unsupported' % name |
- named_port_pair_map[name] = (forwarders.PortPair( |
- port, local_server_controller.GetRemotePort(port))) |
+ http_port = None |
+ for p in named_ports: |
+ if p.name == 'http': |
+ http_port = p.port |
+ assert http_port and len(named_ports) == 1, ( |
+ 'Only http port is supported: %s' % named_ports) |
self.forwarder = local_server_controller.CreateForwarder( |
- forwarders.PortPairs(**named_port_pair_map)) |
+ forwarders.PortPair(http_port, |
+ local_server_controller.GetRemotePort(http_port))) |
def _GetNamedPortsFromBackend(self): |
named_ports_json = None |
@@ -178,8 +180,8 @@ |
import traceback |
traceback.print_exc() |
- def CreateForwarder(self, port_pairs): |
- return self._platform_backend.forwarder_factory.Create(port_pairs) |
+ def CreateForwarder(self, port_pair): |
+ return self._platform_backend.forwarder_factory.Create(port_pair) |
def GetRemotePort(self, port): |
return self._platform_backend.GetRemotePort(port) |