Create fixture for creating heat client.

Change-Id: Iaae648511c157f4caefe021820142d11d47645e7

tobiko/openstack/heat/__init__.py

@@ -0,0 +1,19 @@
+# Copyright 2019 Red Hat
+#
+#    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.
+from __future__ import absolute_import
+
+from tobiko.openstack.heat import client
+
+get_heat_client = client.get_heat_client
+HeatClientFixture = client.HeatClientFixture

tobiko/openstack/heat/client.py

@@ -0,0 +1,38 @@
+# Copyright 2019 Red Hat
+#
+#    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.
+from __future__ import absolute_import
+
+from heatclient import client as heatclient
+
+from tobiko.openstack import client as _client
+
+
+class HeatClientFixture(_client.OpenstackClientFixture):
+
+    def init_client(self, session):
+        return heatclient.Client(
+            '1', session=session, endpoint_type='public',
+            service_type='orchestration')
+
+
+CLIENTS = _client.OpenstackClientManager(init_client=HeatClientFixture)
+
+
+def get_heat_client(session=None, shared=True, init_client=None,
+                    manager=None):
+    manager = manager or CLIENTS
+    client = manager.get_client(session=session, shared=shared,
+                                init_client=init_client)
+    client.setUp()
+    return client.client

tobiko/tests/openstack/heat/test_client.py

@@ -0,0 +1,46 @@
+# Copyright 2019 Red Hat
+#
+#    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.
+from __future__ import absolute_import
+
+from tobiko.openstack import keystone
+from tobiko.openstack import heat
+from tobiko.tests.openstack import base
+from tobiko.tests.openstack import test_client
+
+
+class HeatClientFixtureTest(test_client.OpenstackClientFixtureTest):
+
+    def create_client(self, session=None):
+        return heat.HeatClientFixture(session=session)
+
+
+class GetHeatClientTest(base.OpenstackTest):
+
+    def test_get_heat_client(self, session=None, shared=True):
+        client1 = heat.get_heat_client(session=session, shared=shared)
+        client2 = heat.get_heat_client(session=session, shared=shared)
+        if shared:
+            self.assertIs(client1, client2)
+            self.assertIsNotNone(client1)
+        else:
+            self.assertIsNot(client1, client2)
+            self.assertIsNotNone(client1)
+            self.assertIsNotNone(client2)
+
+    def test_get_heat_client_with_not_shared(self):
+        self.test_get_heat_client(shared=False)
+
+    def test_get_heat_client_with_session(self):
+        session = keystone.get_keystone_session()
+        self.test_get_heat_client(session=session)