Add setup and cleanup for maintenance test

JIRA: DOCTOR-106

Change-Id: Idb95e1391e8f9094f68e2c7bc79fd2d7e01af9de
Signed-off-by: Tomi Juvonen <tomi.juvonen@nokia.com>

1 files changed, 7 insertions, 0 deletions

diff --git a/doctor_tests/os_clients.py b/doctor_tests/os_clients.py
index 640281df..7ab4e9b4 100644
--- a/doctor_tests/os_clients.py
+++ b/doctor_tests/os_clients.py
@@ -11,6 +11,7 @@ from oslo_config import cfg
 import aodhclient.client as aodhclient

 from congressclient.v1 import client as congressclient

 import glanceclient.client as glanceclient

+import heatclient.client as heatclient

 from keystoneclient import client as ks_client

 from neutronclient.v2_0 import client as neutronclient

 import novaclient.client as novaclient

@@ -23,6 +24,7 @@ OPTS = [
     cfg.StrOpt('aodh_version', default='2', help='aodh version'),

     cfg.StrOpt('vitrage_version', default='1', help='vitrage version'),

     cfg.StrOpt('keystone_version', default='v3', help='keystone version'),

+    cfg.StrOpt('heat_version', default='1', help='heat version'),

 ]

 

 

@@ -31,6 +33,11 @@ def glance_client(version, session):
                                session=session)

 

 

+def heat_client(version, session):

+    return heatclient.Client(version=version,

+                             session=session)

+

+

 def keystone_client(version, session):

     return ks_client.Client(version=version,

                             session=session)