Removing flake in TemporaryHTTPServer.

tools/telemetry/telemetry/cros_interface.py

 # Copyright (c) 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.
 """A wrapper around ssh for common operations on a CrOS-based device"""
 import logging
 import os
 import re
 import subprocess
 import sys
 import time
 import tempfile
 
 from telemetry import util
 
-_next_remote_port = 9224
-
 # TODO(nduca): This whole file is built up around making individual ssh calls
 # for each operation. It really could get away with a single ssh session built
 # around pexpect, I suspect, if we wanted it to be faster. But, this was
 # convenient.
 
 def RunCmd(args, cwd=None, quiet=False):
   """Opens a subprocess to execute a program and returns its return value.
 
   Args:
     args: A string or a sequence of program arguments. The program to execute is
@@ skipping 339 matching lines @@
     return running
 
   def GetCmdOutput(self, args, quiet=False):
     stdout, stderr = self.GetAllCmdOutput(args, quiet=True)
     assert stderr == ''
     if not quiet:
       logging.debug("GetCmdOutput(%s)->%s" % (repr(args), stdout))
     return stdout
 
   def GetRemotePort(self):
-    global _next_remote_port
-    port = _next_remote_port
-    _next_remote_port += 1
-    return port
+    netstat = self.GetAllCmdOutput(['netstat', '-ant'])
+    netstat = netstat[0].split('\n')
+    ports_in_use = []
+
+    for line in netstat[2:]:
+      if not line:
+        continue
+      address_in_use = line.split()[3]
+      port_in_use = address_in_use.split(':')[-1]
+      ports_in_use.append(int(port_in_use))
+
+    return sorted(ports_in_use)[-1] + 1
+
+  def IsHTTPServerRunningOnPort(self, port):
+    wget_output = self.GetAllCmdOutput(
+        ['wget', 'localhost:%i' % (port), '-T1', '-t1'])
+
+    if 'Connection refused' in wget_output[1]:
+      return False
+
+    return True