Add setup and cleanup for maintenance test

[doctor.git] / doctor_tests / os_clients.py

diff --git a/doctor_tests/os_clients.py b/doctor_tests/os_clients.py
index 640281d..7ab4e9b 100644 (file)
--- 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\r
 from congressclient.v1 import client as congressclient\r
 import glanceclient.client as glanceclient\r
 import aodhclient.client as aodhclient\r
 from congressclient.v1 import client as congressclient\r
 import glanceclient.client as glanceclient\r

+import heatclient.client as heatclient\r

 from keystoneclient import client as ks_client\r
 from neutronclient.v2_0 import client as neutronclient\r
 import novaclient.client as novaclient\r
 from keystoneclient import client as ks_client\r
 from neutronclient.v2_0 import client as neutronclient\r
 import novaclient.client as novaclient\r


@@ -23,6 +24,7 @@ OPTS = [
     cfg.StrOpt('aodh_version', default='2', help='aodh version'),\r
     cfg.StrOpt('vitrage_version', default='1', help='vitrage version'),\r
     cfg.StrOpt('keystone_version', default='v3', help='keystone version'),\r
     cfg.StrOpt('aodh_version', default='2', help='aodh version'),\r
     cfg.StrOpt('vitrage_version', default='1', help='vitrage version'),\r
     cfg.StrOpt('keystone_version', default='v3', help='keystone version'),\r

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

 ]\r
 \r
 \r
 ]\r
 \r
 \r


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

+def heat_client(version, session):\r
+    return heatclient.Client(version=version,\r
+                             session=session)\r
+\r
+\r

 def keystone_client(version, session):\r
     return ks_client.Client(version=version,\r
                             session=session)\r
 def keystone_client(version, session):\r
     return ks_client.Client(version=version,\r
                             session=session)\r