¡@

Home 

OpenStack Study: test_manage.py

OpenStack Index

**** CubicPower OpenStack Study ****

# 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

import testtools

import glance

from glance.cmd import manage

from glance.db import migration as db_migration

from glance.db.sqlalchemy import api as db_api

from glance.openstack.common.db.sqlalchemy import migration

**** CubicPower OpenStack Study ****

class TestManageBase(testtools.TestCase):

**** CubicPower OpenStack Study ****

    def setUp(self):

        super(TestManageBase, self).setUp()

        def clear_conf():

            manage.CONF.reset()

            manage.CONF.unregister_opt(manage.command_opt)

            manage.CONF.db_enforce_mysql_charset = True

        self.addCleanup(clear_conf)

        self.patcher = mock.patch('glance.db.sqlalchemy.api.get_engine')

        self.patcher.start()

        self.addCleanup(self.patcher.stop)

**** CubicPower OpenStack Study ****

        def clear_conf():

            manage.CONF.reset()

            manage.CONF.unregister_opt(manage.command_opt)

            manage.CONF.db_enforce_mysql_charset = True

        self.addCleanup(clear_conf)

        self.patcher = mock.patch('glance.db.sqlalchemy.api.get_engine')

        self.patcher.start()

        self.addCleanup(self.patcher.stop)

**** CubicPower OpenStack Study ****

    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)

**** CubicPower OpenStack Study ****

class TestLegacyManage(TestManageBase):

**** CubicPower OpenStack Study ****

    def test_legacy_db_version(self):

        migration.db_version = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db_version'],

                               glance.openstack.common.db.sqlalchemy.

                               migration.db_version,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, 0)

**** CubicPower OpenStack Study ****

    def test_legacy_db_sync(self):

        migration.db_sync = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db_sync'],

                               glance.openstack.common.db.sqlalchemy.

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, None,

                               sanity_check=True)

**** CubicPower OpenStack Study ****

    def test_legacy_db_upgrade(self):

        migration.db_sync = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db_upgrade'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, None,

                               sanity_check=True)

**** CubicPower OpenStack Study ****

    def test_legacy_db_version_control(self):

        migration.db_version_control = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db_version_control'],

                               migration.db_version_control,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, None)

**** CubicPower OpenStack Study ****

    def test_legacy_db_sync_version(self):

        migration.db_sync = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db_sync', '20'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, '20',

                               sanity_check=True)

**** CubicPower OpenStack Study ****

    def test_legacy_db_upgrade_version(self):

        migration.db_sync = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db_upgrade', '20'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, '20',

                               sanity_check=True)

**** CubicPower OpenStack Study ****

    def test_legacy_db_downgrade_version(self):

        migration.db_sync = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db_downgrade', '20'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, '20',

                               sanity_check=True)

**** CubicPower OpenStack Study ****

    def test_legacy_db_sync_version_without_sanity_check(self):

        migration.db_sync = mock.Mock()

        manage.CONF.db_enforce_mysql_charset = False

        self._main_test_helper(['glance.cmd.manage', 'db_sync', '20'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, '20',

                               sanity_check=False)

**** CubicPower OpenStack Study ****

    def test_legacy_db_upgrade_version_without_sanity_check(self):

        migration.db_sync = mock.Mock()

        manage.CONF.db_enforce_mysql_charset = False

        self._main_test_helper(['glance.cmd.manage', 'db_upgrade', '40'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, '40',

                               sanity_check=False)

**** CubicPower OpenStack Study ****

    def test_legacy_db_downgrade_version_without_sanity_check(self):

        migration.db_sync = mock.Mock()

        manage.CONF.db_enforce_mysql_charset = False

        self._main_test_helper(['glance.cmd.manage', 'db_downgrade', '20'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, '20',

                               sanity_check=False)

**** CubicPower OpenStack Study ****

class TestManage(TestManageBase):

**** CubicPower OpenStack Study ****

    def test_db_version(self):

        migration.db_version = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db', 'version'],

                               migration.db_version,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, 0)

**** CubicPower OpenStack Study ****

    def test_db_sync(self):

        migration.db_sync = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db', 'sync'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, None,

                               sanity_check=True)

**** CubicPower OpenStack Study ****

    def test_db_upgrade(self):

        migration.db_sync = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db', 'upgrade'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, None,

                               sanity_check=True)

**** CubicPower OpenStack Study ****

    def test_db_version_control(self):

        migration.db_version_control = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db', 'version_control'],

                               migration.db_version_control,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, None)

**** CubicPower OpenStack Study ****

    def test_db_sync_version(self):

        migration.db_sync = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db', 'sync', '20'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, '20',

                               sanity_check=True)

**** CubicPower OpenStack Study ****

    def test_db_upgrade_version(self):

        migration.db_sync = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db', 'upgrade', '20'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, '20',

                               sanity_check=True)

**** CubicPower OpenStack Study ****

    def test_db_downgrade_version(self):

        migration.db_sync = mock.Mock()

        self._main_test_helper(['glance.cmd.manage', 'db', 'downgrade', '20'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, '20',

                               sanity_check=True)

**** CubicPower OpenStack Study ****

    def test_db_sync_version_without_sanity_check(self):

        migration.db_sync = mock.Mock()

        manage.CONF.db_enforce_mysql_charset = False

        self._main_test_helper(['glance.cmd.manage', 'db', 'sync', '20'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, u'20',

                               sanity_check=False)

**** CubicPower OpenStack Study ****

    def test_db_upgrade_version_without_sanity_check(self):

        migration.db_sync = mock.Mock()

        manage.CONF.db_enforce_mysql_charset = False

        self._main_test_helper(['glance.cmd.manage', 'db', 'upgrade', '40'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, '40',

                               sanity_check=False)

**** CubicPower OpenStack Study ****

    def test_db_downgrade_version_without_sanity_check(self):

        migration.db_sync = mock.Mock()

        manage.CONF.db_enforce_mysql_charset = False

        self._main_test_helper(['glance.cmd.manage', 'db', 'downgrade', '20'],

                               migration.db_sync,

                               db_api.get_engine(),

                               db_migration.MIGRATE_REPO_PATH, '20',

                               sanity_check=False)