def test_necessary_upgrade_scripts_ran(database_mock_setup):
with pytest.raises(SystemExit):
database.upgrade_schema()
# Assert the necessary scripts were called
for i in range(CDV - 3, CDV):
script_name = 'upgrade_to_{}'.format(i)
assert getattr(database, script_name).called
# Fixes are inclusive of the "current" database version
for i in range(CDV - 4, CDV):
available_fixes = fixes.AVAILABLE_FIXES.get(i, [])
for fix_id in available_fixes:
assert getattr(fixes, fix_id).called
# But not the scripts before it
for i in range(1, CDV - 4):
script_name = 'upgrade_to_{}'.format(i)
assert getattr(database, script_name).called is False
# Fixes included in the previous version, could have run
for i in range(1, CDV - 5):
available_fixes = fixes.AVAILABLE_FIXES.get(i, [])
for fix_id in available_fixes:
assert getattr(fixes, fix_id).called is False
for check_id in checks.AVAILABLE_CHECKS:
assert getattr(checks, check_id).called
def test_necessary_upgrade_scripts_ran(database_mock_setup):
with pytest.raises(SystemExit):
database.upgrade_schema()
# Assert the necessary scripts were called
for i in range(CDV-3, CDV):
script_name = 'upgrade_to_{}'.format(i)
assert getattr(database, script_name).called
# But not the scripts before it
for i in range(1, CDV-4):
script_name = 'upgrade_to_{}'.format(i)
assert getattr(database, script_name).called is False
def test_necessary_upgrade_scripts_ran(database_mock_setup):
with pytest.raises(SystemExit):
database.upgrade_schema()
# Assert the necessary scripts were called
for i in range(CDV - 3, CDV):
script_name = 'upgrade_to_{}'.format(i)
assert getattr(database, script_name).called
# But not the scripts before it
for i in range(1, CDV - 4):
script_name = 'upgrade_to_{}'.format(i)
assert getattr(database, script_name).called is False
def test_all_upgrade_scripts_ran(database_mock_setup):
with pytest.raises(SystemExit):
database.upgrade_schema()
for i in range(1, CDV):
script_name = 'upgrade_to_{}'.format(i)
assert getattr(database, script_name).called
for available_fixes in fixes.AVAILABLE_FIXES.values():
for fix_id in available_fixes:
assert getattr(fixes, fix_id).called
for check_id in checks.AVAILABLE_CHECKS:
assert getattr(checks, check_id).called
def test_all_upgrade_scripts_ran(database_mock_setup, api_db):
with pytest.raises(SystemExit):
database.upgrade_schema()
for i in range(1, CDV):
script_name = 'upgrade_to_{}'.format(i)
assert getattr(database, script_name).called
for available_fixes in fixes.AVAILABLE_FIXES.values():
for fix_id in available_fixes:
assert getattr(fixes, fix_id).called
for check_id in checks.AVAILABLE_CHECKS:
assert getattr(checks, check_id).called
# Upgrade Cleanup
# After the upgrade scripts run we will have extra providers since those
# Are needed for all the previous tests. We should either make this test run first
# For undo any changes that adjust our seed data state
api_db.providers.delete_one({'label': 'Primary Storage'})
api_db.providers.delete_one({'label': 'Static Compute'})
def test_all_upgrade_scripts_ran(database_mock_setup):
with pytest.raises(SystemExit):
database.upgrade_schema()
for i in range(1, CDV):
script_name = 'upgrade_to_{}'.format(i)
assert getattr(database, script_name).called
def test_all_upgrade_scripts_ran(database_mock_setup):
with pytest.raises(SystemExit):
database.upgrade_schema()
for i in range(1, CDV):
script_name = 'upgrade_to_{}'.format(i)
assert getattr(database, script_name).called