Modify TestCase constructor attributes

[functest.git] / functest / core / pytest_suite_runner.py

diff --git a/functest/core/pytest_suite_runner.py b/functest/core/pytest_suite_runner.py
index c168d7d..775f0a6 100644 (file)
--- a/functest/core/pytest_suite_runner.py
+++ b/functest/core/pytest_suite_runner.py
@@ -1,5 +1,3 @@
-#!/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
@@ -7,18 +5,18 @@
 #
 # http://www.apache.org/licenses/LICENSE-2.0
 
-import testcase_base as base
+import testcase as base
 import unittest
 import time
 
 
-class PyTestSuiteRunner(base.TestcaseBase):
+class PyTestSuiteRunner(base.TestCase):
     """
     This superclass is designed to execute pre-configured unittest.TestSuite()
     objects
     """
-    def __init__(self):
-        super(PyTestSuiteRunner, self).__init__()
+    def __init__(self, **kwargs):
+        super(PyTestSuiteRunner, self).__init__(**kwargs)
         self.suite = None
 
     def run(self, **kwargs):
@@ -44,9 +42,9 @@ class PyTestSuiteRunner(base.TestcaseBase):
         # a result can be PASS or FAIL
         # But in this case it means that the Execution was OK
         # we shall distinguish Execution Error from FAIL results
-        # TestcaseBase.EX_RUN_ERROR means that the test case was not run
+        # TestCase.EX_RUN_ERROR means that the test case was not run
         # not that it was run but the result was FAIL
-        exit_code = base.TestcaseBase.EX_OK
+        exit_code = base.TestCase.EX_OK
         if ((result.errors and len(result.errors) > 0)
                 or (result.failures and len(result.failures) > 0)):
             self.logger.info("%s FAILED" % self.case_name)