Improvement of debug messages to the user

[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 9191a17..0fe8554 100644 (file)
--- a/yardstick/vTC/apexlake/tests/heat_manager_test.py
+++ b/yardstick/vTC/apexlake/tests/heat_manager_test.py
@@ -12,13 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__author__ = 'gpetralx'
-
-
 import unittest
+import logging
+import experimental_framework.common as common
+
 from experimental_framework import heat_manager
 import mock
 
+__author__ = 'gpetralx'
+
 
 def get_mock_heat(version, *args, **kwargs):
     return MockHeat()
@@ -173,14 +175,22 @@ class TestHeatManager_2(unittest.TestCase):
         pass
 
     def test_delete_stack_for_success_2(self):
+        common.LOG = logging.getLogger()
         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):