# Copyright (c) 2014 Mirantis, Inc.
#
# 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 os

from tempest import clients
from tempest import config
from tempest_lib.common import rest_client
import testtools

from mistralclient.api import base
from mistralclient.api.v1 import client as mclient

CONF = config.CONF


class ClientAuth(rest_client.RestClient):
    def __init__(self, auth_provider, url):
        super(ClientAuth, self).__init__(
            auth_provider=auth_provider,
            service='workflow',
            region=CONF.identity.region)

        self.mistral_client = mclient.Client(
            auth_token=self.auth_provider.get_token(),
            mistral_url=url)


class MistralBase(testtools.TestCase):

    _mistral_url = None

    @classmethod
    def setUpClass(cls):
        super(MistralBase, cls).setUpClass()

        if 'WITHOUT_AUTH' in os.environ:
            cls.mistral_client = mclient.Client(
                mistral_url=cls._mistral_url)
        else:
            mgr = clients.Manager()
            cls.mistral_client = ClientAuth(
                mgr.auth_provider, cls._mistral_url).mistral_client

        cls.definition = open(os.path.relpath(
            'functionaltests/resources/v1/wb_v1.yaml',
            os.getcwd()), 'rb').read()

        cls.wb = cls.mistral_client.workbooks.create(
            "wb", "Description", ["tags"])

    @classmethod
    def tearDownClass(cls):
        super(MistralBase, cls).tearDownClass()

        for wb in cls.mistral_client.workbooks.list():
            cls.mistral_client.workbooks.delete(wb.name)

    def tearDown(self):
        super(MistralBase, self).tearDown()
        for ex in self.mistral_client.executions.list(None):
            # TODO(akuznetsova): remove try/except after
            # https://bugs.launchpad.net/mistral/+bug/1353306
            # will be fixed
            """ try/except construction was added because of problem
                with concurrent transactions in sqlite and appropriate
                error: APIException: (OperationalError) database is locked.
                This isn't a tests problem, so they are considered passed.
            """
            try:
                self.mistral_client.executions.delete(None, ex.id)
            except base.APIException:
                pass

    def create_execution(self):
        self.mistral_client.workbooks.upload_definition("wb", self.definition)
        execution = self.mistral_client.executions.create("wb", "hello")
        return execution

    def assert_item_in_list(self, items, **props):
        def _matches(item, **props):
            for prop_name, prop_val in props.iteritems():
                v = item[prop_name] if isinstance(item, dict) else getattr(
                    item, prop_name)

                if v != prop_val:
                    return False

            return True

        filtered_items = filter(lambda item: _matches(item, **props), items)

        return len(filtered_items) != 0