1 files changed, 89 insertions, 0 deletions
diff --git a/vnftest/tests/unit/core/test_commands.py b/vnftest/tests/unit/core/test_commands.py
new file mode 100755
index 0000000..35660bd
--- /dev/null
+++ b/vnftest/tests/unit/core/test_commands.py
@@ -0,0 +1,89 @@
+##############################################################################
+# Copyright 2018 EuropeanSoftwareMarketingLtd.
+# ===================================================================
+#  Licensed under the ApacheLicense, Version2.0 (the"License");
+#  you may not use this file except in compliance with the License.
+#  You may obtain a copy of the License at
+#  http://www.apache.org/licenses/LICENSE-2.0
+#
+# software distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and limitations under
+# the License
+##############################################################################
+# vnftest comment: this is a modified copy of
+# yardstick/tests/functional/test_cli_runner.py
+
+import unittest
+
+from vnftest.cmd.commands.runner import RunnerCommands
+from vnftest.cmd.commands.step import StepCommands
+from vnftest.core import Param
+from vnftest.core.runner import Runners
+from vnftest.core.step import Steps
+from vnftest.runners.iteration import IterationRunner
+from vnftest.runners.duration import DurationRunner
+from vnftest.onap.onap_api_call import OnapApiCall
+from cStringIO import StringIO
+import sys
+
+
+class Capture(list):
+
+    def __enter__(self):
+        self._stdout = sys.stdout
+        sys.stdout = self._stringio = StringIO()
+        return self
+
+    def __exit__(self, *args):
+        self.extend(self._stringio.getvalue().splitlines())
+        del self._stringio
+        sys.stdout = self._stdout
+
+
+class CommandsTestCase(unittest.TestCase):
+
+    def setUp(self):
+        super(CommandsTestCase, self).setUp()
+
+    def test_runner_list(self):
+        runner_cmd = RunnerCommands()
+        with Capture() as output:
+            runner_cmd.do_list(None)
+        self.assert_text_in_lines(output, ["Duration", "Iteration"])
+
+    def test_step_list(self):
+        step_cmd = StepCommands()
+        with Capture() as output:
+            step_cmd.do_list(None)
+        self.assert_text_in_lines(output, ["OnapApiCall"])
+
+    def test_runner_show_Duration(self):
+        param = Param({})
+        setattr(param, 'type', ['Duration'])
+        with Capture() as output:
+            Runners().show(param)
+        self.assert_text_in_lines(output, ["duration - amount of time"])
+
+    def test_runner_show_Iteration(self):
+        param = Param({})
+        setattr(param, 'type', ['Iteration'])
+        with Capture() as output:
+            Runners().show(param)
+        self.assert_text_in_lines(output, ["iterations - amount of times"])
+
+    def test_step_show_OnapApiCall(self):
+        param = Param({})
+        setattr(param, 'type', ['OnapApiCall'])
+        with Capture() as output:
+            Steps().show(param)
+        self.assert_text_in_lines(output, ["Call ONAP API"])
+
+    def assert_text_in_lines(self, lines, texts):
+        for text in texts:
+            found = False
+            for line in lines:
+                if text in line:
+                    found = True
+                    break
+            self.assertTrue(found, "Not Found: " + text)