OLD | NEW |
1 # Copyright 2016 The Chromium Authors. All rights reserved. | 1 # Copyright 2016 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import json | 5 import json |
6 import logging | 6 import logging |
7 import unittest | 7 import unittest |
8 import urllib2 | 8 import urllib2 |
9 | 9 |
10 from webkitpy.common.net.rietveld import latest_try_jobs, TryJob, get_latest_try
_job_results | 10 from webkitpy.common.net.rietveld import latest_try_jobs, TryJob, get_latest_try
_job_results |
11 from webkitpy.common.net.web_mock import MockWeb | 11 from webkitpy.common.net.web_mock import MockWeb |
12 from webkitpy.common.system.outputcapture import OutputCapture | 12 from webkitpy.common.system.outputcapture import OutputCapture |
13 | 13 |
14 | 14 |
15 _log = logging.getLogger(__name__) | 15 _log = logging.getLogger(__name__) |
16 | 16 |
17 | 17 |
18 class RietveldTest(unittest.TestCase): | 18 class RietveldTest(unittest.TestCase): |
19 | 19 |
20 def setUp(self): | 20 def setUp(self): |
21 self.web = MockWeb(urls={ | 21 self.web = MockWeb(urls={ |
22 'https://codereview.chromium.org/api/11112222': json.dumps({ | 22 'https://codereview.chromium.org/api/11112222': json.dumps({ |
23 'patchsets': [1, 2, 3], | 23 'patchsets': [1, 2, 3], |
24 }), | 24 }), |
25 'https://codereview.chromium.org/api/11112222/2': json.dumps({ | 25 'https://codereview.chromium.org/api/11112222/2': json.dumps({ |
26 'try_job_results': [ | 26 'try_job_results': [ |
27 { | 27 { |
28 'builder': 'some builder', | 28 'builder': 'foo-builder', |
29 'master': 'some master', | 29 'master': 'master.tryserver.foo', |
30 'buildnumber': 10, | 30 'buildnumber': 10, |
31 'result': -1 | 31 'result': -1 |
32 }, | 32 }, |
33 { | 33 { |
34 'builder': 'my builder', | 34 'builder': 'bar-builder', |
35 'master': 'my master', | 35 'master': 'master.tryserver.bar', |
36 'buildnumber': 50, | 36 'buildnumber': 50, |
37 'results': 0 | 37 'results': 0 |
38 }, | 38 }, |
39 ], | 39 ], |
40 }), | 40 }), |
41 'https://codereview.chromium.org/api/11112222/3': json.dumps({ | 41 'https://codereview.chromium.org/api/11112222/3': json.dumps({ |
42 'try_job_results': [ | 42 'try_job_results': [ |
43 { | 43 { |
44 'builder': 'some builder', | 44 'builder': 'foo-builder', |
45 'master': 'some master', | 45 'master': 'master.tryserver.foo', |
46 'buildnumber': 20, | 46 'buildnumber': 20, |
47 'result': 1 | 47 'result': 1 |
48 }, | 48 }, |
49 { | 49 { |
50 'builder': 'my builder', | 50 'builder': 'bar-builder', |
51 'master': 'my master', | 51 'master': 'master.tryserver.bar', |
52 'buildnumber': 60, | 52 'buildnumber': 60, |
53 'result': 0 | 53 'result': 0 |
54 }, | 54 }, |
55 ], | 55 ], |
56 }), | 56 }), |
57 'https://codereview.chromium.org/api/11113333': 'my non-json content
s', | 57 'https://codereview.chromium.org/api/11113333': 'my non-JSON content
s', |
58 }) | 58 }) |
59 | 59 |
60 def test_latest_try_jobs(self): | 60 def test_latest_try_jobs(self): |
61 self.assertEqual( | 61 self.assertEqual( |
62 latest_try_jobs(11112222, ('my builder', 'my other builder'), self.w
eb), | 62 latest_try_jobs(11112222, ('bar-builder', 'other-builder'), self.web
), |
63 [TryJob('my builder', 'my master', 60)]) | 63 [TryJob('bar-builder', 'tryserver.bar', 60)]) |
64 | 64 |
65 def test_latest_try_jobs_http_error(self): | 65 def test_latest_try_jobs_http_error(self): |
66 def raise_error(_): | 66 def raise_error(_): |
67 raise urllib2.URLError('Some request error message') | 67 raise urllib2.URLError('Some request error message') |
68 self.web.get_binary = raise_error | 68 self.web.get_binary = raise_error |
69 oc = OutputCapture() | 69 oc = OutputCapture() |
70 try: | 70 try: |
71 oc.capture_output() | 71 oc.capture_output() |
72 self.assertEqual(latest_try_jobs(11112222, ('my builder',), self.web
), []) | 72 self.assertEqual(latest_try_jobs(11112222, ('bar-builder',), self.we
b), []) |
73 finally: | 73 finally: |
74 _, _, logs = oc.restore_output() | 74 _, _, logs = oc.restore_output() |
75 self.assertEqual(logs, 'Request failed to URL: https://codereview.chromi
um.org/api/11112222\n') | 75 self.assertEqual(logs, 'Request failed to URL: https://codereview.chromi
um.org/api/11112222\n') |
76 | 76 |
77 def test_latest_try_jobs_non_json_response(self): | 77 def test_latest_try_jobs_non_json_response(self): |
78 oc = OutputCapture() | 78 oc = OutputCapture() |
79 try: | 79 try: |
80 oc.capture_output() | 80 oc.capture_output() |
81 self.assertEqual(latest_try_jobs(11113333, ('my builder',), self.web
), []) | 81 self.assertEqual(latest_try_jobs(11113333, ('bar-builder',), self.we
b), []) |
82 finally: | 82 finally: |
83 _, _, logs = oc.restore_output() | 83 _, _, logs = oc.restore_output() |
84 self.assertEqual(logs, 'Invalid JSON: my non-json contents\n') | 84 self.assertEqual(logs, 'Invalid JSON: my non-JSON contents\n') |
85 | 85 |
86 def test_latest_try_jobs_with_patchset(self): | 86 def test_latest_try_jobs_with_patchset(self): |
87 self.assertEqual( | 87 self.assertEqual( |
88 latest_try_jobs(11112222, ('my builder', 'my other builder'), self.w
eb, patchset_number=2), | 88 latest_try_jobs(11112222, ('bar-builder', 'other-builder'), self.web
, patchset_number=2), |
89 [TryJob('my builder', 'my master', 50)]) | 89 [TryJob('bar-builder', 'tryserver.bar', 50)]) |
90 | 90 |
91 def test_latest_try_jobs_no_relevant_builders(self): | 91 def test_latest_try_jobs_no_relevant_builders(self): |
92 self.assertEqual(latest_try_jobs(11112222, ('foo', 'bar'), self.web), []
) | 92 self.assertEqual(latest_try_jobs(11112222, ('foo', 'bar'), self.web), []
) |
93 | 93 |
94 def test_get_latest_try_job_results(self): | 94 def test_get_latest_try_job_results(self): |
95 self.assertEqual(get_latest_try_job_results(11112222, self.web), {'some
builder': 1, 'my builder': 0}) | 95 self.assertEqual(get_latest_try_job_results(11112222, self.web), {'foo-b
uilder': 1, 'bar-builder': 0}) |
OLD | NEW |