def testGetRecentCompletedBuilds(self, mock_fn):
mock_fn.return_value = SearchBuildsResponse(
builds=[Build(number=34),
Build(number=33),
Build(number=32)])
self.assertEqual([34, 33, 32],
buildbot.GetRecentCompletedBuilds(
'm', 'b', RetryHttpClient()))
def testHttpRequestSuccess(self, mocked_Get):
http_client = RetryHttpClient()
expected_result = {
'git_sha': 'sha',
'repo_url': 'url',
}
self.assertDictEqual(expected_result,
crrev.RedirectByCommitPosition(http_client, 5000))
mocked_Get.assert_called_once_with(
'https://cr-rev.appspot.com/_ah/api/crrev/v1/redirect/5000')
def testListBuildersOnMaster(self, mocked_fn):
mocked_fn.return_value = json.dumps({
'builders': {
'builder_1': {},
'builder_2': {}
}})
self.assertEqual(
['builder_1', 'builder_2'],
buildbot.ListBuildersOnMaster('master', RetryHttpClient()))
def setUp(self):
super(BuildBotTest, self).setUp()
self.http_client = RetryHttpClient()
self.master_name = 'tryserver.m'
self.wf_master_name = 'm'
self.builder_name = 'b'
self.build_number = 123
self.step_name = 'browser_tests on platform'
self.stdout_stream = 'stdout_stream'
self.step_metadata_stream = 'step_metadata_stream'
def testDownloadJsonData(self, mock_fn_1, mock_fn_2):
mocked_response_json = {'a': 'a'}
mocked_response = json.dumps(mocked_response_json)
mock_fn_1.return_value = mocked_response
mock_fn_2.return_value = mocked_response_json
url = 'url'
data = {'data': 'data'}
http_client = RetryHttpClient()
response_json = rpc_util.DownloadJsonData(url, data, http_client)
self.assertEqual(response_json, mocked_response_json)
mock_fn_1.assert_called_once_with(url, data, http_client)
mock_fn_2.assert_called_once_with(mocked_response)
def setUp(self):
super(LogDogUtilTest, self).setUp()
self.http_client = RetryHttpClient()
self.master_name = 'tryserver.m'
self.builder_name = 'b'
self.build_number = 123
self.step_name = 'browser_tests on platform'
self.build_data = {
'result_details_json':
json.dumps({'properties': {
'log_location': 'logdog://h/p/path'
}})
}
self.stdout_stream = 'stdout_stream'
self.step_metadata_stream = 'step_metadata_stream'
def testGetRecentCompletedBuilds(self, mock_fn):
builders_data = {
'builders': {
'b': {
'cachedBuilds': [33, 32, 34, 35],
'currentBuilds': [35]
}
}
}
builders_json_data = json.dumps(builders_data)
compressed_file = io.BytesIO()
with gzip.GzipFile(fileobj=compressed_file, mode='w') as compressed:
compressed.write(builders_json_data)
data = {
'data': base64.b64encode(compressed_file.getvalue())
}
compressed_file.close()
mock_fn.return_value = json.dumps(data)
self.assertEqual(
[34, 33, 32],
buildbot.GetRecentCompletedBuilds('m', 'b', RetryHttpClient()))
def testHttpRequestFailure(self, mocked_Get):
http_client = RetryHttpClient()
self.assertIsNone(crrev.RedirectByCommitPosition(http_client, 5000))
mocked_Get.assert_called_once_with(
'https://cr-rev.appspot.com/_ah/api/crrev/v1/redirect/5000')
def testListBuildersOnMasterError(self, _):
self.assertEqual(
[],
buildbot.ListBuildersOnMaster('master', RetryHttpClient()))