37dbbb355c
There is a typo in SetUpClass in base.py, this typo hinders running integration tests for client. Change-Id: I50fba55025729737f4850c4687dac9093dcf235f
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
import os
|
|
import testtools
|
|
|
|
from tempest import clients
|
|
from tempest.common import rest_client
|
|
|
|
from mistralclient.api import client as mclient
|
|
|
|
|
|
class ClientAuth(rest_client.RestClient):
|
|
def __init__(self, auth_provider):
|
|
super(ClientAuth, self).__init__(auth_provider)
|
|
|
|
self.mistral_client = mclient.Client(
|
|
auth_token=self.auth_provider.get_token())
|
|
|
|
|
|
class MistralBase(testtools.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(MistralBase, cls).setUpClass()
|
|
|
|
mgr = clients.Manager()
|
|
cls.mistral_client = ClientAuth(mgr.auth_provider).mistral_client
|
|
|
|
__location = os.path.realpath(os.path.join(os.getcwd(),
|
|
os.path.dirname(__file__)))
|
|
|
|
cls.definition = open(os.path.join(
|
|
__location, 'hello.yaml'), '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):
|
|
self.mistral_client.executions.delete(None, ex.id)
|
|
|
|
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
|