# 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 ExtensionsTestJSON(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 = map(lambda x: x['alias'], 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'])
