def test_upgrade_from_version(self, File):
from palladium import __version__
persister = File('model-{version}')
with patch.object(File, 'upgrade_steps') as upgrade_steps:
persister.upgrade(from_version='0.34')
upgrade_steps.run.assert_called_with(persister, '0.34', __version__)
assert persister.list_properties()['db-version'] == __version__
def test_upgrade_from_version(self, File):
from palladium import __version__
persister = File('model-{version}')
with patch.object(File, 'upgrade_steps') as upgrade_steps:
persister.upgrade(from_version='0.34')
upgrade_steps.run.assert_called_with(persister, '0.34', __version__)
assert persister.list_properties()['db-version'] == __version__
def test_upgrade_no_args(self, File):
from palladium import __version__
persister = File('model-{version}')
with patch.object(File, '_read_md',
return_value={'properties': {'db-version': '0.33'}}):
with patch.object(File, 'upgrade_steps') as upgrade_steps:
persister.upgrade()
upgrade_steps.run.assert_called_with(persister, '0.33', __version__)
assert persister.list_properties()['db-version'] == __version__
def test_upgrade_with_legacy_md(self, File):
from palladium import __version__
persister = File('model-{version}')
legacy_md = [{'some': 'model'}]
with patch.object(File, '_read_md',
side_effect=[legacy_md, {'properties': {}}]):
with patch.object(File, 'upgrade_steps') as upgrade_steps:
persister.upgrade()
upgrade_steps.run.assert_called_with(persister, '0.0', __version__)
assert persister.list_properties()['db-version'] == __version__
def test_upgrade_no_args(self, File):
from palladium import __version__
persister = File('model-{version}')
with patch.object(File,
'_read_md',
return_value={'properties': {
'db-version': '0.33'
}}):
with patch.object(File, 'upgrade_steps') as upgrade_steps:
persister.upgrade()
upgrade_steps.run.assert_called_with(persister, '0.33', __version__)
assert persister.list_properties()['db-version'] == __version__
def test_upgrade_with_legacy_md(self, File):
from palladium import __version__
persister = File('model-{version}')
legacy_md = [{'some': 'model'}]
with patch.object(File,
'_read_md',
side_effect=[legacy_md, {
'properties': {}
}]):
with patch.object(File, 'upgrade_steps') as upgrade_steps:
persister.upgrade()
upgrade_steps.run.assert_called_with(persister, '0.0', __version__)
assert persister.list_properties()['db-version'] == __version__
