def testGetLimits(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetAll.return_value = limit_pb2.LimitGetAllResponse(
limits=[limit_pb2.Limit(name=TEST_LIMIT_NAME)])
getStubMock.return_value = stubMock
limits = opencue.api.getLimits()
stubMock.GetAll.assert_called_with(limit_pb2.LimitGetAllRequest(), timeout=mock.ANY)
self.assertEqual(len(limits), 1)
self.assertEqual(limits[0].name(), TEST_LIMIT_NAME)
def testFindLimit(self, getStubMock):
stubMock = mock.Mock()
stubMock.Find.return_value = limit_pb2.LimitFindResponse(
limit=limit_pb2.Limit(name=TEST_LIMIT_NAME, max_value=42))
getStubMock.return_value = stubMock
limit = opencue.api.findLimit(TEST_LIMIT_NAME)
self.assertEqual(TEST_LIMIT_NAME, limit.name())
self.assertEqual(42, limit.maxValue())
stubMock.Find.assert_called_with(
limit_pb2.LimitFindRequest(name=TEST_LIMIT_NAME),
timeout=mock.ANY)
def testDelete(self, getStubMock):
stubMock = mock.Mock()
stubMock.Delete.return_value = limit_pb2.LimitDeleteResponse()
getStubMock.return_value = stubMock
limit = opencue.wrappers.limit.Limit(
limit_pb2.Limit(name=TEST_LIMIT_NAME,
max_value=TEST_LIMIT_MAX_VALUE))
limit.delete()
stubMock.Delete.assert_called_with(
limit_pb2.LimitDeleteRequest(name=TEST_LIMIT_NAME),
timeout=mock.ANY)
def testRename(self, getStubMock):
test_new_name = 'new_name'
stubMock = mock.Mock()
stubMock.Rename.return_value = limit_pb2.LimitRenameResponse()
getStubMock.return_value = stubMock
limit = opencue.wrappers.limit.Limit(
limit_pb2.Limit(name=TEST_LIMIT_NAME,
max_value=TEST_LIMIT_MAX_VALUE))
limit.rename(test_new_name)
stubMock.Rename.assert_called_with(limit_pb2.LimitRenameRequest(
old_name=TEST_LIMIT_NAME, new_name=test_new_name),
timeout=mock.ANY)
def testGet(self, getStubMock):
stubMock = mock.Mock()
stubMock.Get.return_value = limit_pb2.LimitGetResponse(
limit=limit_pb2.Limit(name=TEST_LIMIT_NAME,
max_value=TEST_LIMIT_MAX_VALUE))
getStubMock.return_value = stubMock
limit = opencue.wrappers.limit.Limit()
responseLimit = limit.get(TEST_LIMIT_ID)
stubMock.Get.assert_called_with(
limit_pb2.LimitGetRequest(id=TEST_LIMIT_ID), timeout=mock.ANY)
self.assertEqual(responseLimit.name(), TEST_LIMIT_NAME)
self.assertEqual(responseLimit.maxValue(), TEST_LIMIT_MAX_VALUE)
def testSetMaxValue(self, getStubMock):
max_value = 16
stubMock = mock.Mock()
stubMock.SetMaxValue.return_value = limit_pb2.LimitSetMaxValueResponse(
)
getStubMock.return_value = stubMock
limit = opencue.wrappers.limit.Limit(
limit_pb2.Limit(name=TEST_LIMIT_NAME,
max_value=TEST_LIMIT_MAX_VALUE))
limit.setMaxValue(max_value)
stubMock.SetMaxValue.assert_called_with(
limit_pb2.LimitSetMaxValueRequest(name=TEST_LIMIT_NAME,
max_value=max_value),
timeout=mock.ANY)