def test_do_select_non_existent_id(test_dao):
""" do_select() should return None when a non-existent DesignMechanic ID is requested. """
DUT = dtmDesignMechanic(test_dao)
_design_electric = DUT.do_select(100)
assert _design_electric is None
def test_do_select_all(test_dao):
""" do_select_all() should return a Tree() object populated with RAMSTKDesignMechanic instances on success. """
DUT = dtmDesignMechanic(test_dao)
_tree = DUT.do_select_all(hardware_id=2)
assert isinstance(_tree, Tree)
assert isinstance(_tree.get_node(2).data, RAMSTKDesignMechanic)
def test_data_model_create(test_dao):
""" __init__() should return a DesignMechanic model. """
DUT = dtmDesignMechanic(test_dao)
assert isinstance(DUT, dtmDesignMechanic)
assert isinstance(DUT.tree, Tree)
assert isinstance(DUT.dao, DAO)
assert DUT._tag == 'DesignMechanic'
def test_do_delete_non_existent_id(test_dao):
""" do_delete() should return a non-zero error code when passed a DesignMechanic ID that doesn't exist. """
DUT = dtmDesignMechanic(test_dao)
DUT.do_select_all(hardware_id=3)
_error_code, _msg = DUT.do_delete(300)
assert _error_code == 2005
assert _msg == (' RAMSTK ERROR: Attempted to delete non-existent '
'DesignMechanic record ID 300.')
def test_do_delete(test_dao):
""" do_delete() should return a zero error code on success. """
DUT = dtmDesignMechanic(test_dao)
DUT.do_select_all(hardware_id=4)
_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 RAMSTKDesignMechanic data model on success. """
DUT = dtmDesignMechanic(test_dao)
DUT.do_select_all(hardware_id=2)
_design_mechanic = DUT.do_select(2)
assert isinstance(_design_mechanic, RAMSTKDesignMechanic)
assert _design_mechanic.hardware_id == 2
assert _design_mechanic.lubrication_id == 0
def test_do_update_all(test_dao):
""" do_update_all() should return a zero error code on success. """
DUT = dtmDesignMechanic(test_dao)
DUT.do_select_all(hardware_id=3)
_error_code, _msg = DUT.do_update_all()
assert _error_code == 0
assert _msg == ("RAMSTK SUCCESS: Updating all records in the mechanical "
"design table.")
def test_do_insert(test_dao):
""" do_insert() should return False on success when inserting a DesignMechanic record. """
DUT = dtmDesignMechanic(test_dao)
DUT.do_select_all(hardware_id=3)
_error_code, _msg = DUT.do_insert(hardware_id=90)
assert _error_code == 0
assert _msg == (
'RAMSTK SUCCESS: Adding one or more items to the RAMSTK Program '
'database.')
def test_do_update(test_dao):
""" do_update() should return a zero error code on success. """
DUT = dtmDesignMechanic(test_dao)
DUT.do_select_all(hardware_id=3)
_design_electric = DUT.do_select(3)
_design_electric.resistance = 0.9832
_error_code, _msg = DUT.do_update(3)
assert _error_code == 0
assert _msg == ('RAMSTK SUCCESS: Updating the RAMSTK Program database.')