250 lines
11 KiB
Python
250 lines
11 KiB
Python
# Copyright 2014 Rackspace Hosting
|
|
# 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 fixtures
|
|
import mock
|
|
|
|
from glance.cmd import manage
|
|
from glance.db.sqlalchemy import api as db_api
|
|
from glance.db.sqlalchemy import metadata as db_metadata
|
|
from glance.tests import utils as test_utils
|
|
|
|
|
|
class TestManageBase(test_utils.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestManageBase, self).setUp()
|
|
|
|
def clear_conf():
|
|
manage.CONF.reset()
|
|
manage.CONF.unregister_opt(manage.command_opt)
|
|
clear_conf()
|
|
self.addCleanup(clear_conf)
|
|
|
|
self.useFixture(fixtures.MonkeyPatch(
|
|
'oslo_log.log.setup', lambda product_name, version='test': None))
|
|
|
|
patcher = mock.patch('glance.db.sqlalchemy.api.get_engine')
|
|
patcher.start()
|
|
self.addCleanup(patcher.stop)
|
|
|
|
def _main_test_helper(self, argv, func_name=None, *exp_args, **exp_kwargs):
|
|
self.useFixture(fixtures.MonkeyPatch('sys.argv', argv))
|
|
manage.main()
|
|
func_name.assert_called_once_with(*exp_args, **exp_kwargs)
|
|
|
|
|
|
class TestLegacyManage(TestManageBase):
|
|
|
|
@mock.patch.object(manage.DbCommands, 'version')
|
|
def test_legacy_db_version(self, db_upgrade):
|
|
self._main_test_helper(['glance.cmd.manage', 'db_version'],
|
|
manage.DbCommands.version)
|
|
|
|
@mock.patch.object(manage.DbCommands, 'sync')
|
|
def test_legacy_db_sync(self, db_sync):
|
|
self._main_test_helper(['glance.cmd.manage', 'db_sync'],
|
|
manage.DbCommands.sync, None)
|
|
|
|
@mock.patch.object(manage.DbCommands, 'upgrade')
|
|
def test_legacy_db_upgrade(self, db_upgrade):
|
|
self._main_test_helper(['glance.cmd.manage', 'db_upgrade'],
|
|
manage.DbCommands.upgrade, None)
|
|
|
|
@mock.patch.object(manage.DbCommands, 'version_control')
|
|
def test_legacy_db_version_control(self, db_version_control):
|
|
self._main_test_helper(['glance.cmd.manage', 'db_version_control'],
|
|
manage.DbCommands.version_control, None)
|
|
|
|
@mock.patch.object(manage.DbCommands, 'sync')
|
|
def test_legacy_db_sync_version(self, db_sync):
|
|
self._main_test_helper(['glance.cmd.manage', 'db_sync', 'liberty'],
|
|
manage.DbCommands.sync, 'liberty')
|
|
|
|
@mock.patch.object(manage.DbCommands, 'upgrade')
|
|
def test_legacy_db_upgrade_version(self, db_upgrade):
|
|
self._main_test_helper(['glance.cmd.manage', 'db_upgrade', 'liberty'],
|
|
manage.DbCommands.upgrade, 'liberty')
|
|
|
|
def test_db_metadefs_unload(self):
|
|
db_metadata.db_unload_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db_unload_metadefs'],
|
|
db_metadata.db_unload_metadefs,
|
|
db_api.get_engine())
|
|
|
|
def test_db_metadefs_load(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db_load_metadefs'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
None, None, None, None)
|
|
|
|
def test_db_metadefs_load_with_specified_path(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db_load_metadefs',
|
|
'/mock/'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
'/mock/', None, None, None)
|
|
|
|
def test_db_metadefs_load_from_path_merge(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db_load_metadefs',
|
|
'/mock/', 'True'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
'/mock/', 'True', None, None)
|
|
|
|
def test_db_metadefs_load_from_merge_and_prefer_new(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db_load_metadefs',
|
|
'/mock/', 'True', 'True'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
'/mock/', 'True', 'True', None)
|
|
|
|
def test_db_metadefs_load_from_merge_and_prefer_new_and_overwrite(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db_load_metadefs',
|
|
'/mock/', 'True', 'True', 'True'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
'/mock/', 'True', 'True', 'True')
|
|
|
|
def test_db_metadefs_export(self):
|
|
db_metadata.db_export_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db_export_metadefs'],
|
|
db_metadata.db_export_metadefs,
|
|
db_api.get_engine(),
|
|
None)
|
|
|
|
def test_db_metadefs_export_with_specified_path(self):
|
|
db_metadata.db_export_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db_export_metadefs',
|
|
'/mock/'],
|
|
db_metadata.db_export_metadefs,
|
|
db_api.get_engine(),
|
|
'/mock/')
|
|
|
|
|
|
class TestManage(TestManageBase):
|
|
|
|
@mock.patch.object(manage.DbCommands, 'version')
|
|
def test_db_version(self, version):
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'version'],
|
|
manage.DbCommands.version)
|
|
|
|
@mock.patch.object(manage.DbCommands, 'sync')
|
|
def test_db_sync(self, sync):
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'sync'],
|
|
manage.DbCommands.sync)
|
|
|
|
@mock.patch.object(manage.DbCommands, 'upgrade')
|
|
def test_db_upgrade(self, upgrade):
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'upgrade'],
|
|
manage.DbCommands.upgrade)
|
|
|
|
@mock.patch.object(manage.DbCommands, 'version_control')
|
|
def test_db_version_control(self, version_control):
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'version_control'],
|
|
manage.DbCommands.version_control)
|
|
|
|
@mock.patch.object(manage.DbCommands, 'sync')
|
|
def test_db_sync_version(self, sync):
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'sync', 'liberty'],
|
|
manage.DbCommands.sync, 'liberty')
|
|
|
|
@mock.patch.object(manage.DbCommands, 'upgrade')
|
|
def test_db_upgrade_version(self, upgrade):
|
|
self._main_test_helper(['glance.cmd.manage', 'db',
|
|
'upgrade', 'liberty'],
|
|
manage.DbCommands.upgrade, 'liberty')
|
|
|
|
def test_db_metadefs_unload(self):
|
|
db_metadata.db_unload_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'unload_metadefs'],
|
|
db_metadata.db_unload_metadefs,
|
|
db_api.get_engine())
|
|
|
|
def test_db_metadefs_load(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'load_metadefs'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
None, False, False, False)
|
|
|
|
def test_db_metadefs_load_with_specified_path(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'load_metadefs',
|
|
'--path', '/mock/'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
'/mock/', False, False, False)
|
|
|
|
def test_db_metadefs_load_prefer_new_with_path(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'load_metadefs',
|
|
'--path', '/mock/', '--merge', '--prefer_new'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
'/mock/', True, True, False)
|
|
|
|
def test_db_metadefs_load_prefer_new(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'load_metadefs',
|
|
'--merge', '--prefer_new'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
None, True, True, False)
|
|
|
|
def test_db_metadefs_load_overwrite_existing(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'load_metadefs',
|
|
'--merge', '--overwrite'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
None, True, False, True)
|
|
|
|
def test_db_metadefs_load_prefer_new_and_overwrite_existing(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'load_metadefs',
|
|
'--merge', '--prefer_new', '--overwrite'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
None, True, True, True)
|
|
|
|
def test_db_metadefs_load_from_path_overwrite_existing(self):
|
|
db_metadata.db_load_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'load_metadefs',
|
|
'--path', '/mock/', '--merge', '--overwrite'],
|
|
db_metadata.db_load_metadefs,
|
|
db_api.get_engine(),
|
|
'/mock/', True, False, True)
|
|
|
|
def test_db_metadefs_export(self):
|
|
db_metadata.db_export_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'export_metadefs'],
|
|
db_metadata.db_export_metadefs,
|
|
db_api.get_engine(),
|
|
None)
|
|
|
|
def test_db_metadefs_export_with_specified_path(self):
|
|
db_metadata.db_export_metadefs = mock.Mock()
|
|
self._main_test_helper(['glance.cmd.manage', 'db', 'export_metadefs',
|
|
'--path', '/mock/'],
|
|
db_metadata.db_export_metadefs,
|
|
db_api.get_engine(),
|
|
'/mock/')
|