def test_acquire_specific_vessels(self):
# 8 vessels will ultimately be acquired.
calls_results = [True] * 8
mocklib.mock_backend_acquire_vessel(calls_results)
# Create a user who will be doing the acquiring.
user = maindb.create_user("testuser", "password", "*****@*****.**", "affiliation", "1 2", "2 2 2", "3 4")
userport = user.usable_vessel_port
vesselcount = maindb.get_user_free_vessel_credits(user)
# We use userport + 1 to make sure the user isn't being restricted to only
# vessels that have their user port in their port list.
testutil.create_nodes_on_different_subnets(vesselcount + 10, [userport + 1])
vessels = list(maindb._get_queryset_of_all_available_vessels_for_a_port_include_nat_nodes(userport + 1))
# Request the first 4 vessels in the list.
first_vessel_list = interface.acquire_specific_vessels(user, vessels[:4])
# Now request the first 6 vessels in the list. We should only get 2.
second_vessel_list = interface.acquire_specific_vessels(user, vessels[:6])
self.assertEqual(4, len(first_vessel_list))
self.assertEqual(2, len(second_vessel_list))
# Now ask for more vessels than the user has available, regardless of the
# fact that some of the requested vessels aren't available.
requestcount = vesselcount - 6 + 1
func = interface.acquire_specific_vessels
args = (user, vessels[:requestcount])
self.assertRaises(InsufficientUserResourcesError, func, *args)
def test_acquire_specific_vessels(self):
# 8 vessels will ultimately be acquired.
calls_results = [True] * 8
mocklib.mock_backend_acquire_vessel(calls_results)
# Create a user who will be doing the acquiring.
user = maindb.create_user("testuser", "password", "*****@*****.**", "affiliation", "1 2", "2 2 2", "3 4")
userport = user.usable_vessel_port
vesselcount = maindb.get_user_free_vessel_credits(user)
# We use userport + 1 to make sure the user isn't being restricted to only
# vessels that have their user port in their port list.
testutil.create_nodes_on_different_subnets(vesselcount + 10, [userport + 1])
vessels = list(maindb._get_queryset_of_all_available_vessels_for_a_port_include_nat_nodes(userport + 1))
# Request the first 4 vessels in the list.
first_vessel_list = interface.acquire_specific_vessels(user, vessels[:4])
# Now request the first 6 vessels in the list. We should only get 2.
second_vessel_list = interface.acquire_specific_vessels(user, vessels[:6])
self.assertEqual(4, len(first_vessel_list))
self.assertEqual(2, len(second_vessel_list))
# Now ask for more vessels than the user has available, regardless of the
# fact that some of the requested vessels aren't available.
requestcount = vesselcount - 6 + 1
func = interface.acquire_specific_vessels
args = (user, vessels[:requestcount])
self.assertRaises(InsufficientUserResourcesError, func, *args)
# key error.
vesselhandle_list = list(set(vesselhandle_list))
# since we're given a list of vessel 'handles', we need to convert them to a
# list of actual Vessel objects; as release_vessels_of_user expects Vessel objs.
try:
list_of_vessel_objs = interface.get_vessel_list(vesselhandle_list)
except DoesNotExistError, err:
# given handle refers to a non-existant vessel
raise xmlrpclib.Fault(FAULTCODE_INVALIDREQUEST, str(err))
except InvalidRequestError, err:
# A handle is of an invalid format or the list of vessels is empty.
raise xmlrpclib.Fault(FAULTCODE_INVALIDREQUEST, str(err))
try:
acquired_vessels = interface.acquire_specific_vessels(geni_user, list_of_vessel_objs)
except InvalidRequestError, err:
raise xmlrpclib.Fault(FAULTCODE_INVALIDREQUEST, str(err))
except InsufficientUserResourcesError, err:
raise xmlrpclib.Fault(FAULTCODE_NOTENOUGHCREDITS, "You do not have enough " +
"vessel credits to acquire the number of vessels requested.")
# Convert the list of vessel objects into a list of 'info' dictionaries.
return interface.get_vessel_infodict_list(acquired_vessels)
@staticmethod
@log_function_call
def release_resources(auth, vesselhandle_list):
"""
# key error.
vesselhandle_list = list(set(vesselhandle_list))
# since we're given a list of vessel 'handles', we need to convert them to a
# list of actual Vessel objects; as release_vessels_of_user expects Vessel objs.
try:
list_of_vessel_objs = interface.get_vessel_list(vesselhandle_list)
except DoesNotExistError, err:
# given handle refers to a non-existant vessel
raise xmlrpclib.Fault(FAULTCODE_INVALIDREQUEST, str(err))
except InvalidRequestError, err:
# A handle is of an invalid format or the list of vessels is empty.
raise xmlrpclib.Fault(FAULTCODE_INVALIDREQUEST, str(err))
try:
acquired_vessels = interface.acquire_specific_vessels(
geni_user, list_of_vessel_objs)
except InvalidRequestError, err:
raise xmlrpclib.Fault(FAULTCODE_INVALIDREQUEST, str(err))
except InsufficientUserResourcesError, err:
raise xmlrpclib.Fault(
FAULTCODE_NOTENOUGHCREDITS, "You do not have enough " +
"vessel credits to acquire the number of vessels requested.")
# Convert the list of vessel objects into a list of 'info' dictionaries.
return interface.get_vessel_infodict_list(acquired_vessels)
@staticmethod
@log_function_call
def release_resources(auth, vesselhandle_list):
"""
<Purpose>
