Add a webkit-patch command which gets a list of relevant try jobs.

third_party/WebKit/Tools/Scripts/webkitpy/common/net/rietveld_unittest.py

Index: third_party/WebKit/Tools/Scripts/webkitpy/common/net/rietveld_unittest.py
diff --git a/third_party/WebKit/Tools/Scripts/webkitpy/common/net/rietveld_unittest.py b/third_party/WebKit/Tools/Scripts/webkitpy/common/net/rietveld_unittest.py
new file mode 100644
index 0000000000000000000000000000000000000000..4614c5ac35c4b0ea19bca507bc123cf9aeaa68bb
--- /dev/null
+++ b/third_party/WebKit/Tools/Scripts/webkitpy/common/net/rietveld_unittest.py
@@ -0,0 +1,88 @@
+# Copyright 2016 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 json
+import logging
+import unittest
+import urllib2
+
+from webkitpy.common.net.rietveld import latest_try_jobs, TryJob
+from webkitpy.common.net.web_mock import MockWeb
+from webkitpy.common.system.outputcapture import OutputCapture
+
+
+_log = logging.getLogger(__name__)
+
+
+class RietveldTest(unittest.TestCase):
+
+    def setUp(self):
+        self.web = MockWeb(urls={
+            'https://codereview.chromium.org/api/11112222': json.dumps({
+                'patchsets': [1, 2, 3],
+            }),
+            'https://codereview.chromium.org/api/11112222/2': json.dumps({
+                'try_job_results': [
+                    {
+                        'builder': 'some builder',
+                        'master': 'some master',
+                        'buildnumber': 10,
+                    },
+                    {
+                        'builder': 'my builder',
+                        'master': 'my master',
+                        'buildnumber': 50,
+                    },
+                ],
+            }),
+            'https://codereview.chromium.org/api/11112222/3': json.dumps({
+                'try_job_results': [
+                    {
+                        'builder': 'some builder',
+                        'master': 'some master',
+                        'buildnumber': 20,
+                    },
+                    {
+                        'builder': 'my builder',
+                        'master': 'my master',
+                        'buildnumber': 60,
+                    },
+                ],
+            }),
+            'https://codereview.chromium.org/api/11113333': 'my non-json contents',
+        })
+
+    def test_latest_try_jobs(self):
+        self.assertEqual(
+            latest_try_jobs(11112222, ('my builder', 'my other builder'), self.web),
+            [TryJob('my builder', 'my master', 60)])
+
+    def test_latest_try_jobs_http_error(self):
+        def raise_error(_):
+            raise urllib2.URLError('Some request error message')
+        self.web.get_binary = raise_error
+        oc = OutputCapture()
+        try:
+            oc.capture_output()
+            self.assertEqual(latest_try_jobs(11112222, ('my builder',), self.web), [])
+        finally:
+            _, _, logs = oc.restore_output()
+        self.assertEqual(logs, 'Request failed to URL: https://codereview.chromium.org/api/11112222\n')
+
+    def test_latest_try_jobs_non_json_response(self):
+        oc = OutputCapture()
+        try:
+            oc.capture_output()
+            self.assertEqual(latest_try_jobs(11113333, ('my builder',), self.web), [])
+        finally:
+            _, _, logs = oc.restore_output()
+        self.assertEqual(logs, 'Invalid JSON: my non-json contents\n')
+
+    def test_latest_try_jobs_with_patchset(self):
+        self.assertEqual(
+            latest_try_jobs(11112222, ('my builder', 'my other builder'), self.web, patchset_number=2),
+            [TryJob('my builder', 'my master', 50)])
+
+    def test_latest_try_jobs_no_relevant_builders(self):
+        self.assertEqual(latest_try_jobs(11112222, ('foo', 'bar'), self.web), [])