class TestComponent(unittest.TestCase):
""" Testing of module.py
"""
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()
def tearDown(self):
""" Closes database connection and removes database from disk
"""
Database.db_connection.close()
os.remove(Database.db_file)
def test_create_table(self):
self.comp.create_table()
self.assertIsNotNone(self.connection,
"Database isn't correctly initialized")
self.cursor.execute("""SELECT name FROM sqlite_master
WHERE name = 'Component';""")
table_exists = False
for i in self.cursor.fetchone():
if 'Component' in i:
table_exists = True
self.assertTrue(table_exists)
def test_drop_table(self):
self.comp.create_table()
try:
self.comp.drop_table()
except sqlite3.OperationalError:
self.assertFalse(True, "Error while table drop")
self.cursor.execute("""SELECT name FROM sqlite_master
WHERE name = 'Component';""")
self.assertIsNone(self.cursor.fetchone())
def test_getitem(self):
""" Tests if access to instance objects with names like c_ are
possible. Errors or Fails need to be fixed in class Database.
"""
self.comp.create_table()
self.assertEqual(self.comp["c_id"], "name")
self.assertEqual(self.comp["c_exe"], "/path/to/exe")
try:
self.comp["db_file"]
self.assertTrue(False, "Got access to wrong variable")
except AttributeError:
pass
def test_get_attributes(self):
""" Tests if all instance objects are found by get_attributes(). Errors
or Fails need to be fixed in class Database.
"""
self.comp.create_table()
attrs = self.comp.get_attributes()
self.assertTrue(attrs, msg="Got no attributes")
self.assertTrue("c_id" in attrs,
msg="Caught the following attributes: %s" % attrs)
self.assertTrue("c_exe" in attrs,
msg="Caught the following attributes: %s" % attrs)
self.assertFalse("__doc__" in attrs,
msg="Caught __doc__ attribute which is wrong")
def test_save(self):
""" Checks if data can be made persistent in the database
"""
with self.assertRaises(NoSuchTable):
self.comp.save()
self.assertEqual(self.comp.c_pk, -1, "no temporary primary key?")
self.comp.create_table()
self.comp.save()
self.assertEqual(self.comp.c_pk, 1,
"not the correct primary key for first element")
# connection restart
self.connection.close()
Database()
# test if data still present
cursor = self.comp.db_connection.cursor()
cursor.execute("SELECT * FROM Component;")
res = cursor.fetchone()
self.assertIsNotNone(res, "Could not read test data from db.")
res = tuple(res)
self.assertEqual(res[5], "name", "Got unexpected data")
self.assertEqual(res[4], "hash", "Got unexpected data")
self.assertEqual(res[3], "/path/to/exe", "Got unexpected data")
def test_update(self):
""" Checks if data can be changed
"""
self.comp.create_table()
self.comp.save()
self.comp.c_exe = "updatedexe"
self.comp.save()
# raw database check
self.cursor.execute("SELECT * FROM Component;")
result = self.cursor.fetchall()
self.assertEqual(len(result), 1, "too much or too less objects")
self.cursor.execute("SELECT * FROM Component;")
res = self.cursor.fetchone()
res = tuple(res)
self.assertEqual(res[3], "updatedexe",
"Seems not to be updated correctly")
# Object oriented check
comp_db = Component.get_exactly(self.comp.c_id)
self.assertEqual(self.comp.c_exe, comp_db.c_exe)
def test_remove(self):
self.comp.create_table()
self.comp.save()
try:
self.comp.remove()
except OSError:
pass # The error that is risen because of the missing component
# file. This test will be repeated in class TestComponentAdd
self.cursor.execute("""SELECT * FROM Component
WHERE c_id = 'name';""")
self.assertTrue(self.cursor.fetchone() is None,
"Removing Entries has failed")
def test_conform(self):
""" Test for sqlite3 representation
"""
self.comp.create_table()
repr = self.comp.__conform__(sqlite3.PrepareProtocol)
self.assertEqual(repr,
"c_id=name;c_exe_hash=hash;c_exe=/path/to/exe",
msg="Error in generation of representation")
def test_convert(self):
""" Tests if a __conform__ representation could be successfully
rebuild.
"""
self.comp.create_table()
repr = (-1, "c_id=name;c_exe_hash=hash;c_exe=/path/to/exe",)
comp = Component.convert(repr)
self.assertEqual(comp, self.comp)
def test_get(self):
from datetime import datetime
from datetime import timedelta
self.comp.create_table()
self.comp.save()
# Test get all
comps = Component.get()
self.assertIsNotNone(comps, "Could not retrieve Component")
self.assertEqual(comps[0], self.comp, "Could not deserialize data")
time_since = datetime.today() - timedelta(minutes=10)
comps = Component.get(time_since)
Log.debug("test_get(): comps: " + str(comps))
self.assertEqual(comps[0], self.comp, "Could not deserialize data")
def test_get_exactly(self):
self.comp.create_table()
self.comp.save()
comp = Component.get_exactly(self.comp.c_id)
self.assertEqual(comp, self.comp, "Could not deserialize data")
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)
@unittest.skip('Test and function not implemented yet.')
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()
class TestComponent(unittest.TestCase):
""" Testing of module.py
"""
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()
def tearDown(self):
""" Closes database connection and removes database from disk
"""
Database.db_connection.close()
os.remove(Database.db_file)
def test_create_table(self):
self.comp.create_table()
self.assertIsNotNone(self.connection,
"Database isn't correctly initialized")
self.cursor.execute("""SELECT name FROM sqlite_master
WHERE name = 'Component';""")
table_exists = False
for i in self.cursor.fetchone():
if 'Component' in i:
table_exists = True
self.assertTrue(table_exists)
def test_drop_table(self):
self.comp.create_table()
try:
self.comp.drop_table()
except sqlite3.OperationalError:
self.assertFalse(True, "Error while table drop")
self.cursor.execute("""SELECT name FROM sqlite_master
WHERE name = 'Component';""")
self.assertIsNone(self.cursor.fetchone())
def test_getitem(self):
""" Tests if access to instance objects with names like c_ are
possible. Errors or Fails need to be fixed in class Database.
"""
self.comp.create_table()
self.assertEqual(self.comp["c_id"], "name")
self.assertEqual(self.comp["c_exe"], "/path/to/exe")
try:
self.comp["db_file"]
self.assertTrue(False, "Got access to wrong variable")
except AttributeError:
pass
def test_get_attributes(self):
""" Tests if all instance objects are found by get_attributes(). Errors
or Fails need to be fixed in class Database.
"""
self.comp.create_table()
attrs = self.comp.get_attributes()
self.assertTrue(attrs, msg="Got no attributes")
self.assertTrue("c_id" in attrs,
msg="Caught the following attributes: %s" % attrs)
self.assertTrue("c_exe" in attrs,
msg="Caught the following attributes: %s" % attrs)
self.assertFalse("__doc__" in attrs,
msg="Caught __doc__ attribute which is wrong")
def test_save(self):
""" Checks if data can be made persistent in the database
"""
with self.assertRaises(NoSuchTable):
self.comp.save()
self.assertEqual(self.comp.c_pk, -1, "no temporary primary key?")
self.comp.create_table()
self.comp.save()
self.assertEqual(self.comp.c_pk, 1,
"not the correct primary key for first element")
# connection restart
self.connection.close()
Database()
# test if data still present
cursor = self.comp.db_connection.cursor()
cursor.execute("SELECT * FROM Component;")
res = cursor.fetchone()
self.assertIsNotNone(res, "Could not read test data from db.")
res = tuple(res)
self.assertEqual(res[5], "name", "Got unexpected data")
self.assertEqual(res[4], "hash", "Got unexpected data")
self.assertEqual(res[3], "/path/to/exe", "Got unexpected data")
def test_update(self):
""" Checks if data can be changed
"""
self.comp.create_table()
self.comp.save()
self.comp.c_exe = "updatedexe"
self.comp.save()
# raw database check
self.cursor.execute("SELECT * FROM Component;")
result = self.cursor.fetchall()
self.assertEqual(len(result), 1, "too much or too less objects")
self.cursor.execute("SELECT * FROM Component;")
res = self.cursor.fetchone()
res = tuple(res)
self.assertEqual(res[3], "updatedexe",
"Seems not to be updated correctly")
# Object oriented check
comp_db = Component.get_exactly(self.comp.c_id)
self.assertEqual(self.comp.c_exe, comp_db.c_exe)
def test_remove(self):
self.comp.create_table()
self.comp.save()
try:
self.comp.remove()
except OSError:
pass # The error that is risen because of the missing component
# file. This test will be repeated in class TestComponentAdd
self.cursor.execute("""SELECT * FROM Component
WHERE c_id = 'name';""")
self.assertTrue(self.cursor.fetchone() is None,
"Removing Entries has failed")
def test_conform(self):
""" Test for sqlite3 representation
"""
self.comp.create_table()
repr = self.comp.__conform__(sqlite3.PrepareProtocol)
self.assertEqual(repr,
"c_id=name;c_exe_hash=hash;c_exe=/path/to/exe",
msg="Error in generation of representation")
def test_convert(self):
""" Tests if a __conform__ representation could be successfully
rebuild.
"""
self.comp.create_table()
repr = (
-1,
"c_id=name;c_exe_hash=hash;c_exe=/path/to/exe",
)
comp = Component.convert(repr)
self.assertEqual(comp, self.comp)
def test_get(self):
from datetime import datetime
from datetime import timedelta
self.comp.create_table()
self.comp.save()
# Test get all
comps = Component.get()
self.assertIsNotNone(comps, "Could not retrieve Component")
self.assertEqual(comps[0], self.comp, "Could not deserialize data")
time_since = datetime.today() - timedelta(minutes=10)
comps = Component.get(time_since)
Log.debug("test_get(): comps: " + str(comps))
self.assertEqual(comps[0], self.comp, "Could not deserialize data")
def test_get_exactly(self):
self.comp.create_table()
self.comp.save()
comp = Component.get_exactly(self.comp.c_id)
self.assertEqual(comp, self.comp, "Could not deserialize data")
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)
@unittest.skip('Test and function not implemented yet.')
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()
