def test_do_select_all_non_existent_id(test_dao): """ do_select_all() should return an empty Tree() when passed a Revision ID that doesn't exist. """ DUT = dtmUsageProfile(test_dao) _tree = DUT.do_select_all(revision_id=100) assert isinstance(_tree, Tree) assert _tree.get_node(0).tag == 'Usage Profiles' assert _tree.get_node(1) is None
def test_do_select(test_dao): """ do_select() should return a Tree() on success. """ DUT = dtmUsageProfile(test_dao) DUT.do_select_all(revision_id=1) _entity = DUT.do_select(1) assert isinstance(_entity, RAMSTKMission) assert _entity.description == 'Test Mission'
def test_do_update(test_dao): """ do_update() should return a zero error code on success. """ DUT = dtmUsageProfile(test_dao) DUT.do_select_all(revision_id=1) _error_code, _msg = DUT.do_update(1) assert _error_code == 0 assert _msg == ("RAMSTK SUCCESS: Updating the RAMSTK Program database.")
def test_do_select_all(test_dao): """ do_select_all() should return an empty Tree() when passed a Revision ID that doesn't exist. """ DUT = dtmUsageProfile(test_dao) _tree = DUT.do_select_all(revision_id=1) assert isinstance(_tree, Tree) assert _tree.get_node(0).tag == 'Usage Profiles' assert isinstance(_tree.get_node(1).data, RAMSTKMission) assert isinstance(_tree.get_node(11).data, RAMSTKMissionPhase) assert isinstance(_tree.get_node(111).data, RAMSTKEnvironment)
def test_do_update_all(test_dao): """ do_update_all() should return a zero error code on success. """ DUT = dtmUsageProfile(test_dao) DUT.do_select_all(revision_id=1) _error_code, _msg = DUT.do_update_all() assert _error_code == 0 assert _msg == ("RAMSTK SUCCESS: Updating all line items in the usage " "profile.")
def test_do_delete_non_existent_node_id(test_dao): """ do_delete() should return a 2005 error code when attempting to remove a non-existant item from the Profile. """ DUT = dtmUsageProfile(test_dao) DUT.do_select_all(revision_id=1) _error_code, _msg = DUT.do_delete(4) assert _error_code == 2005 assert _msg == (" RAMSTK ERROR: Attempted to delete non-existent Usage " "Profile entity with Node ID 4.")
def test_do_delete_environment(test_dao): """ do_delete() should return a zero error code on success when removing an Environment. """ DUT = dtmUsageProfile(test_dao) DUT.do_select_all(revision_id=1) _error_code, _msg = DUT.do_delete(223) assert _error_code == 0 assert _msg == ("RAMSTK SUCCESS: Deleting an item from the RAMSTK Program " "database.")
def test_create_data_model(test_dao): """ __init__() should create a Usage Profile data model. """ DUT = dtmUsageProfile(test_dao) assert isinstance(DUT, dtmUsageProfile) assert isinstance(DUT.dtm_mission, dtmMission) assert isinstance(DUT.dtm_phase, dtmMissionPhase) assert isinstance(DUT.dtm_environment, dtmEnvironment) assert isinstance(DUT.tree, Tree) assert DUT.last_id is None assert DUT.dao == test_dao
def test_do_update_non_existent_node_id(test_dao): """ do_update() should return a 2006 error code when attempting to update a non-existent Node ID. """ DUT = dtmUsageProfile(test_dao) DUT.do_select_all(revision_id=1) _error_code, _msg = DUT.do_update(100) assert _error_code == 2006 assert _msg == ( "RAMSTK ERROR: Attempted to save non-existent Usage Profile " "entity with Node ID 100.")
def test_do_insert_phase(test_dao): """ do_insert() should return a zero error code on success when adding a new Mission Phase. """ DUT = dtmUsageProfile(test_dao) DUT.do_select_all(revision_id=1) _error_code, _msg = DUT.do_insert(entity_id=2, parent_id=1, level='phase') assert _error_code == 0 assert _msg == ( "RAMSTK SUCCESS: Adding one or more items to the RAMSTK Program " "database.") assert isinstance(DUT.do_select(12), RAMSTKMissionPhase)
def test_do_insert_environment(test_dao): """ do_insert() should return a zero error code on success when adding a new Environment. """ DUT = dtmUsageProfile(test_dao) DUT.do_select_all(revision_id=1) _error_code, _msg = DUT.do_insert(entity_id=2, parent_id=11, level='environment') assert _error_code == 0 assert _msg == ( "RAMSTK SUCCESS: Adding one or more items to the RAMSTK Program " "database.") assert isinstance(DUT.tree.get_node(112).data, RAMSTKEnvironment)
def test_do_insert_non_existent_type(test_dao): """ do_insert() should return a 2105 error code when attempting to add something other than a Mission, Phase, or Environment. """ DUT = dtmUsageProfile(test_dao) DUT.do_select_all(revision_id=1) _error_code, _msg = DUT.do_insert(entity_id=1, parent_id=0, level='scadamoosh') assert _error_code == 2105 assert _msg == ( "RAMSTK ERROR: Attempted to add an item to the Usage Profile " "with an undefined indenture level. Level scadamoosh was " "requested. Must be one of mission, phase, or " "environment.")