def test_tables_names(self):
with sc.SqliteDb(DB_NAME, DB_INIT_SCRIPT_NAME) as db:
tables = db.tables_names()
self.assertListEqual(['test_points'], tables)
db.execute('CREATE TABLE abc (x INTEGER)')
tables = db.tables_names()
self.assertListEqual(['test_points', 'abc'], tables)
def test_construct_and_open_in_memory(self):
with sc.SqliteDb() as db:
self.assertIsNotNone(db.connection)
self.assertIsNotNone(db.cursor)
self.assertTrue(db.is_in_memory())
self.assertTrue(db.is_empty_db())
db.execute('CREATE TABLE aaa(x INTEGER)')
self.assertFalse(db.is_empty_db())
self.assertIsNone(db.connection)
self.assertIsNone(db.cursor)
def test_construct_and_open_and_init_on_file(self):
with sc.SqliteDb(DB_NAME, DB_INIT_SCRIPT_NAME) as db:
self.assertIsNotNone(db.connection)
self.assertIsNotNone(db.cursor)
self.assertFalse(db.is_in_memory())
self.assertFalse(db.is_empty_db())
self.assertTrue(os.path.isfile(DB_NAME))
cursor = db.execute('SELECT count(*) FROM test_points')
amount = cursor.fetchone()[0]
self.assertEqual(3, amount)
self.assertIsNone(db.connection)
self.assertIsNone(db.cursor)
def test_reopening_closed_connection(self):
db = sc.SqliteDb(DB_NAME, DB_INIT_SCRIPT_NAME)
db.open()
db.execute('INSERT INTO test_points(x, y) VALUES (100, 200)')
db.commit()
db.close()
self.assertTrue(os.path.isfile(DB_NAME))
self.assertIsNone(db.connection)
self.assertIsNone(db.cursor)
db.open()
self.assertIsNotNone(db.connection)
self.assertIsNotNone(db.cursor)
self.assertFalse(db.is_empty_db())
cursor = db.execute('SELECT count(*) FROM test_points')
amount = cursor.fetchone()[0]
self.assertEqual(3 + 1, amount)
def test_constructor_on_file_with_init_script(self):
db = sc.SqliteDb(DB_NAME, DB_INIT_SCRIPT_NAME)
self.assertIsNone(db.connection)
self.assertIsNone(db.cursor)
self.assertFalse(db.is_in_memory())
def test_closed_at_construction_on_file(self):
db = sc.SqliteDb(DB_NAME)
self.assertIsNone(db.connection)
self.assertIsNone(db.cursor)
self.assertFalse(db.is_in_memory())
def test_closed_at_construction_in_memory(self):
db = sc.SqliteDb()
self.assertIsNone(db.connection)
self.assertIsNone(db.cursor)
self.assertTrue(db.is_in_memory())
def test_cursor_column_name(self):
db = sc.SqliteDb(DB_NAME, DB_INIT_SCRIPT_NAME)
db.open()
cursor = db.execute('SELECT * FROM test_points')
columns = sc.cursor_column_names(cursor)
self.assertListEqual(['x', 'y'], columns)
def test_column_names(self):
with sc.SqliteDb(DB_NAME, DB_INIT_SCRIPT_NAME) as db:
columns = db.columns_names('test_points')
self.assertListEqual(['x', 'y'], columns)
columns = db.columns_names('UNEXISTING_TABLE____Aaaaaa')
self.assertListEqual([], columns)