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__