def run_movingto_twopercent_to_canonical_test(): transitionlist = [] # Change the vessel_dict and reset all the mock functions in order to have the appropriate info testuser = maindb.get_user(mockutil.testusername) all_donations = maindb.get_donations_by_user(testuser, include_inactive_and_broken=True) node = all_donations[0].node # Create 9 vessels for this node 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)) vessels_dict[mockutil.extra_vessel_name]["userkeys"] = [node_transition_lib.transition_state_keys['movingto_twopercent']] vessels_dict[mockutil.extra_vessel_name]["ownerkey"] = rsa_string_to_publickey(node.owner_pubkey) mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test", vessels_dict) mockutil.mock_backend_set_vessel_owner_key() mockutil.mock_backend_join_vessels() mockutil.mock_backend_set_vessel_user_keylist([mockutil._mock_pubkey_to_string( node_transition_lib.transition_state_keys['canonical'])]) transitionlist.append(("movingto_twopercent", "canonical", node_transition_lib.combine_vessels, node_transition_lib.noop, False)) print "Starting movingto_twopercent to canonical test....." (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_before_completed() assert(mockutil.set_vessel_owner_key_call_count == 0) assert(mockutil.set_vessel_user_keylist_call_count == 1) all_donations = maindb.get_donations_by_user(testuser, include_inactive_and_broken=True) node = all_donations[0].node vessel_list_per_node = maindb.get_vessels_on_node(node) #testing to see if the vessels were deleted after join_vessels was called assert(mockutil.join_vessels_call_count == 9) assert(len(vessel_list_per_node) == 0) assert(node.extra_vessel_name == "extra_vessel_join9")
def run_moving2onepercent_to_canonical_test(): transitionlist = [] # Change the vessel_dict and reset all the mock functions in order to have the appropriate info testuser = maindb.get_user(mockutil.testusername) all_donations = maindb.get_donations_by_user(testuser, include_inactive_and_broken=True) node = all_donations[0].node # Create 9 vessels for this node 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)) vessels_dict[mockutil.extra_vessel_name]["userkeys"] = [node_transition_lib.transition_state_keys['movingto_onepercentmanyevents']] vessels_dict[mockutil.extra_vessel_name]["ownerkey"] = rsa_string_to_publickey(node.owner_pubkey) mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test", vessels_dict) mockutil.mock_backend_set_vessel_owner_key() mockutil.mock_backend_join_vessels() mockutil.mock_backend_set_vessel_user_keylist([mockutil._mock_pubkey_to_string( node_transition_lib.transition_state_keys['canonical'])]) transitionlist.append(("movingto_onepercentmanyevents", "canonical", node_transition_lib.combine_vessels, node_transition_lib.noop, False)) print "Starting canonical to movingtoonepercentmanyevents test....." (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_before_completed() assert(mockutil.set_vessel_owner_key_call_count == 0) assert(mockutil.set_vessel_user_keylist_call_count == 1) all_donations = maindb.get_donations_by_user(testuser, include_inactive_and_broken=True) node = all_donations[0].node vessel_list_per_node = maindb.get_vessels_on_node(node) #testing to see if the vessels were deleted after join_vessels was called assert(mockutil.join_vessels_call_count == 9) assert(len(vessel_list_per_node) == 0) assert(node.extra_vessel_name == "extra_vessel_join9")
def setup_movingto_canonical_to_canonical(): """ <Purpose> Setup everything thats needed to run the movingto_canonical_to_canonical test. <Arguments> None. <Side Effects> None. <Return> None. """ # Do the setup for this test. vessels_dict[mockutil.extra_vessel_name]['userkeys'] = [ node_transition_lib.transition_state_keys['movingto_canonical'] ] mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test", vessels_dict) mockutil.mock_backend_join_vessels() mockutil.mock_backend_set_vessel_user_keylist([ mockutil._mock_pubkey_to_string( node_transition_lib.transition_state_keys['canonical']) ])
def setup_movingto_canonical_to_canonical(): """ <Purpose> Setup everything thats needed to run the movingto_canonical_to_canonical test. <Arguments> None. <Side Effects> None. <Return> None. """ # Do the setup for this test. vessels_dict[mockutil.extra_vessel_name]['userkeys'] = [node_transition_lib.transition_state_keys['movingto_canonical']] mockutil.mock_nodemanager_get_node_info(mockutil.nodeid_key, "10.0test", vessels_dict) mockutil.mock_backend_join_vessels() mockutil.mock_backend_set_vessel_user_keylist([mockutil._mock_pubkey_to_string( node_transition_lib.transition_state_keys['canonical'])])