def test_override(self):
self.comp.create_table()
self.comp.save()
# Test if component seems to be new, so the update mechanism is used
comp3 = Component('name', '/path/to/exe3', 'hash3')
def yes(msg):
return 'y'
def no(msg):
return 'n'
# override yes
comp3._input = yes
comp3.save()
comp_db = Component.get_exactly('name')
self.assertEqual(comp3.c_exe, comp_db.c_exe)
self.assertEqual(comp3.c_exe_hash, comp_db.c_exe_hash)
# override no
comp4 = Component('name', '/path/to/exe4', 'hash4')
comp4._input = no
comp4.save()
comp_db = Component.get_exactly('name')
self.assertNotEqual(comp4.c_exe, comp_db.c_exe)
self.assertNotEqual(comp4.c_exe_hash, comp_db.c_exe_hash)
self.assertEqual(comp3.c_exe, comp_db.c_exe)
self.assertEqual(comp3.c_exe_hash, comp_db.c_exe_hash)
def test_merge(self):
self.comp.create_table()
self.comp.save()
# Test if components seems to be the same so the "merge" mechanism
# is used
comp2 = Component('name', '/path/to/exe', 'hash')
comp2.save()
def setUp(self):
""" Establishes database connection
"""
Database.db_file = "test.db" # DB File for Testcase without config
Log.streamoutput()
# Log.set_verbosity(5)
# Instance objects
self.comp = Component("name", "/path/to/exe", 'hash')
self.connection = Database.db_connection
self.cursor = self.connection.cursor()
