def setup_test():
"""
<Purpose>
Prepare everything in order to run the tests.
<Arguments>
None
<Exceptions>
None
<Side Efects>
None
<Return>
None
"""
testlib.setup_test_db()
# Make sure all the variables are its default values
mockutil.mockutil_cleanup_variables()
# Create a user who has the donation key.
user_object = maindb.create_user(
mockutil.testusername,
"password",
"*****@*****.**",
"affiliation",
"10 11",
"2 2 2",
mockutil.donor_key_str,
)
# Create a database entry for the node
node_object = maindb.create_node(
mockutil.nodeid_key_str,
mockutil.node_ip,
mockutil.node_port,
"10.0test",
False,
mockutil.per_node_key_str,
mockutil.extra_vessel_name,
)
# Create a donation for user
maindb.create_donation(node_object, user_object, "Making a donation")
# Setup all the mock functions
mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test", vessels_dict)
mockutil.mock_lockserver_calls()
mockutil.mock_backend_generate_key([mockutil.per_node_key_str])
mockutil.mock_nodemanager_get_vesselresources()
mockutil.mock_transitionlib_do_advertise_lookup([mockutil.node_address])
mockutil.mock_backend_set_vessel_owner_key()
mockutil.mock_backend_split_vessel()
mockutil.mock_backend_set_vessel_user_keylist(
[mockutil._mock_pubkey_to_string(node_transition_lib.movingtoonepercentmanyeventspublickey)]
)
def run_database_stituation_three_test():
"""
<Purpose>
The purpose of this test is to test the case where a record
for the node has been registered in the database and a
record for the donation has been registered, but the
owner key on the node has not been changed.
<Arguments>
None
<Exceptions>
None
<Side Effects>
None
<Return>
None
"""
# Create a database entry for the node
node_object = maindb.create_node(
mockutil.nodeid_key_str,
mockutil.node_ip,
mockutil.node_port,
"10.0test",
False,
mockutil.per_node_key_str,
mockutil.extra_vessel_name,
)
user_object = maindb.get_user(mockutil.testusername)
# Create a donation for user
maindb.create_donation(node_object, user_object, "Making a donation")
transitionlist = []
transitionlist.append(
(
("donation_state", node_transition_lib.acceptdonationpublickey),
("canonical_state", node_transition_lib.canonicalpublickey),
node_transition_lib.noop,
node_transition_lib.noop,
)
)
print "Running test where database has record of node, has record of donation and node has donor key"
(success_count, failure_count) = node_transition_lib.do_one_processnode_run(transitionlist, "startstatename", 1)[0]
assert success_count == 1
assert failure_count == 0
assert_database_info()
assert mockutil.set_vessel_owner_key_call_count == 1
assert mockutil.set_vessel_user_keylist_call_count == 1
def setup_test():
"""
<Purpose>
Prepare everything in order to run the tests.
<Arguments>
None
<Exceptions>
None
<Side Efects>
None
<Return>
None
"""
testlib.setup_test_db()
# Create a user who has the donation key.
user_object = maindb.create_user(
mockutil.testusername,
"password",
"*****@*****.**",
"affiliation",
"10 11",
"2 2 2",
mockutil.donor_key_str,
)
# Create a database entry for the node
node_object = maindb.create_node(
mockutil.nodeid_key_str,
mockutil.node_ip,
mockutil.node_port,
"10.0test",
True,
mockutil.per_node_key_str,
mockutil.extra_vessel_name,
)
# Create a donation for user
maindb.create_donation(node_object, user_object, "Making a donation")
# Setup all the mock functions
mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test", vessels_dict)
mockutil.mock_lockserver_calls()
mockutil.mock_backend_generate_key([mockutil.per_node_key_str])
mockutil.mock_nodemanager_get_vesselresources()
mockutil.mock_transitionlib_do_advertise_lookup([mockutil.node_address])
mockutil.mock_backend_set_vessel_owner_key()
mockutil.mock_backend_split_vessel()
# set_vessel_user_keylist_call_count won't be called because the
# node_transition_lib will see that it's a state change to the same state.
mockutil.mock_backend_set_vessel_user_keylist(None)
def run_database_stituation_three_test():
"""
<Purpose>
The purpose of this test is to test the case where a record
for the node has been registered in the database and a
record for the donation has been registered, but the
owner key on the node has not been changed.
<Arguments>
None
<Exceptions>
None
<Side Effects>
None
<Return>
None
"""
# Create a database entry for the node
node_object = maindb.create_node(mockutil.nodeid_key_str, mockutil.node_ip,
mockutil.node_port, "10.0test", False,
mockutil.per_node_key_str,
mockutil.extra_vessel_name)
user_object = maindb.get_user(mockutil.testusername)
# Create a donation for user
maindb.create_donation(node_object, user_object, "Making a donation")
transitionlist = []
transitionlist.append(
(("donation_state", node_transition_lib.acceptdonationpublickey),
("canonical_state", node_transition_lib.canonicalpublickey),
node_transition_lib.noop, node_transition_lib.noop))
print "Running test where database has record of node, has record of donation and node has donor key"
(success_count,
failure_count) = node_transition_lib.do_one_processnode_run(
transitionlist, "startstatename", 1)[0]
assert (success_count == 1)
assert (failure_count == 0)
assert_database_info()
assert (mockutil.set_vessel_owner_key_call_count == 1)
assert (mockutil.set_vessel_user_keylist_call_count == 1)
def setup_test():
"""
<Purpose>
Prepare everything in order to run the tests.
<Arguments>
None
<Exceptions>
None
<Side Efects>
None
<Return>
None
"""
testlib.setup_test_db()
# Make sure all the variables are its default values
mockutil.mockutil_cleanup_variables()
# Create a user who has the donation key.
user_object = maindb.create_user(mockutil.testusername, "password",
"*****@*****.**", "affiliation",
"10 11", "2 2 2", mockutil.donor_key_str)
# Create a database entry for the node
node_object = maindb.create_node(mockutil.nodeid_key_str, mockutil.node_ip,
mockutil.node_port, "10.0test", False,
mockutil.per_node_key_str,
mockutil.extra_vessel_name)
# Create a donation for user
maindb.create_donation(node_object, user_object, "Making a donation")
# Setup all the mock functions
mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test",
vessels_dict)
mockutil.mock_lockserver_calls()
mockutil.mock_backend_generate_key([mockutil.per_node_key_str])
mockutil.mock_nodemanager_get_vesselresources()
mockutil.mock_transitionlib_do_advertise_lookup([mockutil.node_address])
mockutil.mock_backend_set_vessel_owner_key()
mockutil.mock_backend_split_vessel()
mockutil.mock_backend_set_vessel_user_keylist([
mockutil._mock_pubkey_to_string(
node_transition_lib.
transition_state_keys['movingto_onepercentmanyevents'])
])
def setup_test():
"""
<Purpose>
Prepare everything in order to run the tests.
<Arguments>
None
<Exceptions>
None
<Side Efects>
None
<Return>
None
"""
testlib.setup_test_db()
# Create a user who has the donation key.
user_object = maindb.create_user(mockutil.testusername, "password",
"*****@*****.**", "affiliation",
"10 11", "2 2 2", mockutil.donor_key_str)
# Create a database entry for the node
node_object = maindb.create_node(mockutil.nodeid_key_str, mockutil.node_ip,
mockutil.node_port, "10.0test", True,
mockutil.per_node_key_str,
mockutil.extra_vessel_name)
# Create a donation for user
maindb.create_donation(node_object, user_object, "Making a donation")
# Setup all the mock functions
mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test",
vessels_dict)
mockutil.mock_lockserver_calls()
mockutil.mock_backend_generate_key([mockutil.per_node_key_str])
mockutil.mock_nodemanager_get_vesselresources()
mockutil.mock_transitionlib_do_advertise_lookup([mockutil.node_address])
mockutil.mock_backend_set_vessel_owner_key()
mockutil.mock_backend_split_vessel()
# set_vessel_user_keylist_call_count won't be called because the
# node_transition_lib will see that it's a state change to the same state.
mockutil.mock_backend_set_vessel_user_keylist(None)
def create_new_node_object(node_string, node_info, donated_vesselname):
"""
<Purpose>
Create a new node entry in the database with a newly generated
per node key
<Arguments>
node_string - the address of the node
node_info - information about the node
vesselname - the name of the extra vessel
<Exceptions>
None
<Side Effects>
None
<Return>
Returns a database node object
"""
nodeID = _do_rsa_publickey_to_string(node_info['nodekey'])
(ip_or_nat_string, port_num) = split_node_string(node_string)
# Generate a new set of keys for the node. The pubkey gets passed
# down to us while the private key is stored in a database elsewhere.
log("Generating new keys for node owner_key....")
new_node_owner_pubkey = backend.generate_key(ip_or_nat_string+". "+nodeID)
log("Generated publickey for node "+node_string+" : "+str(new_node_owner_pubkey))
try:
# Attempt to add new node to db.
database_nodeobject = maindb.create_node(nodeID, ip_or_nat_string, port_num,
node_info['version'], False,
new_node_owner_pubkey,
donated_vesselname)
log("Added node to the database with nodeID: " + nodeID)
except:
raise DatabaseError("Failed to create node and add to database. " + traceback.format_exc())
return database_nodeobject
def run_database_situation_two_test():
"""
<Purpose>
The purpose of this test is to test the case where a record
for the node has been registered in the database, but the
owner key on the node has not been changed and no donation
record has been created.
<Arguments>
None
<Exceptions>
None
<Side Effects>
None
<Return>
None
"""
# Create a database entry for the node
node_object = maindb.create_node(mockutil.nodeid_key_str, mockutil.node_ip, mockutil.node_port, "10.0test",
False, mockutil.per_node_key_str, mockutil.extra_vessel_name)
transitionlist = []
transitionlist.append(("acceptdonation", "canonical",
node_transition_lib.noop,
node_transition_lib.noop, False))
print "Running test where database has record of node, no record of donation and node has donor key"
(success_count, failure_count) = node_transition_lib.do_one_processnode_run(transitionlist, "startstatename", 1)[0]
assert(success_count == 1)
assert(failure_count == 0)
assert_database_info()
assert(mockutil.set_vessel_owner_key_call_count == 1)
assert(mockutil.set_vessel_user_keylist_call_count == 1)
def create_node_and_vessels_with_one_port_each(ip, portlist, is_active=True):
global next_nodeid_number
next_nodeid_number += 1
nodeid = "node" + str(next_nodeid_number)
port = 1234
version = "10.0test"
owner_pubkey = "1 2"
extra_vessel_name = "v1"
node = maindb.create_node(nodeid, ip, port, version, is_active, owner_pubkey, extra_vessel_name)
single_vessel_number = 2
for vesselport in portlist:
single_vessel_name = "v" + str(single_vessel_number)
single_vessel_number += 1
vessel = maindb.create_vessel(node, single_vessel_name)
maindb.set_vessel_ports(vessel, [vesselport])
return node
def create_node_and_vessels_with_one_port_each(ip, portlist):
global next_nodeid_number
next_nodeid_number += 1
nodeid = "node" + str(next_nodeid_number)
port = 1234
version = "10.0test"
is_active = True
owner_pubkey = "1 2"
extra_vessel_name = "v1"
node = maindb.create_node(nodeid, ip, port, version, is_active, owner_pubkey, extra_vessel_name)
single_vessel_number = 2
for vesselport in portlist:
single_vessel_name = "v" + str(single_vessel_number)
single_vessel_number += 1
vessel = maindb.create_vessel(node, single_vessel_name)
maindb.set_vessel_ports(vessel, [vesselport])
return node
def setup_general():
"""
<Purpose>
Prepare all the general stuff in order to run the tests.
<Arguments>
None
<Exceptions>
None
<Side Efects>
None
<Return>
None
"""
testlib.setup_test_db()
# Make sure all the variables are its default values
mockutil.mockutil_cleanup_variables()
# Create a user who has the donation key.
user_object = maindb.create_user(mockutil.testusername, "password",
"*****@*****.**", "affiliation",
"10 11", "2 2 2", mockutil.donor_key_str)
# Create a database entry for the node
node_object = maindb.create_node(mockutil.nodeid_key_str, mockutil.node_ip,
mockutil.node_port, "10.0test", True,
mockutil.per_node_key_str,
mockutil.extra_vessel_name)
# Create a donation for user
maindb.create_donation(node_object, user_object, "Making a donation")
testuser = maindb.get_user(mockutil.testusername)
# Retrieve the info about the donation we just made.
all_donations = maindb.get_donations_by_user(
testuser, include_inactive_and_broken=True)
node = all_donations[0].node
global vessels_dict
vessels_dict = vessels_dict_basic.copy()
# Create 9 different vessels, assuming that the original vessel got split into 9.
for i in range(9):
vessels_dict["vessel" + str(i)] = {}
vessels_dict["vessel" + str(i)]["userkeys"] = []
vessels_dict["vessel" + str(i)]["ownerkey"] = rsa_string_to_publickey(
node.owner_pubkey)
vessels_dict["vessel" + str(i)]["ownerinfo"] = ""
vessels_dict["vessel" + str(i)]["status"] = ""
vessels_dict["vessel" + str(i)]["advertise"] = True
maindb.create_vessel(node, "vessel" + str(i))
# Setup all the mock functions
mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test",
vessels_dict)
mockutil.mock_lockserver_calls()
mockutil.mock_backend_generate_key([mockutil.per_node_key_str])
mockutil.mock_nodemanager_get_vesselresources()
mockutil.mock_transitionlib_do_advertise_lookup([mockutil.node_address])
mockutil.mock_backend_set_vessel_owner_key()
def run_multiple_donation_test():
"""
<Purpose>
The purpose of this test is to see how the transtion script would
react if we provide a dictionary where a vessel is in the acceptdonation
state but the node has already in the canonical state. This is possible
if the node had two vessel in the acceptdonation state to start with.
This is essentially if the node has two donations. Currently we don't handle
multiple donations from the second node, so when we run the test, the success
count should be 0 and the failure count should be 1. Also the name of the extra
vessel should not change.
<Arguments>
None
<Exceptions>
None
<Side Effects>
None
<Return>
None
"""
# Have the extra vessel in canonical form
vessels_dict[mockutil.extra_vessel_name]["userkeys"] = [
node_transition_lib.canonicalpublickey
]
# Have a second vessel in acceptdonation state. Essentially a second donation.
vessels_dict["second_donation"] = {
"userkeys": [node_transition_lib.acceptdonationpublickey],
"ownerkey": mockutil.donor_key,
"ownerinfo": "",
"status": "",
"advertise": True
}
# Create a database entry for the node
node_object = maindb.create_node(mockutil.nodeid_key_str, mockutil.node_ip,
mockutil.node_port, "10.0test", False,
mockutil.per_node_key_str,
mockutil.extra_vessel_name)
user_object = maindb.get_user(mockutil.testusername)
# Create a donation for user
maindb.create_donation(node_object, user_object, "Making a donation")
# Register the new vesseldict so get_info returns this new modified vesseldict
mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test",
vessels_dict)
transitionlist = []
transitionlist.append(
(("donation_state", node_transition_lib.acceptdonationpublickey),
("canonical_state", node_transition_lib.canonicalpublickey),
node_transition_lib.noop, node_transition_lib.noop))
print "Running test where database has record of node, no record of donation and node has donor key"
(success_count,
failure_count) = node_transition_lib.do_one_processnode_run(
transitionlist, "startstatename", 1)[0]
assert (success_count == 0)
assert (failure_count == 1)
assert_database_info()
assert (mockutil.set_vessel_owner_key_call_count == 0)
assert (mockutil.set_vessel_user_keylist_call_count == 0)
def setup_general():
"""
<Purpose>
Prepare all the general stuff in order to run the tests.
<Arguments>
None
<Exceptions>
None
<Side Efects>
None
<Return>
None
"""
testlib.setup_test_db()
# Make sure all the variables are its default values
mockutil.mockutil_cleanup_variables()
# Create a user who has the donation key.
user_object = maindb.create_user(mockutil.testusername, "password", "*****@*****.**", "affiliation",
"10 11", "2 2 2", mockutil.donor_key_str)
# Create a database entry for the node
node_object = maindb.create_node(mockutil.nodeid_key_str, mockutil.node_ip, mockutil.node_port, "10.0test",
True, mockutil.per_node_key_str, mockutil.extra_vessel_name)
# Create a donation for user
maindb.create_donation(node_object, user_object, "Making a donation")
testuser = maindb.get_user(mockutil.testusername)
# Retrieve the info about the donation we just made.
all_donations = maindb.get_donations_by_user(testuser, include_inactive_and_broken=True)
node = all_donations[0].node
global vessels_dict
vessels_dict = vessels_dict_basic.copy()
# Create 9 different vessels, assuming that the original vessel got split into 9.
for i in range(9):
vessels_dict["vessel"+str(i)]={}
vessels_dict["vessel"+str(i)]["userkeys"] = []
vessels_dict["vessel"+str(i)]["ownerkey"] = rsa_string_to_publickey(node.owner_pubkey)
vessels_dict["vessel"+str(i)]["ownerinfo"] = ""
vessels_dict["vessel"+str(i)]["status"] = ""
vessels_dict["vessel"+str(i)]["advertise"] = True
maindb.create_vessel(node, "vessel"+str(i))
# Setup all the mock functions
mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test", vessels_dict)
mockutil.mock_lockserver_calls()
mockutil.mock_backend_generate_key([mockutil.per_node_key_str])
mockutil.mock_nodemanager_get_vesselresources()
mockutil.mock_transitionlib_do_advertise_lookup([mockutil.node_address])
mockutil.mock_backend_set_vessel_owner_key()
def run_multiple_donation_test():
"""
<Purpose>
The purpose of this test is to see how the transtion script would
react if we provide a dictionary where a vessel is in the acceptdonation
state but the node has already in the canonical state. This is possible
if the node had two vessel in the acceptdonation state to start with.
This is essentially if the node has two donations. Currently we don't handle
multiple donations from the second node, so when we run the test, the success
count should be 0 and the failure count should be 1. Also the name of the extra
vessel should not change.
<Arguments>
None
<Exceptions>
None
<Side Effects>
None
<Return>
None
"""
# Have the extra vessel in canonical form
vessels_dict[mockutil.extra_vessel_name]["userkeys"] = [node_transition_lib.transition_state_keys['canonical']]
# Have a second vessel in acceptdonation state. Essentially a second donation.
vessels_dict["second_donation"] = {"userkeys" : [node_transition_lib.transition_state_keys['acceptdonation']],
"ownerkey" : mockutil.donor_key,
"ownerinfo" : "",
"status" : "",
"advertise" : True}
# Create a database entry for the node
node_object = maindb.create_node(mockutil.nodeid_key_str, mockutil.node_ip, mockutil.node_port, "10.0test",
False, mockutil.per_node_key_str, mockutil.extra_vessel_name)
user_object = maindb.get_user(mockutil.testusername)
# Create a donation for user
maindb.create_donation(node_object, user_object, "Making a donation")
# Register the new vesseldict so get_info returns this new modified vesseldict
mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test", vessels_dict)
transitionlist = []
transitionlist.append(("acceptdonation", "canonical",
node_transition_lib.noop,
node_transition_lib.noop, False))
print "Running test where database has record of node, no record of donation and node has donor key"
(success_count, failure_count) = node_transition_lib.do_one_processnode_run(transitionlist, "startstatename", 1)[0]
assert(success_count == 0)
assert(failure_count == 1)
assert_database_info()
assert(mockutil.set_vessel_owner_key_call_count == 0)
assert(mockutil.set_vessel_user_keylist_call_count == 0)
# Create GeniUser records user_list = [] for i in range(10): username = '******' + str(i) user = maindb.create_user(username, 'mypassword', 'myemail', 'myaff', 'mypubkey', 'myprivkey', 'mydonorpubkey') user.save() user_list.append(user) # Create Node records node_list = [] for i in range(100): node_identifier = 'node' + str(i) extra_vessel_name = 'v2' ip = '127.0.' + str(i) + '.0' node = maindb.create_node(node_identifier, ip, 1234, '0.1a', True, 'the owner pubkey', extra_vessel_name) node_list.append(node) # Create Donation records donation_list = [] for i in range(100): node = node_list[i] donor = user_list[i % len(user_list)] donation = maindb.create_donation(node, donor, 'some resource description text') donation_list.append(donation) # Create Vessel records vessel_list = [] for i in range(100):
# Create GeniUser records
user_list = []
for i in range(10):
username = '******' + str(i)
user = maindb.create_user(username, 'mypassword', 'myemail', 'myaff',
'mypubkey', 'myprivkey', 'mydonorpubkey')
user.save()
user_list.append(user)
# Create Node records
node_list = []
for i in range(100):
node_identifier = 'node' + str(i)
extra_vessel_name = 'v2'
ip = '127.0.' + str(i) + '.0'
node = maindb.create_node(node_identifier, ip, 1234, '0.1a', True,
'the owner pubkey', extra_vessel_name)
node_list.append(node)
# Create Donation records
donation_list = []
for i in range(100):
node = node_list[i]
donor = user_list[i % len(user_list)]
donation = maindb.create_donation(node, donor,
'some resource description text')
donation_list.append(donation)
# Create Vessel records
vessel_list = []
for i in range(100):
node = node_list[i]