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)