Merge "Unit Test for Ansible Module"

1 files changed, 31 insertions, 0 deletions

diff --git a/tests/unit/cli/ansible_test.py b/tests/unit/cli/ansible_test.py
new file mode 100644
index 00000000..02941333
--- /dev/null
+++ b/tests/unit/cli/ansible_test.py
@@ -0,0 +1,31 @@
+##############################################################################
+# Copyright (c) 2016 ZTE Corp and others.
+#
+# 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
+##############################################################################
+
+import pytest
+from click.testing import CliRunner
+from qtip.cli.entry import cli
+
+
+class TestClass(object):
+
+    @pytest.fixture()
+    def runner(self):
+        return CliRunner()
+
+    def test_prepare(self, runner):
+        result = runner.invoke(cli, ['ansible', 'prepare'])
+        assert result.output == "prepare ansible env\n"
+
+    def test_show(self, runner):
+        result = runner.invoke(cli, ['ansible', 'show'])
+        assert result.output == "show ansible configuration\n"
+
+    def test_status(self, runner):
+        result = runner.invoke(cli, ['ansible', 'status'])
+        assert result.output == "check connectivity\n"