def testDropLimit(self, getStubMock):
test_limit_id = 'lll-llll-lll'
stubMock = mock.Mock()
stubMock.DropLimit.return_value = job_pb2.LayerDropLimitResponse()
getStubMock.return_value = stubMock
layer = opencue.wrappers.layer.Layer(
job_pb2.Layer(name=TEST_LAYER_NAME))
layer.dropLimit(test_limit_id)
stubMock.DropLimit.assert_called_with(job_pb2.LayerDropLimitRequest(
layer=layer.data, limit_id=test_limit_id),
timeout=mock.ANY)
def dropLimit(self, limit_id):
"""Remove a limit on the current layer."""
return self.stub.DropLimit(job_pb2.LayerDropLimitRequest(
layer=self.data, limit_id=limit_id),
timeout=Cuebot.Timeout)