Index: third_party/WebKit/Tools/Scripts/webkitpy/tool/commands/rebaseline_from_try_jobs_unittest.py |
diff --git a/third_party/WebKit/Tools/Scripts/webkitpy/tool/commands/rebaseline_from_try_jobs_unittest.py b/third_party/WebKit/Tools/Scripts/webkitpy/tool/commands/rebaseline_from_try_jobs_unittest.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a9d67d2e106e8a616876fc39a09647790360af01 |
--- /dev/null |
+++ b/third_party/WebKit/Tools/Scripts/webkitpy/tool/commands/rebaseline_from_try_jobs_unittest.py |
@@ -0,0 +1,53 @@ |
+# 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 |
+ |
+from webkitpy.common.net.web_mock import MockWeb |
+from webkitpy.common.system.outputcapture import OutputCapture |
+from webkitpy.tool.commands.rebaseline_from_try_jobs import RebaselineFromTryJobs |
+from webkitpy.tool.commands.rebaseline_unittest import BaseTestCase |
+from webkitpy.tool.mocktool import MockOptions |
+ |
+ |
+class RebaselineFromTryJobsTest(BaseTestCase): |
+ command_constructor = RebaselineFromTryJobs |
+ |
+ def setUp(self): |
+ super(RebaselineFromTryJobsTest, self).setUp() |
+ self.command.web = MockWeb(urls={ |
+ 'https://codereview.chromium.org/api/11112222': json.dumps({ |
+ 'patchsets': [1, 2], |
+ }), |
+ 'https://codereview.chromium.org/api/11112222/2': json.dumps({ |
+ 'try_job_results': [ |
+ { |
+ 'builder': 'win_chromium_rel_ng', |
+ 'master': 'tryserver.chromium.win', |
+ 'buildnumber': 5000, |
+ }, |
+ ], |
+ }), |
+ }) |
+ |
+ def test_execute_with_issue_number_given(self): |
+ oc = OutputCapture() |
+ try: |
+ oc.capture_output() |
+ self.command.execute(MockOptions(issue=11112222), [], self.tool) |
+ finally: |
+ _, _, logs = oc.restore_output() |
+ self.assertEqual(logs, ('Getting results for Rietveld issue 11112222.\n' |
+ ' Builder: win_chromium_rel_ng\n' |
+ ' Master: tryserver.chromium.win\n' |
+ ' Build: 5000\n')) |
+ |
+ def test_execute_with_no_issue_number(self): |
+ oc = OutputCapture() |
+ try: |
+ oc.capture_output() |
+ self.command.execute(MockOptions(issue=None), [], self.tool) |
+ finally: |
+ _, _, logs = oc.restore_output() |
+ self.assertEqual(logs, 'No issue number provided.\n') |