| 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')
|
|
|