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.")
