def testFetchMissingPieces(self, mocked_fetch):
response_pb = rpc_pb2.SearchBuildsResponse()
build_pb = response_pb.builds.add()
build_pb.number = 123
build_pb.status = common_pb2.SCHEDULED
mocked_fetch.return_value = response_pb
builds = lkgr_lib.FetchBuildbucketBuildsForBuilder(
'project1/bucket1', 'builder1')
self.assertEquals(builds, [])
def testFetchBuildFailed(self, mocked_fetch):
response_pb = rpc_pb2.SearchBuildsResponse()
build_pb = response_pb.builds.add()
build_pb.number = 123
build_pb.status = common_pb2.FAILURE
build_pb.input.gitiles_commit.id = 'a' * 40
mocked_fetch.return_value = response_pb
builds = lkgr_lib.FetchBuildbucketBuildsForBuilder(
'project1/bucket1', 'builder1')
self.assertEquals(
builds, [lkgr_lib.Build(123, lkgr_lib.STATUS.FAILURE, 'a' * 40)])
def testProtoError(self, mocked_fetch):
mocked_fetch.side_effect = google.protobuf.message.DecodeError
builds = lkgr_lib.FetchBuildbucketBuildsForBuilder(
'project1/bucket1', 'builder1')
self.assertEquals(builds, None)
def testMalformattedBucket(self, mocked_fetch):
builds = lkgr_lib.FetchBuildbucketBuildsForBuilder(
'project1_bucket1', 'builder1')
self.assertEquals(builds, None)
mocked_fetch.assert_not_called()
def testFetchFailed(self, mocked_fetch):
mocked_fetch.side_effect = httplib2.HttpLib2Error
builds = lkgr_lib.FetchBuildbucketBuildsForBuilder(
'project1/bucket1', 'builder1')
self.assertEquals(builds, None)
