def test_reopen(tmpdir):
"""Test that we can reopen a database of the correct version."""
db = tmpdir.join("test.db")
s = database.create_session(str(db))
s.close()
s = database.create_session(str(db))
s.close()
def test_arg_override(tmpdir, db_env):
# check that if we supply an argument, that
# is used rather than the environment variable
db = tmpdir.join('test_other.db')
s = database.create_session(str(db))
assert (not db_env.check())
assert (db.check())
def test_default(tmpdir):
db = tmpdir.join('test.db')
# override the NCI-specific default
database.__DEFAULT_DB__ = str(db)
s = database.create_session()
assert (db.check())
def test_outdated(tmpdir):
"""Test that we can't use an outdated database"""
db = tmpdir.join("test.db")
s = database.create_session(str(db))
# check that the current version matches that defined in the module
ver = s.execute("PRAGMA user_version").fetchone()[0]
assert ver == database.__DB_VERSION__
# reset version to one prior
s.execute("PRAGMA user_version={}".format(database.__DB_VERSION__ - 1))
s.close()
# recreate the session
with pytest.raises(Exception, match="Incompatible database versions"):
s = database.create_session(str(db))
def test_outdated_notmodified(tmpdir):
"""Test that we don't try to modify an outdated database.
This includes adding tables that don't yet exist because
it's a previous version.
"""
# set up an empty database with a previous version
db = tmpdir.join("test.db")
conn = sa.create_engine("sqlite:///" + str(db)).connect()
conn.execute("PRAGMA user_version={}".format(database.__DB_VERSION__ - 1))
conn.close()
# try to create the session
# this should fail and not modify the existing database
with pytest.raises(Exception):
s = database.create_session(str(db))
# reopen the connection and ensure tables weren't created
conn = sa.create_engine("sqlite:///" + str(db)).connect()
with pytest.raises(sa.exc.OperationalError, match="no such table"):
conn.execute("SELECT * FROM ncfiles")
def session_db(tmpdir):
db = tmpdir.join('test.db')
s = database.create_session(str(db))
yield s, db
s.close()
def test_env_var(db_env):
# make sure we use the environment variable
# override with no arguments supplied
s = database.create_session()
assert (db_env.check())