def test_upgrade_dryrun(self, needupdate):
args = self.Args({"dry_run": True})
db = self.PartialMockDb(args, None)
self.mox.StubOutWithMock(db, "get_current_db_version")
self.mox.StubOutWithMock(db, "sql_version_matrix")
self.mox.StubOutWithMock(db, "sql_version_resolve")
self.mox.StubOutWithMock(db, "check_connection")
# Stub out to ensure it's NOT called
self.mox.StubOutWithMock(db, "psql")
db.check_connection()
versions = ["OMERO4.4__0", "OMERO5.0__0"]
if needupdate:
db.get_current_db_version().AndReturn(("OMERO4.4", "0"))
db.sql_version_matrix().AndReturn(([], versions))
db.sql_version_resolve([], versions, versions[0]).AndReturn(
["./sql/psql/OMERO5.0__0/OMERO4.4__0.sql"]
)
else:
db.get_current_db_version().AndReturn(("OMERO5.0", "0"))
db.sql_version_matrix().AndReturn(([], versions))
self.mox.ReplayAll()
if needupdate:
with pytest.raises(Stop) as excinfo:
db.upgrade()
assert excinfo.value.rc == 2
assert excinfo.value.msg == (
"Database upgrade required OMERO4.4__0->OMERO5.0__0"
)
else:
db.upgrade()
self.mox.VerifyAll()
def test_upgrade(self, needupdate):
args = self.Args({"dry_run": False})
db = self.PartialMockDb(args, None)
self.mox.StubOutWithMock(db, "get_current_db_version")
self.mox.StubOutWithMock(db, "sql_version_matrix")
self.mox.StubOutWithMock(db, "sql_version_resolve")
self.mox.StubOutWithMock(db, "check_connection")
self.mox.StubOutWithMock(db, "psql")
db.check_connection()
versions = ["OMERO3.0__0", "OMERO4.4__0", "OMERO5.0__0"]
if needupdate:
db.get_current_db_version().AndReturn(("OMERO3.0", "0"))
db.sql_version_matrix().AndReturn(([], versions))
db.sql_version_resolve([], versions, versions[0]).AndReturn(
[
"./sql/psql/OMERO4.4__0/OMERO3.0__0.sql",
"./sql/psql/OMERO5.0__0/OMERO4.4__0.sql",
]
)
db.psql("-f", "./sql/psql/OMERO4.4__0/OMERO3.0__0.sql")
db.psql("-f", "./sql/psql/OMERO5.0__0/OMERO4.4__0.sql")
else:
db.get_current_db_version().AndReturn(("OMERO5.0", "0"))
db.sql_version_matrix().AndReturn(([], versions))
self.mox.ReplayAll()
db.upgrade()
self.mox.VerifyAll()
def test_init(self, sqlfile, dryrun):
ext = self.mox.CreateMock(external.External)
if sqlfile != "notprovided":
omerosql = "omero.sql"
else:
omerosql = None
args = self.Args(
{"omerosql": omerosql, "rootpass": "******", "dry_run": dryrun}
)
db = self.PartialMockDb(args, ext)
self.mox.StubOutWithMock(db, "psql")
self.mox.StubOutWithMock(omero_database.db, "timestamp_filename")
self.mox.StubOutWithMock(os.path, "exists")
self.mox.StubOutWithMock(db, "check_connection")
self.mox.StubOutWithMock(db, "upgrade")
self.mox.StubOutWithMock(os, "remove")
db.check_connection()
if sqlfile == "notprovided":
omerosql = "omero-00000000-000000-000000.sql"
omero_database.db.timestamp_filename("omero", "sql").AndReturn(
omerosql
)
else:
os.path.exists(omerosql).AndReturn(sqlfile == "exists")
if sqlfile == "notprovided" and not dryrun:
ext.omero_cli(
["db", "script", "-f", omerosql, "", "", args.rootpass]
)
if sqlfile == "exists":
db.upgrade()
if sqlfile != "missing" and not dryrun:
db.psql("-f", omerosql)
if sqlfile == "notprovided" and not dryrun:
os.remove("omero-00000000-000000-000000.sql")
self.mox.ReplayAll()
if sqlfile == "missing":
with pytest.raises(Stop) as excinfo:
db.init()
assert excinfo.value.msg == "SQL file not found"
else:
db.init()
self.mox.VerifyAll()
def test_upgrade_not_initialised(self, dryrun):
args = self.Args({"dry_run": dryrun})
db = self.PartialMockDb(args, None)
self.mox.StubOutWithMock(db, "get_current_db_version")
self.mox.StubOutWithMock(db, "check_connection")
db.check_connection()
exc = external.RunException(
"test psql failure", "psql", [], -1, "", ""
)
db.get_current_db_version().AndRaise(exc)
self.mox.ReplayAll()
with pytest.raises(Stop) as excinfo:
db.upgrade()
assert excinfo.value.rc == 3
assert excinfo.value.msg == "Unable to get database version"
self.mox.VerifyAll()
def test_check_connection(self, connected):
db = self.PartialMockDb(None, None)
self.mox.StubOutWithMock(db, "psql")
if connected:
db.psql("-c", r"\conninfo")
else:
db.psql("-c", r"\conninfo").AndRaise(
external.RunException("", "", [], 1, "", "")
)
self.mox.ReplayAll()
if connected:
db.check_connection()
else:
with pytest.raises(Stop) as excinfo:
db.check_connection()
assert excinfo.value.msg == "Database connection check failed"
self.mox.VerifyAll()
def test_dump(self, dumpfile, dryrun):
args = self.Args({"dry_run": dryrun, "dumpfile": dumpfile})
db = self.PartialMockDb(args, None)
self.mox.StubOutWithMock(omero_database.db, "timestamp_filename")
self.mox.StubOutWithMock(db, "get_db_args_env")
self.mox.StubOutWithMock(db, "pgdump")
self.mox.StubOutWithMock(db, "check_connection")
db.check_connection()
if not dumpfile:
db.get_db_args_env().AndReturn(self.create_db_test_params())
dumpfile = "omero-database-name-00000000-000000-000000.pgdump"
omero_database.db.timestamp_filename(
"omero-database-name", "pgdump"
).AndReturn(dumpfile)
if not dryrun:
db.pgdump("-Fc", "-f", dumpfile).AndReturn("")
self.mox.ReplayAll()
db.dump()
self.mox.VerifyAll()