Fix some small things in ApexLake tests

[yardstick.git] / yardstick / vTC / apexlake / tests / heat_manager_test.py

diff --git a/yardstick/vTC/apexlake/tests/heat_manager_test.py b/yardstick/vTC/apexlake/tests/heat_manager_test.py
index f89835c..a2798a7 100644 (file)
--- a/yardstick/vTC/apexlake/tests/heat_manager_test.py
+++ b/yardstick/vTC/apexlake/tests/heat_manager_test.py
@@ -144,11 +144,13 @@ class TestHeatManager(unittest.TestCase):
     def test_delete_stack_for_success_2(self):
         self.assertTrue(self.heat_manager.delete_stack('stack_1'))
 
+    @mock.patch('experimental_framework.common.LOG')
     @mock.patch('heatclient.common.template_utils.get_template_contents')
     @mock.patch('heatclient.client.Client')
     # @mock.patch('heatclient.client.Client', side_effect=DummyHeatClient)
     def test_create_stack_for_success(self, mock_stack_create,
-                                      mock_get_template_contents):
+                                      mock_get_template_contents,
+                                      mock_log):
         return_value = ({'template': 'template'}, 'template')
         mock_get_template_contents.return_value = return_value
         self.heat_manager.create_stack('template', 'stack_n', 'parameters')
@@ -174,11 +176,18 @@ class TestHeatManager_2(unittest.TestCase):
         self.assertFalse(self.heat_manager.delete_stack('stack_1'))
 
 
+class ServiceCatalog():
+    def url_for(self, service_type):
+        return 'http://heat_url'
+
+
 class KeystoneMock(object):
     @property
     def auth_token(self):
         return 'token'
 
+    service_catalog = ServiceCatalog()
+
 
 class TestHeatInit(unittest.TestCase):
     def setUp(self):