# Copyright 2012 OpenStack Foundation
# 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.

from oslo_log import log as logging

from tempest.api.compute import base
from tempest.common import utils
from tempest import config
from tempest.lib import decorators

CONF = config.CONF


LOG = logging.getLogger(__name__)


class ExtensionsTest(base.BaseV2ComputeTest):

    @decorators.idempotent_id('3bb27738-b759-4e0d-a5fa-37d7a6df07d1')
    def test_list_extensions(self):
        # List of all extensions
        if not CONF.compute_feature_enabled.api_extensions:
            raise self.skipException('There are not any extensions configured')
        extensions = self.extensions_client.list_extensions()['extensions']
        ext = CONF.compute_feature_enabled.api_extensions[0]

        # Log extensions list
        extension_list = [x['alias'] for x in extensions]
        LOG.debug("Nova extensions: %s", ','.join(extension_list))

        if ext == 'all':
            self.assertIn('Hosts', map(lambda x: x['name'], extensions))
        elif ext:
            self.assertIn(ext, extension_list)
        else:
            raise self.skipException('There are not any extensions configured')

    @decorators.idempotent_id('05762f39-bdfa-4cdb-9b46-b78f8e78e2fd')
    @utils.requires_ext(extension='os-consoles', service='compute')
    def test_get_extension(self):
        # get the specified extensions
        extension = self.extensions_client.show_extension('os-consoles')
        self.assertEqual('os-consoles', extension['extension']['alias'])
