def testUpdate(self, getStubMock):
stubMock = mock.Mock()
stubMock.Update.return_value = service_pb2.ServiceUpdateResponse()
getStubMock.return_value = stubMock
wrapper = opencue.wrappers.service.Service(service=service_pb2.Service(
name=TEST_SERVICE_NAME))
wrapper.update()
stubMock.Update.assert_called_with(
service_pb2.ServiceUpdateRequest(service=wrapper.data), timeout=mock.ANY)
def testUpdate(self, stubMock):
stubMock.GetService.return_value = service_pb2.ServiceGetServiceResponse(
service=self.service)
stubMock.Update.return_value = service_pb2.ServiceUpdateResponse()
updatedService = opencue.api.getService(self.testName)
updatedService.setTags(['util'])
updatedService.update()
stubMock.Update.assert_called_with(
service_pb2.ServiceUpdateRequest(service=updatedService.data), timeout=mock.ANY)
def update(self):
"""Updates the service database record with the current instance data."""
return self.stub.Update(
service_pb2.ServiceUpdateRequest(service=self.data),
timeout=Cuebot.Timeout)
def update(self):
return self.stub.Update(
service_pb2.ServiceUpdateRequest(service=self.data),
timeout=Cuebot.Timeout)
