diff options
Diffstat (limited to 'tests/sqlalchemy/test_migrations.py')
| -rw-r--r-- | tests/sqlalchemy/test_migrations.py | 154 |
1 files changed, 154 insertions, 0 deletions
diff --git a/tests/sqlalchemy/test_migrations.py b/tests/sqlalchemy/test_migrations.py new file mode 100644 index 0000000..f39bb73 --- /dev/null +++ b/tests/sqlalchemy/test_migrations.py @@ -0,0 +1,154 @@ +# Copyright 2010-2011 OpenStack Foundation +# Copyright 2012-2013 IBM Corp. +# All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); you may +# not use this file except in compliance with the License. You may obtain +# a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT +# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the +# License for the specific language governing permissions and limitations +# under the License. + +import mock +from oslotest import base as test_base + +from oslo.db.sqlalchemy import test_migrations as migrate + + +class TestWalkVersions(test_base.BaseTestCase, migrate.WalkVersionsMixin): + def setUp(self): + super(TestWalkVersions, self).setUp() + self.migration_api = mock.MagicMock() + self.engine = mock.MagicMock() + self.REPOSITORY = mock.MagicMock() + self.INIT_VERSION = 4 + + def test_migrate_up(self): + self.migration_api.db_version.return_value = 141 + + self._migrate_up(self.engine, 141) + + self.migration_api.upgrade.assert_called_with( + self.engine, self.REPOSITORY, 141) + self.migration_api.db_version.assert_called_with( + self.engine, self.REPOSITORY) + + def test_migrate_up_with_data(self): + test_value = {"a": 1, "b": 2} + self.migration_api.db_version.return_value = 141 + self._pre_upgrade_141 = mock.MagicMock() + self._pre_upgrade_141.return_value = test_value + self._check_141 = mock.MagicMock() + + self._migrate_up(self.engine, 141, True) + + self._pre_upgrade_141.assert_called_with(self.engine) + self._check_141.assert_called_with(self.engine, test_value) + + def test_migrate_down(self): + self.migration_api.db_version.return_value = 42 + + self.assertTrue(self._migrate_down(self.engine, 42)) + self.migration_api.db_version.assert_called_with( + self.engine, self.REPOSITORY) + + def test_migrate_down_not_implemented(self): + self.migration_api.downgrade.side_effect = NotImplementedError + self.assertFalse(self._migrate_down(self.engine, 42)) + + def test_migrate_down_with_data(self): + self._post_downgrade_043 = mock.MagicMock() + self.migration_api.db_version.return_value = 42 + + self._migrate_down(self.engine, 42, True) + + self._post_downgrade_043.assert_called_with(self.engine) + + @mock.patch.object(migrate.WalkVersionsMixin, '_migrate_up') + @mock.patch.object(migrate.WalkVersionsMixin, '_migrate_down') + def test_walk_versions_all_default(self, _migrate_up, _migrate_down): + self.REPOSITORY.latest = 20 + self.migration_api.db_version.return_value = self.INIT_VERSION + + self._walk_versions() + + self.migration_api.version_control.assert_called_with( + None, self.REPOSITORY, self.INIT_VERSION) + self.migration_api.db_version.assert_called_with( + None, self.REPOSITORY) + + versions = range(self.INIT_VERSION + 1, self.REPOSITORY.latest + 1) + upgraded = [mock.call(None, v, with_data=True) for v in versions] + self.assertEqual(self._migrate_up.call_args_list, upgraded) + + downgraded = [mock.call(None, v - 1) for v in reversed(versions)] + self.assertEqual(self._migrate_down.call_args_list, downgraded) + + @mock.patch.object(migrate.WalkVersionsMixin, '_migrate_up') + @mock.patch.object(migrate.WalkVersionsMixin, '_migrate_down') + def test_walk_versions_all_true(self, _migrate_up, _migrate_down): + self.REPOSITORY.latest = 20 + self.migration_api.db_version.return_value = self.INIT_VERSION + + self._walk_versions(self.engine, snake_walk=True, downgrade=True) + + versions = range(self.INIT_VERSION + 1, self.REPOSITORY.latest + 1) + upgraded = [] + for v in versions: + upgraded.append(mock.call(self.engine, v, with_data=True)) + upgraded.append(mock.call(self.engine, v)) + upgraded.extend( + [mock.call(self.engine, v) for v in reversed(versions)] + ) + self.assertEqual(upgraded, self._migrate_up.call_args_list) + + downgraded_1 = [ + mock.call(self.engine, v - 1, with_data=True) for v in versions + ] + downgraded_2 = [] + for v in reversed(versions): + downgraded_2.append(mock.call(self.engine, v - 1)) + downgraded_2.append(mock.call(self.engine, v - 1)) + downgraded = downgraded_1 + downgraded_2 + self.assertEqual(self._migrate_down.call_args_list, downgraded) + + @mock.patch.object(migrate.WalkVersionsMixin, '_migrate_up') + @mock.patch.object(migrate.WalkVersionsMixin, '_migrate_down') + def test_walk_versions_true_false(self, _migrate_up, _migrate_down): + self.REPOSITORY.latest = 20 + self.migration_api.db_version.return_value = self.INIT_VERSION + + self._walk_versions(self.engine, snake_walk=True, downgrade=False) + + versions = range(self.INIT_VERSION + 1, self.REPOSITORY.latest + 1) + + upgraded = [] + for v in versions: + upgraded.append(mock.call(self.engine, v, with_data=True)) + upgraded.append(mock.call(self.engine, v)) + self.assertEqual(upgraded, self._migrate_up.call_args_list) + + downgraded = [ + mock.call(self.engine, v - 1, with_data=True) for v in versions + ] + self.assertEqual(self._migrate_down.call_args_list, downgraded) + + @mock.patch.object(migrate.WalkVersionsMixin, '_migrate_up') + @mock.patch.object(migrate.WalkVersionsMixin, '_migrate_down') + def test_walk_versions_all_false(self, _migrate_up, _migrate_down): + self.REPOSITORY.latest = 20 + self.migration_api.db_version.return_value = self.INIT_VERSION + + self._walk_versions(self.engine, snake_walk=False, downgrade=False) + + versions = range(self.INIT_VERSION + 1, self.REPOSITORY.latest + 1) + + upgraded = [ + mock.call(self.engine, v, with_data=True) for v in versions + ] + self.assertEqual(upgraded, self._migrate_up.call_args_list) |