def createFacility(name):
"""Create 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').Create(
facility_pb2.FacilityCreateRequest(name=name),
timeout=Cuebot.Timeout).facility
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 createFacility(name):
return Cuebot.getStub('facility').Create(
facility_pb2.FacilityCreateRequest(name=name),
timeout=Cuebot.Timeout).facility
