diff --git a/karbor/cmd/manage.py b/karbor/cmd/manage.py index f7df49f8..6d11b3dc 100644 --- a/karbor/cmd/manage.py +++ b/karbor/cmd/manage.py @@ -65,7 +65,7 @@ class DbCommands(object): sys.exit(1) try: return db_migration.db_sync(version) - except db_exc.DbMigrationError as ex: + except db_exc.DBMigrationError as ex: print("Error during database migration: %s" % ex) sys.exit(1) diff --git a/karbor/tests/unit/test_cmd.py b/karbor/tests/unit/test_cmd.py index d615cf12..f0042d38 100644 --- a/karbor/tests/unit/test_cmd.py +++ b/karbor/tests/unit/test_cmd.py @@ -80,7 +80,7 @@ class TestKarborManageCmd(base.TestCase): @mock.patch("oslo_db.sqlalchemy.migration.db_sync") def test_db_commands_script_not_present(self, db_sync): - db_sync.side_effect = db_exc.DbMigrationError + db_sync.side_effect = db_exc.DBMigrationError(None) db_cmds = karbor_manage.DbCommands() exit = self.assertRaises(SystemExit, db_cmds.sync, 101) self.assertEqual(1, exit.code)