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 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 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 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 test_add(self):
""" depends kind of the previous tests, checks only if the database
entry it really done and the file is moved to the storage
"""
insert = Component.add(self.component)
output = Component.get_exactly("example")
self.assertEqual(insert, output)
# check if component is stored.
import os
self.assertTrue(
os.path.isfile(os.path.join(Database.store, 'example.tgz')))
def test_add(self):
""" depends kind of the previous tests, checks only if the database
entry it really done and the file is moved to the storage
"""
insert = Component.add(self.component)
output = Component.get_exactly("example")
self.assertEqual(insert, output)
# check if component is stored.
import os
self.assertTrue(os.path.isfile(os.path.join(Database.store,
'example.tgz')))
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(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_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_read_control(self):
""" checks if the control parsing is done correctly
"""
data = Component._read_control(self.control)
self.assertEqual(data, {"c_id": "example",
"c_exe": "/home/foo/example.exe",
"c_exe_hash": "hash",
},
)
def test_unpack(self):
""" checks if unpacking is done correctly
"""
data = Component._unpack(self.component)
self.assertEqual(data, {"c_id": "example",
"c_exe": "/home/foo/example.exe",
"c_exe_hash": "hash",
},
)
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 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_unpack(self):
""" checks if unpacking is done correctly
"""
data = Component._unpack(self.component)
self.assertEqual(
data,
{
"c_id": "example",
"c_exe": "/home/foo/example.exe",
"c_exe_hash": "hash",
},
)
def test_read_control(self):
""" checks if the control parsing is done correctly
"""
data = Component._read_control(self.control)
self.assertEqual(
data,
{
"c_id": "example",
"c_exe": "/home/foo/example.exe",
"c_exe_hash": "hash",
},
)
def test_remove(self):
""" This Test is a complete test for the removal function. It tests the
removal of a database entry as well as the correlated file in the store
"""
component = Component.add(self.component)
component.remove()
cursor = Database.db_connection.cursor()
cursor.execute("""SELECT * FROM Component
WHERE c_id = 'name';""")
self.assertTrue(cursor.fetchone() is None,
"Removing Entries has failed")
import os
component_file = os.path.join(Database.store, "example.tgz")
self.assertFalse(os.path.isfile(component_file))
def test_remove(self):
""" This Test is a complete test for the removal function. It tests the
removal of a database entry as well as the correlated file in the store
"""
component = Component.add(self.component)
component.remove()
cursor = Database.db_connection.cursor()
cursor.execute("""SELECT * FROM Component
WHERE c_id = 'name';""")
self.assertTrue(cursor.fetchone() is None,
"Removing Entries has failed")
import os
component_file = os.path.join(Database.store, "example.tgz")
self.assertFalse(os.path.isfile(component_file))
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_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_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)
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()
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")
