# Copyright 2014 Mirantis Inc.
# 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 tempest import config
from tempest.lib.common.utils import data_utils
from tempest.lib import decorators
from testtools import testcase as tc

from manila_tempest_tests.tests.api import base


CONF = config.CONF


class ShareMultiBackendTest(base.BaseSharesAdminTest):

    @staticmethod
    def _share_protocol(protocol):
        protocols_list = protocol.lower().split('_')
        allowed_protocols = [
            i for i in CONF.share.enable_protocols
            if i.lower() in protocols_list
        ]
        return allowed_protocols

    @classmethod
    def resource_setup(cls):
        super(ShareMultiBackendTest, cls).resource_setup()
        if not CONF.share.multi_backend:
            raise cls.skipException("Manila multi-backend tests are disabled.")
        elif len(CONF.share.backend_names) < 2:
            raise cls.skipException("For running multi-backend tests required"
                                    " two names in config. Skipping.")
        elif any(not name for name in CONF.share.backend_names):
            raise cls.skipException("Share backend names can not be empty. "
                                    "Skipping.")
        cls.sts = []
        cls.shares = []
        share_data_list = []

        pools = cls.shares_v2_client.list_pools(detail=True)['pools']
        backends_protocols = {
            pool['capabilities']['share_backend_name']: pool[
                'capabilities']['storage_protocol'] for pool in pools
        }
        # Create share types
        for backend in CONF.share.backend_names:
            share_protocol = cls._share_protocol(backends_protocols[backend])
            if not share_protocol:
                continue
            st_name = data_utils.rand_name(
                cls.__name__ + "-share-type-%s" % backend)
            extra_specs = {
                "share_backend_name": backend,
                "driver_handles_share_servers":
                    CONF.share.multitenancy_enabled,
            }
            st = cls.create_share_type(name=st_name, extra_specs=extra_specs)
            cls.sts.append(st)
            st_id = st["id"]
            share_data_list.append({"kwargs": {
                "share_type_id": st_id,
                "share_protocol": share_protocol[0]}})

        if not share_data_list:
            raise cls.skipException("Enabled protocols not supported by any "
                                    "of the enabled backends.")

        # Create shares using precreated share types
        cls.shares = cls.create_shares(share_data_list)

    @decorators.idempotent_id('888add9d-4ad0-4d88-a62f-af7ee1d9f9ab')
    @tc.attr(base.TAG_POSITIVE, base.TAG_API_WITH_BACKEND)
    def test_share_backend_name_reporting(self):
        # Share's 'host' should be like "hostname@backend_name"
        for share in self.shares:
            get = self.shares_client.get_share(share['id'])['share']
            self.assertEqual(2, len(get["host"].split("@")))

    @decorators.idempotent_id('691fbcef-6d8d-4ad9-b493-501bbb3dcf3c')
    @tc.attr(base.TAG_POSITIVE, base.TAG_API_WITH_BACKEND)
    def test_share_share_type(self):
        # Share type should be the same as provided with share creation
        for share, share_type in zip(self.shares, self.sts):
            share_details = self.shares_v2_client.get_share(
                share['id'], version="2.5")['share']
            self.assertEqual(share_type["name"], share_details["share_type"])

    @decorators.idempotent_id('f25e0cb0-d656-4f16-a761-ec23992cd9e7')
    @tc.attr(base.TAG_POSITIVE, base.TAG_API_WITH_BACKEND)
    def test_share_share_type_v_2_6(self):
        # Share type should be the same as provided with share creation
        for share, share_type in zip(self.shares, self.sts):
            share_details = self.shares_v2_client.get_share(
                share['id'], version="2.6")['share']
            self.assertEqual(share_type["id"], share_details["share_type"])
            self.assertEqual(
                share_type["name"], share_details["share_type_name"])

    @decorators.idempotent_id('bfa0c056-0a15-40e1-bdff-f1e10b95736c')
    @tc.attr(base.TAG_POSITIVE, base.TAG_API_WITH_BACKEND)
    def test_share_backend_name_distinction(self):
        # Different share backends should have different host records
        if CONF.share.backend_names[0] == CONF.share.backend_names[1]:
            raise self.skipException("Share backends "
                                     "configured with same name. Skipping.")
        get1 = self.shares_client.get_share(self.shares[0]['id'])['share']
        get2 = self.shares_client.get_share(self.shares[1]['id'])['share']
        self.assertNotEqual(get1["host"], get2["host"])