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)