def testGetFacility(self, getStubMock):
stubMock = mock.Mock()
stubMock.Get.return_value = facility_pb2.FacilityGetResponse(
facility=facility_pb2.Facility(name=TEST_FACILITY_NAME))
getStubMock.return_value = stubMock
facility = opencue.api.getFacility(TEST_FACILITY_NAME)
stubMock.Get.assert_called_with(
facility_pb2.FacilityGetRequest(name=TEST_FACILITY_NAME), timeout=mock.ANY)
self.assertEqual(TEST_FACILITY_NAME, facility.name)
def testCreateFacility(self, getStubMock):
stubMock = mock.Mock()
stubMock.Create.return_value = facility_pb2.FacilityCreateResponse(
facility=facility_pb2.Facility(name=TEST_FACILITY_NAME))
getStubMock.return_value = stubMock
newFacility = opencue.api.createFacility(TEST_FACILITY_NAME)
stubMock.Create.assert_called_with(
facility_pb2.FacilityCreateRequest(name=TEST_FACILITY_NAME), timeout=mock.ANY)
self.assertEqual(TEST_FACILITY_NAME, newFacility.name)
def testRenameFacility(self, getStubMock):
facility = facility_pb2.Facility(name=TEST_FACILITY_NAME)
newName = 'new-name'
stubMock = mock.Mock()
stubMock.Rename.return_value = facility_pb2.FacilityRenameResponse()
getStubMock.return_value = stubMock
opencue.api.renameFacility(facility, newName)
stubMock.Rename.assert_called_with(
facility_pb2.FacilityRenameRequest(facility=facility, new_name=newName),
timeout=mock.ANY)
def testCreateAlloc(self, getStubMock):
facility = facility_pb2.Facility(name=TEST_FACILITY_NAME)
stubMock = mock.Mock()
stubMock.Create.return_value = facility_pb2.AllocCreateResponse(
allocation=facility_pb2.Allocation(name=TEST_ALLOC_NAME))
getStubMock.return_value = stubMock
alloc = opencue.api.createAllocation(TEST_ALLOC_NAME, TEST_TAG, facility)
stubMock.Create.assert_called_with(
facility_pb2.AllocCreateRequest(name=TEST_ALLOC_NAME, tag=TEST_TAG, facility=facility),
timeout=mock.ANY)
self.assertEqual(TEST_ALLOC_NAME, alloc.name())
