158 lines
7.2 KiB
Python
158 lines
7.2 KiB
Python
# Copyright 2014 IBM Corp.
|
|
#
|
|
# 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.
|
|
|
|
from migrate import exceptions as versioning_exceptions
|
|
from migrate.versioning import api as versioning_api
|
|
import mock
|
|
import sqlalchemy
|
|
|
|
from nova.db.sqlalchemy import api as db_api
|
|
from nova.db.sqlalchemy import migration
|
|
from nova import test
|
|
|
|
|
|
@mock.patch.object(migration, 'db_version', return_value=2)
|
|
@mock.patch.object(migration, '_find_migrate_repo', return_value='repo')
|
|
@mock.patch.object(versioning_api, 'upgrade')
|
|
@mock.patch.object(versioning_api, 'downgrade')
|
|
@mock.patch.object(migration, 'get_engine', return_value='engine')
|
|
class TestDbSync(test.NoDBTestCase):
|
|
|
|
def test_version_none(self, mock_get_engine, mock_downgrade, mock_upgrade,
|
|
mock_find_repo, mock_version):
|
|
database = 'fake'
|
|
migration.db_sync(database=database)
|
|
mock_version.assert_called_once_with(database, context=None)
|
|
mock_find_repo.assert_called_once_with(database)
|
|
mock_get_engine.assert_called_once_with(database, context=None)
|
|
mock_upgrade.assert_called_once_with('engine', 'repo', None)
|
|
self.assertFalse(mock_downgrade.called)
|
|
|
|
def test_downgrade(self, mock_get_engine, mock_downgrade, mock_upgrade,
|
|
mock_find_repo, mock_version):
|
|
database = 'fake'
|
|
migration.db_sync(1, database=database)
|
|
mock_version.assert_called_once_with(database, context=None)
|
|
mock_find_repo.assert_called_once_with(database)
|
|
mock_get_engine.assert_called_once_with(database, context=None)
|
|
mock_downgrade.assert_called_once_with('engine', 'repo', 1)
|
|
self.assertFalse(mock_upgrade.called)
|
|
|
|
|
|
@mock.patch.object(migration, '_find_migrate_repo', return_value='repo')
|
|
@mock.patch.object(versioning_api, 'db_version')
|
|
@mock.patch.object(migration, 'get_engine')
|
|
class TestDbVersion(test.NoDBTestCase):
|
|
|
|
def test_db_version(self, mock_get_engine, mock_db_version,
|
|
mock_find_repo):
|
|
database = 'fake'
|
|
mock_get_engine.return_value = 'engine'
|
|
migration.db_version(database)
|
|
mock_find_repo.assert_called_once_with(database)
|
|
mock_db_version.assert_called_once_with('engine', 'repo')
|
|
|
|
def test_not_controlled(self, mock_get_engine, mock_db_version,
|
|
mock_find_repo):
|
|
database = 'api'
|
|
mock_get_engine.side_effect = ['engine', 'engine', 'engine']
|
|
exc = versioning_exceptions.DatabaseNotControlledError()
|
|
mock_db_version.side_effect = [exc, '']
|
|
metadata = mock.MagicMock()
|
|
metadata.tables.return_value = []
|
|
with mock.patch.object(sqlalchemy, 'MetaData',
|
|
metadata), mock.patch.object(migration,
|
|
'db_version_control') as mock_version_control:
|
|
migration.db_version(database)
|
|
mock_version_control.assert_called_once_with(
|
|
migration.INIT_VERSION['api'], database, context=None)
|
|
db_version_calls = [mock.call('engine', 'repo')] * 2
|
|
self.assertEqual(db_version_calls, mock_db_version.call_args_list)
|
|
engine_calls = [mock.call(database, context=None)] * 3
|
|
self.assertEqual(engine_calls, mock_get_engine.call_args_list)
|
|
|
|
def test_db_version_init_race(self, mock_get_engine, mock_db_version,
|
|
mock_find_repo):
|
|
# This test exercises bug 1804652 by causing
|
|
# versioning_api.version_contro() to raise an unhandleable error the
|
|
# first time it is called.
|
|
database = 'api'
|
|
mock_get_engine.return_value = 'engine'
|
|
exc = versioning_exceptions.DatabaseNotControlledError()
|
|
mock_db_version.side_effect = [exc, '']
|
|
metadata = mock.MagicMock()
|
|
metadata.tables.return_value = []
|
|
with mock.patch.object(sqlalchemy, 'MetaData',
|
|
metadata), mock.patch.object(migration,
|
|
'db_version_control') as mock_version_control:
|
|
# db_version_control raises an unhandleable error because we were
|
|
# racing to initialise with another process.
|
|
mock_version_control.side_effect = test.TestingException
|
|
migration.db_version(database)
|
|
mock_version_control.assert_called_once_with(
|
|
migration.INIT_VERSION['api'], database, context=None)
|
|
db_version_calls = [mock.call('engine', 'repo')] * 2
|
|
self.assertEqual(db_version_calls, mock_db_version.call_args_list)
|
|
engine_calls = [mock.call(database, context=None)] * 3
|
|
self.assertEqual(engine_calls, mock_get_engine.call_args_list)
|
|
|
|
def test_db_version_raise_on_error(self, mock_get_engine, mock_db_version,
|
|
mock_find_repo):
|
|
# This test asserts that we will still raise a persistent error after
|
|
# working around bug 1804652.
|
|
database = 'api'
|
|
mock_get_engine.return_value = 'engine'
|
|
mock_db_version.side_effect = \
|
|
versioning_exceptions.DatabaseNotControlledError
|
|
metadata = mock.MagicMock()
|
|
metadata.tables.return_value = []
|
|
with mock.patch.object(sqlalchemy, 'MetaData',
|
|
metadata), mock.patch.object(migration,
|
|
'db_version_control') as mock_version_control:
|
|
# db_version_control raises an unhandleable error because we were
|
|
# racing to initialise with another process.
|
|
mock_version_control.side_effect = test.TestingException
|
|
self.assertRaises(test.TestingException,
|
|
migration.db_version, database)
|
|
|
|
|
|
@mock.patch.object(migration, '_find_migrate_repo', return_value='repo')
|
|
@mock.patch.object(migration, 'get_engine', return_value='engine')
|
|
@mock.patch.object(versioning_api, 'version_control')
|
|
class TestDbVersionControl(test.NoDBTestCase):
|
|
|
|
def test_version_control(self, mock_version_control, mock_get_engine,
|
|
mock_find_repo):
|
|
database = 'fake'
|
|
migration.db_version_control(database=database)
|
|
mock_find_repo.assert_called_once_with(database)
|
|
mock_version_control.assert_called_once_with('engine', 'repo', None)
|
|
|
|
|
|
class TestGetEngine(test.NoDBTestCase):
|
|
|
|
def test_get_main_engine(self):
|
|
with mock.patch.object(db_api, 'get_engine',
|
|
return_value='engine') as mock_get_engine:
|
|
engine = migration.get_engine()
|
|
self.assertEqual('engine', engine)
|
|
mock_get_engine.assert_called_once_with(context=None)
|
|
|
|
def test_get_api_engine(self):
|
|
with mock.patch.object(db_api, 'get_api_engine',
|
|
return_value='api_engine') as mock_get_engine:
|
|
engine = migration.get_engine('api')
|
|
self.assertEqual('api_engine', engine)
|
|
mock_get_engine.assert_called_once_with()
|