def test_db_open(self):
"""Test database open"""
db_path = os.path.join(self.res_dir, 'regular.sqlite')
db_obj = db.DataBase(db_path)
db_obj.close()
self.assertTrue(True)
def test_db_create(self):
"""Test database create"""
db_path = os.path.join(self.test_dir, 'test.sqlite')
db_obj = db.DataBase(db_path, query=QUERY_CREATE, create=True)
db_obj.close()
self.assertTrue(os.path.isfile(db_path))
def test_db_connection(self):
"""Test connection to database"""
db_path = os.path.join(self.res_dir, 'regular.sqlite')
db_obj = db.DataBase(db_path)
self.assertTrue(isinstance(db_obj.connection, sqlite.Connection))
db_obj.close()
def test_db_cursor(self):
"""Test cursor property"""
db_path = os.path.join(self.res_dir, 'regular.sqlite')
db_obj = db.DataBase(db_path)
self.assertTrue(isinstance(db_obj.cursor, sqlite.Cursor))
db_obj.close()
def test_db_execute(self):
"""Test query execution"""
db_path = os.path.join(self.test_dir, 'execute.sqlite')
db_obj = db.DataBase(db_path, create=True)
db_obj.execute(QUERY_CREATE)
db_obj.execute(QUERY_INSERT)
self.assertTrue(isinstance(db_obj.cursor, sqlite.Cursor))
self.assertRaises(sqlite.Warning, db_obj.execute, QUERY_MULTIPLE)
db_obj.close()
def test_db_get(self):
"""Test getting single row"""
db_path = os.path.join(self.test_dir, 'get.sqlite')
db_obj = db.DataBase(db_path, create=True)
db_obj.execute(QUERY_CREATE)
db_obj.execute(QUERY_INSERT)
res = db_obj.get(QUERY_GET)
self.assertEqual(res[0], 'key')
self.assertEqual(res[1], 'value')
self.assertEqual(res['key'], 'key')
self.assertEqual(res['value'], 'value')
db_obj.close()
def test_db_all(self):
"""Test getting results by query"""
db_path = os.path.join(self.test_dir, 'get.sqlite')
db_obj = db.DataBase(db_path, create=True)
db_obj.execute(QUERY_CREATE)
db_obj.execute(QUERY_INSERT_MULTIPLE)
res = db_obj.all(QUERY_GET)
self.assertEqual(res[0][0], 'one')
self.assertEqual(res[0][1], 'first')
self.assertEqual(res[1][0], 'two')
self.assertEqual(res[1][1], 'second')
self.assertEqual(res[2][0], 'three')
self.assertEqual(res[2][1], 'third')
db_obj.close()