Import wpt@3c8896ae408c8fd594979da7c99970029e7856a7

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

 from __future__ import print_function
 
 import argparse
 import json
 import logging
 import os
 import re
 import stat
 import subprocess
 import sys
@@ skipping 71 matching lines @@
         self.name = name
 
     def __enter__(self):
         print("travis_fold:start:%s" % self.name, file=sys.stderr)
 
     def __exit__(self, type, value, traceback):
         print("travis_fold:end:%s" % self.name, file=sys.stderr)
 
 
 class GitHub(object):
-    def __init__(self, org, repo, token):
+    def __init__(self, org, repo, token, browser):
         self.token = token
         self.headers = {"Accept": "application/vnd.github.v3+json"}
         self.auth = (self.token, "x-oauth-basic")
         self.org = org
         self.repo = repo
         self.base_url = "https://api.github.com/repos/%s/%s/" % (org, repo)
+        self.browser = browser
 
     def _headers(self, headers):
         if headers is None:
             headers = {}
         rv = self.headers.copy()
         rv.update(headers)
         return rv
 
     def post(self, url, data, headers=None):
         logger.debug("POST %s" % url)
         if data is not None:
             data = json.dumps(data)
         resp = requests.post(
             url,
             data=data,
             headers=self._headers(headers),
             auth=self.auth
         )
         resp.raise_for_status()
         return resp
 
+    def patch(self, url, data, headers=None):
+        logger.debug("PATCH %s" % url)
+        if data is not None:
+            data = json.dumps(data)
+        resp = requests.patch(
+            url,
+            data=data,
+            headers=self._headers(headers),
+            auth=self.auth
+        )
+        resp.raise_for_status()
+        return resp
+
     def get(self, url, headers=None):
         logger.debug("GET %s" % url)
         resp = requests.get(
             url,
             headers=self._headers(headers),
             auth=self.auth
         )
         resp.raise_for_status()
         return resp
 
     def post_comment(self, issue_number, body):
-        url = urljoin(self.base_url, "issues/%s/comments" % issue_number)
-        return self.post(url, {"body": body})
+        user = self.get(urljoin(self.base_url, "/user")).json()
+        issue_comments_url = urljoin(self.base_url, "issues/%s/comments" % issue_number)
+        comments = self.get(issue_comments_url).json()
+        title_line = "# %s #" % self.browser.title()
+        data = {"body": body}
+        for comment in comments:
+            if (comment["user"]["login"] == user["login"] and
+                comment["body"].startswith(title_line)):
+                comment_url = urljoin(self.base_url, "issues/comments/%s" % comment["id"])
+                self.patch(comment_url, data)
+                break
+        else:
+            self.post(issue_comments_url, data)
 
     def releases(self):
         url = urljoin(self.base_url, "releases/latest")
         return self.get(url)
 
 
 class GitHubCommentHandler(logging.Handler):
     def __init__(self, github, pull_number):
         logging.Handler.__init__(self)
         self.github = github
@@ skipping 17 matching lines @@
 
     def __init__(self, github_token):
         self.github_token = github_token
 
 
 class Firefox(Browser):
     product = "firefox"
 
     def install(self):
         call("pip", "install", "-r", "w3c/wptrunner/requirements_firefox.txt")
-        resp = get("https://archive.mozilla.org/pub/firefox/nightly/latest-mozilla-central/firefox-52.0a1.en-US.linux-x86_64.tar.bz2")
+        resp = get("https://archive.mozilla.org/pub/firefox/nightly/latest-mozilla-central/firefox-53.0a1.en-US.linux-x86_64.tar.bz2")
         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("w3c", "wptrunner", "requirements_firefox.txt"))
 
     def _latest_geckodriver_version(self):
@@ skipping 56 matching lines @@
 
 def get(url):
     logger.debug("GET %s" % url)
     resp = requests.get(url, stream=True)
     resp.raise_for_status()
     return resp
 
 
 def call(*args):
     logger.debug("%s" % " ".join(args))
-    return subprocess.check_output(args)
+    try:
+        return subprocess.check_output(args)
+    except subprocess.CalledProcessError as e:
+        logger.critical("%s exited with return code %i" %
+                        (e.cmd, e.returncode))
+        logger.critical(e.output)
+        raise
 
 
 def get_git_cmd(repo_path):
     def git(cmd, *args):
         full_cmd = ["git", cmd] + list(args)
         try:
             return subprocess.check_output(full_cmd, cwd=repo_path, stderr=subprocess.STDOUT)
         except subprocess.CalledProcessError as e:
             logger.error("Git command exited with status %i" % e.returncode)
             logger.error(e.output)
@@ skipping 23 matching lines @@
     with zipfile.ZipFile(fileobj) as zip_data:
         for info in zip_data.infolist():
             zip_data.extract(info)
             perm = info.external_attr >> 16 & 0x1FF
             os.chmod(info.filename, perm)
 
 
 def setup_github_logging(args):
     gh_handler = None
     if args.comment_pr:
-        github = GitHub("w3c", "web-platform-tests", args.gh_token)
+        github = GitHub("w3c", "web-platform-tests", args.gh_token, args.browser)
         try:
             pr_number = int(args.comment_pr)
         except ValueError:
             pass
         else:
             gh_handler = GitHubCommentHandler(github, pr_number)
             gh_handler.setLevel(logging.INFO)
             logger.debug("Setting up GitHub logging")
             logger.addHandler(gh_handler)
     else:
@@ skipping 288 matching lines @@
     return retcode
 
 
 if __name__ == "__main__":
     try:
         retcode = main()
     except:
         raise
     else:
         sys.exit(retcode)