Merge "Added test cases for running the Python Tests included with SNAPS."

1 files changed, 56 insertions, 0 deletions

diff --git a/functest/core/pytest_suite_runner.py b/functest/core/pytest_suite_runner.py
new file mode 100644
index 00000000..ba372c3b
--- /dev/null
+++ b/functest/core/pytest_suite_runner.py
@@ -0,0 +1,56 @@
+#!/usr/bin/python
+#
+# Copyright (c) 2015 All rights reserved
+# This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+
+from functest.core import TestCasesBase
+import unittest
+import time
+
+
+class PyTestSuiteRunner(TestCasesBase.TestCasesBase):
+    """
+    This superclass is designed to execute pre-configured unittest.TestSuite()
+    objects
+    """
+    def __init__(self):
+        super(PyTestSuiteRunner, self).__init__()
+        self.suite = None
+
+    def run(self, **kwargs):
+        """
+        Starts test execution from the functest framework
+        """
+        start_time = time.time()
+        result = unittest.TextTestRunner(verbosity=2).run(self.suite)
+        end_time = time.time()
+
+        if result.errors:
+            self.logger.error('Number of errors in test suite - ' +
+                              str(len(result.errors)))
+            for test, message in result.errors:
+                self.logger.error(str(test) + " ERROR with " + message)
+
+        if result.failures:
+            self.logger.error('Number of failures in test suite - ' +
+                              str(len(result.failures)))
+            for test, message in result.failures:
+                self.logger.error(str(test) + " FAILED with " + message)
+
+        if (result.errors and len(result.errors) > 0) \
+                or (result.failures and len(result.failures) > 0):
+            self.logger.info("%s FAILED" % self.case_name)
+            self.criteria = 'FAIL'
+            exit_code = TestCasesBase.TestCasesBase.EX_RUN_ERROR
+        else:
+            self.logger.info("%s OK" % self.case_name)
+            exit_code = TestCasesBase.TestCasesBase.EX_OK
+
+        self.details = {'timestart': start_time,
+                        'duration': round(end_time - start_time, 1),
+                        'status': self.criteria}
+        return exit_code