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()