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..c8daeba5188a91ee205bfc76cbbe886142e1cd4d
--- /dev/null
+++ b/third_party/WebKit/Tools/Scripts/webkitpy/common/net/rietveld_unittest.py
@@ -0,0 +1,60 @@
+# 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 unittest
+
+from webkitpy.common.net.web_mock import MockWeb
+from webkitpy.common.net.rietveld import latest_try_jobs, TryJob
+
+
+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,
+                    },
+                ],
+            }),
+        })
+
+    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_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), [])