def test_do_select_all(test_dao):
""" do_select_all() should return a treelib Tree() on success when selecting OpStresss. """
DUT = dtmOpStress(test_dao)
_tree = DUT.do_select_all(parent_id=1)
assert isinstance(_tree, Tree)
assert isinstance(_tree.get_node(1).data, RAMSTKOpStress)
def test_create_opstress_data_model(test_dao):
""" __init__() should return instance of OpStress data model. """
DUT = dtmOpStress(test_dao)
assert isinstance(DUT, dtmOpStress)
assert isinstance(DUT.tree, Tree)
assert isinstance(DUT.dao, DAO)
def test_do_select_non_existent_id(test_dao):
""" do_select() should return None when a non-existent OpStress ID is requested. """
DUT = dtmOpStress(test_dao)
DUT.do_select_all(parent_id=1)
_opstress = DUT.do_select('100')
assert _opstress is None
def test_do_delete_non_existent_id(test_dao):
""" do_delete() should return a non-zero error code when passed a OpStress ID that doesn't exist. """
DUT = dtmOpStress(test_dao)
DUT.do_select_all(parent_id=1)
_error_code, _msg = DUT.do_delete(300)
assert _error_code == 2005
assert _msg == (" RAMSTK ERROR: Attempted to delete non-existent OpStress "
"ID 300.")
def test_do_delete(test_dao):
""" do_delete() should return a zero error code on success. """
DUT = dtmOpStress(test_dao)
DUT.do_select_all(parent_id=1)
_error_code, _msg = DUT.do_delete(DUT.last_id)
assert _error_code == 0
assert _msg == ("RAMSTK SUCCESS: Deleting an item from the RAMSTK Program "
"database.")
def test_do_select(test_dao):
""" do_select() should return an instance of the RAMSTKOpStress data model on success. """
DUT = dtmOpStress(test_dao)
DUT.do_select_all(parent_id=1)
_opstress = DUT.do_select(1)
assert isinstance(_opstress, RAMSTKOpStress)
assert _opstress.stress_id == 1
assert _opstress.description == 'Test Operating Stress'
def test_do_update_all(test_dao):
""" do_update_all() should return a zero error code on success. """
DUT = dtmOpStress(test_dao)
DUT.do_select_all(parent_id=1)
_error_code, _msg = DUT.do_update_all()
assert _error_code == 0
assert _msg == ("RAMSTK SUCCESS: Updating all operating stresses in the "
"damage modeling worksheet.")
def test_do_insert(test_dao):
""" do_insert() should return a zero error code on success. """
DUT = dtmOpStress(test_dao)
DUT.do_select_all(parent_id=1)
_error_code, _msg = DUT.do_insert(load_id=1)
assert _error_code == 0
assert _msg == ("RAMSTK SUCCESS: Adding one or more items to the RAMSTK "
"Program database.")
assert DUT.last_id == 2
def test_do_update(test_dao):
""" do_update() should return a zero error code on success. """
DUT = dtmOpStress(test_dao)
DUT.do_select_all(parent_id=1)
_opstress = DUT.do_select(1)
_opstress.pof_include = 1
_error_code, _msg = DUT.do_update(1)
assert _error_code == 0
assert _msg == ("RAMSTK SUCCESS: Updating the RAMSTK Program database.")
