Import wpt@a7e9c2abcf65b78fcf1c246fec6681c74e1bc352

third_party/WebKit/LayoutTests/external/wpt/check_stability.py

 from __future__ import print_function
 
 import argparse
 import logging
 import os
 import re
 import stat
 import subprocess
 import sys
 import tarfile
@@ skipping 119 matching lines @@
     # Value must be boxed to support modification from inner function scope
     count = [0]
     capacity -= 2 + len(warning_msg)
     stderr = sys.stderr
 
     def on_write(handle, msg):
         length = len(msg)
         count[0] += length
 
         if count[0] > capacity:
-            sys.stdout.disable()
-            sys.stderr.disable()
+            wrapped_stdout.disable()
+            wrapped_stderr.disable()
             handle.write(msg[0:capacity - count[0]])
             handle.flush()
             stderr.write("\n%s\n" % warning_msg)
             return False
 
         return True
 
-    sys.stdout = FilteredIO(sys.stdout, on_write)
-    sys.stderr = FilteredIO(sys.stderr, on_write)
+    # Store local references to the replaced streams to guard against the case
+    # where other code replace the global references.
+    sys.stdout = wrapped_stdout = FilteredIO(sys.stdout, on_write)
+    sys.stderr = wrapped_stderr = FilteredIO(sys.stderr, on_write)
 
 
 class Browser(object):
     __metaclass__ = ABCMeta
 
     @abstractmethod
     def install(self):
         return NotImplemented
 
     @abstractmethod
@@ skipping 15 matching lines @@
     Includes installation, webdriver installation, and wptrunner setup methods.
     """
 
     product = "firefox"
     binary = "%s/firefox/firefox"
     platform_ini = "%s/firefox/platform.ini"
 
     def install(self):
         """Install Firefox."""
         call("pip", "install", "-r", os.path.join(wptrunner_root, "requirements_firefox.txt"))
-        resp = get("https://archive.mozilla.org/pub/firefox/nightly/latest-mozilla-central/firefox-53.0a1.en-US.linux-x86_64.tar.bz2")
+        index = get("https://archive.mozilla.org/pub/firefox/nightly/latest-mozilla-central/")
+        latest = re.compile("<a[^>]*>(firefox-\d+\.\d(?:\w\d)?.en-US.linux-x86_64\.tar\.bz2)</a>")
+        filename = latest.search(index.text).group(1)
+        resp = get("https://archive.mozilla.org/pub/firefox/nightly/latest-mozilla-central/%s" %
+                   filename)
         untar(resp.raw)
 
         if not os.path.exists("profiles"):
             os.mkdir("profiles")
         with open(os.path.join("profiles", "prefs_general.js"), "wb") as f:
             resp = get("https://hg.mozilla.org/mozilla-central/raw-file/tip/testing/profiles/prefs_general.js")
             f.write(resp.content)
         call("pip", "install", "-r", os.path.join(wptrunner_root, "requirements_firefox.txt"))
 
     def _latest_geckodriver_version(self):
@@ skipping 559 matching lines @@
     return retcode
 
 
 if __name__ == "__main__":
     try:
         retcode = main()
     except:
         raise
     else:
         sys.exit(retcode)