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())
def createAllocation(name, tag, facility): """Creates and returns an allocation. The host tag will be the lowercase of the allocation name. @type name: str @param name: the name of the allocation @type tag: str @param tag: the tag for the allocation @rtype: Allocation @return: the newly created Allocation object""" return Allocation( Cuebot.getStub('allocation').Create(facility_pb2.AllocCreateRequest( name=name, tag=tag, facility=facility), timeout=Cuebot.Timeout).allocation)