def test_get_try_job_results_for_patchset(self): put_builds_response = { 'builds': [ {'id': '1', 'status': 'SCHEDULED'}, {'id': '2', 'status': 'SCHEDULED'}, ], } self.fake_responses = [ {'delegation_token': 'deltok', 'validity_duration': 1000}, put_builds_response, ] actual_builds = buildbucket.get_builds_for_patchset_async( 'project', 1, 2).get_result() self.assertEqual(actual_builds, put_builds_response['builds']) mint_token_req_body = self.requests[0][1]['payload'] self.assertEqual(mint_token_req_body, { 'audience': ['user:test@localhost'], 'services': ['service:cr-buildbucket-test'], 'impersonate': 'user:[email protected]', }) put_builds_req_headers = self.requests[1][1]['headers'] self.assertEqual( put_builds_req_headers['X-Delegation-Token-V1'], 'deltok')
def test_get_try_job_results_for_patchset(self): put_builds_response = { 'builds': [ {'id': '1', 'status': 'SCHEDULED'}, {'id': '2', 'status': 'SCHEDULED'}, ], } self.fake_responses = [ {'token': 'deltok', 'delegationSubtoken': {'validityDuration': 18000}}, put_builds_response, ] actual_builds = buildbucket.get_builds_for_patchset_async( 'project', 1, 2).get_result() self.assertEqual(actual_builds, put_builds_response['builds']) mint_token_req_body = self.requests[0][1]['payload'] self.assertEqual(mint_token_req_body, { 'audience': ['REQUESTOR'], 'services': ['service:cr-buildbucket-test'], 'delegatedIdentity': 'user:[email protected]', 'validityDuration': 18000, }) put_builds_req_headers = self.requests[1][1]['headers'] self.assertEqual( put_builds_req_headers['X-Delegation-Token-V1'], 'deltok')
def test_get_try_job_results_for_patchset(self): response_data = { 'builds': [ {'id': '1', 'status': 'SCHEDULED'}, {'id': '2', 'status': 'SCHEDULED'}, ] } self.fake_responses = [response_data] actual_builds = buildbucket.get_builds_for_patchset_async( 'project', 1, 2).get_result() self.assertEqual(actual_builds, response_data['builds'])