From: Cédric Ollivier <cedric.ollivier@orange.com>
Date: Thu, 18 May 2017 14:40:13 +0000 (+0200)
Subject: Use logger instead of printing in console
X-Git-Tag: opnfv-5.0.RC1~387^2~2
X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=commitdiff_plain;h=510a5cfda3b375b432e0ed17a61e34de969b520c;p=functest.git

Use logger instead of printing in console

It modifies pytest_suite_runner.py to dump unit test results in
functest.log via logging.

Change-Id: Ic4e7567c5e52e00ad0e485ccb070029b135041b2
Signed-off-by: Cédric Ollivier <cedric.ollivier@orange.com>
---

diff --git a/functest/core/pytest_suite_runner.py b/functest/core/pytest_suite_runner.py
index 0dd860d26..a6e476600 100644
--- a/functest/core/pytest_suite_runner.py
+++ b/functest/core/pytest_suite_runner.py
@@ -8,8 +8,10 @@
 # pylint: disable=missing-docstring
 
 import logging
-import unittest
 import time
+import unittest
+
+import six
 
 from functest.core import testcase
 
@@ -19,6 +21,7 @@ class PyTestSuiteRunner(testcase.TestCase):
     This superclass is designed to execute pre-configured unittest.TestSuite()
     objects
     """
+
     def __init__(self, **kwargs):
         super(PyTestSuiteRunner, self).__init__(**kwargs)
         self.suite = None
@@ -30,11 +33,18 @@ class PyTestSuiteRunner(testcase.TestCase):
         """
         try:
             name = kwargs["name"]
-            self.suite = unittest.TestLoader().loadTestsFromName(name)
+            try:
+                self.suite = unittest.TestLoader().loadTestsFromName(name)
+            except ImportError:
+                self.logger.error("Can not import %s", name)
+                return testcase.TestCase.EX_RUN_ERROR
         except KeyError:
             pass
         self.start_time = time.time()
-        result = unittest.TextTestRunner(verbosity=2).run(self.suite)
+        stream = six.StringIO()
+        result = unittest.TextTestRunner(
+            stream=stream, verbosity=2).run(self.suite)
+        self.logger.debug("\n\n%s", stream.getvalue())
         self.stop_time = time.time()
 
         if result.errors: