def testDelete(self, stubMock): stubMock.GetService.return_value = service_pb2.ServiceGetServiceResponse( service=self.service) stubMock.Delete.return_value = service_pb2.ServiceDeleteResponse() opencue.api.getService(self.testName).delete() stubMock.Delete.assert_called_with( service_pb2.ServiceDeleteRequest(service=self.service), timeout=mock.ANY)
def testDelete(self, getStubMock): stubMock = mock.Mock() stubMock.Delete.return_value = service_pb2.ServiceDeleteResponse() getStubMock.return_value = stubMock service = opencue.wrappers.service.Service( service_pb2.Service(name=TEST_SERVICE_NAME)) service.delete() stubMock.Delete.assert_called_with( service_pb2.ServiceDeleteRequest(service=service.data), timeout=mock.ANY)
def delete(self): """Deletes the service.""" return self.stub.Delete( service_pb2.ServiceDeleteRequest(service=self.data), timeout=Cuebot.Timeout)