def test_get_db():
db = get_db(filename=None, isolation_level="DEFERRED", journal_mode='OFF')
assert isinstance(db, SQLiteDB)
db.close()
filename = NamedTemporaryFile().name
db = get_db(filename=filename, isolation_level="DEFERRED", journal_mode='OFF')
assert isinstance(db, SQLiteDB)
assert os.path.exists(filename)
mode = list(db.execute('pragma journal_mode'))[0]
assert mode == {'journal_mode': 'off'}
db.close()
def test_get_tables():
db = get_db(filename=None)
assert list_tables(db) == []
table_name = 'foo'
create_lookup_table(db, table_name, key_type='TEXT', value_type='TEXT',
drop_first=False)
tables = get_tables(db)
assert tables == ['foo']
def test_get_db_journal_modes():
journal_modes = ['DELETE', 'TRUNCATE', 'PERSIST', 'MEMORY', 'WAL', 'OFF']
for journal_mode in journal_modes:
filename = NamedTemporaryFile().name
db = get_db(filename=filename, isolation_level="DEFERRED", journal_mode=journal_mode)
assert isinstance(db, SQLiteDB)
assert os.path.exists(filename)
mode = list(db.execute('pragma journal_mode'))[0]
assert mode == {'journal_mode': journal_mode.lower()}
db.close()
def test_create_lookup_table():
db = get_db(filename=None)
assert list_tables(db) == []
table_name = 'foo'
create_lookup_table(db, table_name, key_type='TEXT', value_type='TEXT',
drop_first=False)
assert list_tables(db) == [{'name': 'foo'}]
drop_table(db, table_name)
assert list_tables(db) == []
db.close()
def __init__(self, db_file, isolation_level=None, journal_mode=None):
if isolation_level is None:
isolation_level = "DEFERRED"
if journal_mode is None:
journal_mode = "OFF"
super().__init__()
self.db_file = db_file
self.db = get_db(filename=db_file,
isolation_level=isolation_level,
journal_mode=journal_mode)
self.db_type = 'sqlite'
logger.info('Tables: %s' % self.tables.__repr__())
def test_db_is_open():
db = get_db(filename=None, isolation_level="DEFERRED", journal_mode='OFF')
assert isinstance(db, SQLiteDB)
assert db_is_open(db)
db.close()
assert not db_is_open(db)