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())