105 lines
3.4 KiB
Python
105 lines
3.4 KiB
Python
# 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
|