def getFacility(name): """Return a given facility by name or unique ID. @type name: str @param name: a facility name or unique ID @rtype: Facility @return: a facility object """ return Cuebot.getStub('facility').Get( facility_pb2.FacilityGetRequest(name=name), timeout=Cuebot.Timeout).facility
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)