def test_get(self):
self.web.save()
webs = Web.get()
self.assertEqual(webs[0], self.web, "Unable to get all data")
import datetime
d = datetime.datetime.now() - datetime.timedelta(minutes=2)
webs = Web.get(d)
self.assertEqual(webs[0], self.web, "Unable to get new data")
def test_get(self):
self.web.save()
webs = Web.get()
self.assertEqual(webs[0], self.web, "Unable to get all data")
import datetime
d = datetime.datetime.now() - datetime.timedelta(minutes=2)
webs = Web.get(d)
self.assertEqual(webs[0], self.web, "Unable to get new data")
def test_update(self):
from database.database import InstanceNotFoundError
self.web.save()
self.web.c_id = 'new'
self.web.save()
web_db = Web.get_exactly('new')
self.assertEqual(web_db.c_id, self.web.c_id)
self.assertEqual(web_db.c_pk, self.web.c_pk)
with self.assertRaises(InstanceNotFoundError):
web_db = Web.get_exactly('url')
def test_update(self):
from database.database import InstanceNotFoundError
self.web.save()
self.web.c_id = "new"
self.web.save()
web_db = Web.get_exactly("new")
self.assertEqual(web_db.c_id, self.web.c_id)
self.assertEqual(web_db.c_pk, self.web.c_pk)
with self.assertRaises(InstanceNotFoundError):
web_db = Web.get_exactly("url")
def test_eq(self):
""" Equals method
"""
web1 = Web("test")
web2 = Web("test")
web3 = Web("na")
user = User("test")
self.assertEqual(web1, web2)
self.assertNotEqual(web1, web3)
self.assertNotEqual(web1, user)
self.assertNotEqual(web1, None)
def setUp(self):
Database.db_file = None # else runtime destroys testing framework
Database(self.gentest_config())
self.web = Web("url")
self.web.create_table()
self.cursor = self.web.db_connection.cursor()
def test_get_exactly(self):
self.web.save()
web = Web.get_exactly(self.web.c_id)
self.assertEqual(web, self.web, "Could not deserialize data")
class TestWeb(unittest.TestCase):
def gentest_config(self):
conffile = "test.conf"
with open(conffile, "w") as file:
file.write("""[Backend]
Hostname = localhost
User = foo
SSH-Port = 22
Database = test.db
Manifest_store = /dev/null
authorized_keys = auth_keys""")
return conffile
def setUp(self):
Database.db_file = None # else runtime destroys testing framework
Database(self.gentest_config())
self.web = Web("url")
self.web.create_table()
self.cursor = self.web.db_connection.cursor()
def tearDown(self):
""" Closes database connection and removes database from disk
"""
Database.db_connection.close()
os.remove(self.gentest_config())
os.remove(Database.db_file)
def test_create_table(self):
self.assertIsNotNone(self.web.db_connection,
"Database isn't correctly initialized")
self.cursor.execute("""SELECT name FROM sqlite_master
WHERE name = 'Web';""")
table_exists = False
for i in self.cursor.fetchone():
if 'Web' in i:
table_exists = True
self.assertTrue(table_exists)
def test_drop_table(self):
try:
self.web.drop_table()
except sqlite3.OperationalError:
self.assertFalse(True, "Error while table drop")
self.cursor.execute("""SELECT name FROM sqlite_master
WHERE name = 'Web';""")
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.assertEqual(self.web["c_id"], "url")
try:
self.web["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.
"""
attrs = self.web.get_attributes()
self.assertTrue(attrs, msg="Got no attributes")
self.assertTrue("c_id" 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
"""
self.web.save()
# connection restart
self.web.db_connection.close()
Database()
# test if data still present
cursor = self.web.db_connection.cursor()
cursor.execute("SELECT * FROM Web;")
res = cursor.fetchone()
self.assertIsNotNone(res, "Could not read test data from db.")
res = tuple(res)
self.assertEqual(res[3], "url", "Got unexpected data")
def test_remove(self):
self.web.save()
self.web.remove()
self.cursor.execute("""SELECT * FROM Web
WHERE c_id = 'url';""")
self.assertTrue(self.cursor.fetchone() is None,
"Removing Entries has failed")
def test_get(self):
self.web.save()
webs = Web.get()
self.assertEqual(webs[0], self.web, "Unable to get all data")
import datetime
d = datetime.datetime.now() - datetime.timedelta(minutes=2)
webs = Web.get(d)
self.assertEqual(webs[0], self.web, "Unable to get new data")
def test_override(self):
self.web.save()
web2 = Web('url')
web2.save()
def test_update(self):
from database.database import InstanceNotFoundError
self.web.save()
self.web.c_id = 'new'
self.web.save()
web_db = Web.get_exactly('new')
self.assertEqual(web_db.c_id, self.web.c_id)
self.assertEqual(web_db.c_pk, self.web.c_pk)
with self.assertRaises(InstanceNotFoundError):
web_db = Web.get_exactly('url')
def test_get_exactly(self):
self.web.save()
web = Web.get_exactly(self.web.c_id)
self.assertEqual(web, self.web, "Could not deserialize data")
def test_override(self):
self.web.save()
web2 = Web('url')
web2.save()
def setUp(self):
Database.db_file = None # else runtime destroys testing framework
Database(self.gentest_config())
self.web = Web("url")
self.web.create_table()
self.cursor = self.web.db_connection.cursor()
def test_get_exactly(self):
self.web.save()
web = Web.get_exactly(self.web.c_id)
self.assertEqual(web, self.web, "Could not deserialize data")
class TestWeb(unittest.TestCase):
def gentest_config(self):
conffile = "test.conf"
with open(conffile, "w") as file:
file.write(
"""[Backend]
Hostname = localhost
User = foo
SSH-Port = 22
Database = test.db
Manifest_store = /dev/null
authorized_keys = auth_keys"""
)
return conffile
def setUp(self):
Database.db_file = None # else runtime destroys testing framework
Database(self.gentest_config())
self.web = Web("url")
self.web.create_table()
self.cursor = self.web.db_connection.cursor()
def tearDown(self):
""" Closes database connection and removes database from disk
"""
Database.db_connection.close()
os.remove(self.gentest_config())
os.remove(Database.db_file)
def test_create_table(self):
self.assertIsNotNone(self.web.db_connection, "Database isn't correctly initialized")
self.cursor.execute(
"""SELECT name FROM sqlite_master
WHERE name = 'Web';"""
)
table_exists = False
for i in self.cursor.fetchone():
if "Web" in i:
table_exists = True
self.assertTrue(table_exists)
def test_drop_table(self):
try:
self.web.drop_table()
except sqlite3.OperationalError:
self.assertFalse(True, "Error while table drop")
self.cursor.execute(
"""SELECT name FROM sqlite_master
WHERE name = 'Web';"""
)
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.assertEqual(self.web["c_id"], "url")
try:
self.web["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.
"""
attrs = self.web.get_attributes()
self.assertTrue(attrs, msg="Got no attributes")
self.assertTrue("c_id" 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
"""
self.web.save()
# connection restart
self.web.db_connection.close()
Database()
# test if data still present
cursor = self.web.db_connection.cursor()
cursor.execute("SELECT * FROM Web;")
res = cursor.fetchone()
self.assertIsNotNone(res, "Could not read test data from db.")
res = tuple(res)
self.assertEqual(res[3], "url", "Got unexpected data")
def test_remove(self):
self.web.save()
self.web.remove()
self.cursor.execute(
"""SELECT * FROM Web
WHERE c_id = 'url';"""
)
self.assertTrue(self.cursor.fetchone() is None, "Removing Entries has failed")
def test_get(self):
self.web.save()
webs = Web.get()
self.assertEqual(webs[0], self.web, "Unable to get all data")
import datetime
d = datetime.datetime.now() - datetime.timedelta(minutes=2)
webs = Web.get(d)
self.assertEqual(webs[0], self.web, "Unable to get new data")
def test_override(self):
self.web.save()
web2 = Web("url")
web2.save()
def test_update(self):
from database.database import InstanceNotFoundError
self.web.save()
self.web.c_id = "new"
self.web.save()
web_db = Web.get_exactly("new")
self.assertEqual(web_db.c_id, self.web.c_id)
self.assertEqual(web_db.c_pk, self.web.c_pk)
with self.assertRaises(InstanceNotFoundError):
web_db = Web.get_exactly("url")
def test_get_exactly(self):
self.web.save()
web = Web.get_exactly(self.web.c_id)
self.assertEqual(web, self.web, "Could not deserialize data")
def test_override(self):
self.web.save()
web2 = Web("url")
web2.save()
