cleanup how message factory is used

[testsuite/python-testing-utils.git] / robotframework-onap / tests / runner.py

diff --git a/robotframework-onap/tests/runner.py b/robotframework-onap/tests/runner.py
index b58215a..3422dfe 100644 (file)
--- a/robotframework-onap/tests/runner.py
+++ b/robotframework-onap/tests/runner.py
@@ -1,16 +1,20 @@
-import unittest
+from unittest import TextTestRunner
+from unittest import TestLoader
+from unittest import TestSuite
 
 # import your test modules
-from tests.vcpeutils.SoUtils_test import *
+from tests.vcpeutils.SoUtils_test import SoUtilsTest
+from tests.ONAPLibrary.ProtobufKeywordsTest import ProtobufKeywordsTest
 
 
 # initialize the test suite
-loader = unittest.TestLoader()
-suite = unittest.TestSuite()
+loader = TestLoader()
+suite = TestSuite()
 
 # add tests to the test suite
+suite.addTests(loader.loadTestsFromTestCase(ProtobufKeywordsTest))
 suite.addTests(loader.loadTestsFromTestCase(SoUtilsTest))
 
 # initialize a runner, pass it your suite and run it
-runner = unittest.TextTestRunner(verbosity=3)
+runner = TextTestRunner(verbosity=3)
 result = runner.run(suite)