Add unit tests for check_pkg_version_format and check_release_format

roles/lib_utils/test/sanity_check_test.py

@@ -0,0 +1,82 @@
+import os
+import pytest
+import sys
+
+from ansible.playbook.play_context import PlayContext
+from ansible.template import Templar
+from ansible import errors
+
+sys.path.insert(1, os.path.join(os.path.dirname(__file__), os.pardir, "action_plugins"))
+from sanity_checks import ActionModule  # noqa: E402
+
+
+@pytest.mark.parametrize('hostvars, host, varname, result', [
+    ({"example.com": {"param": 3.11}}, "example.com", "param", 3.11),
+    ({"example.com": {"param": 3.11}}, "example.com", "another_param", None)
+])
+def test_template_var(hostvars, host, varname, result):
+    task = FakeTask('sanity_checks', {'checks': []})
+    plugin = ActionModule(task, None, PlayContext(), None, Templar(None, None, None), None)
+    check = plugin.template_var(hostvars, host, varname)
+    assert check == result
+
+
+@pytest.mark.parametrize('hostvars, host, result', [
+    ({"example.com": {"openshift_pkg_version": "-3.6.0"}}, "example.com", None),
+    ({"example.com": {"openshift_pkg_version": "-3.7.0-0.126.0.git.0.9351aae.el7"}}, "example.com", None),
+    ({"example.com": {"openshift_pkg_version": "-3.9.0-2.fc28"}}, "example.com", None),
+    ({"example.com": {"openshift_pkg_version": "-3.11*"}}, "example.com", None),
+])
+def test_valid_check_pkg_version_format(hostvars, host, result):
+    task = FakeTask('sanity_checks', {'checks': []})
+    plugin = ActionModule(task, None, PlayContext(), None, Templar(None, None, None), None)
+    check = plugin.check_pkg_version_format(hostvars, host)
+    assert check == result
+
+
+@pytest.mark.parametrize('hostvars, host, result', [
+    ({"example.com": {"openshift_pkg_version": "3.11.0"}}, "example.com", None),
+    ({"example.com": {"openshift_pkg_version": "-3.11"}}, "example.com", None),
+    ({"example.com": {"openshift_pkg_version": "v3.11.0"}}, "example.com", None),
+])
+def test_invalid_check_pkg_version_format(hostvars, host, result):
+    with pytest.raises(errors.AnsibleModuleError):
+        task = FakeTask('sanity_checks', {'checks': []})
+        plugin = ActionModule(task, None, PlayContext(), None, Templar(None, None, None), None)
+        plugin.check_pkg_version_format(hostvars, host)
+
+
+@pytest.mark.parametrize('hostvars, host, result', [
+    ({"example.com": {"openshift_release": "v3"}}, "example.com", None),
+    ({"example.com": {"openshift_release": "v3.11"}}, "example.com", None),
+    ({"example.com": {"openshift_release": "v3.11.0"}}, "example.com", None),
+    ({"example.com": {"openshift_release": "3.11"}}, "example.com", None),
+])
+def test_valid_check_release_format(hostvars, host, result):
+    task = FakeTask('sanity_checks', {'checks': []})
+    plugin = ActionModule(task, None, PlayContext(), None, Templar(None, None, None), None)
+    check = plugin.check_release_format(hostvars, host)
+    assert check == result
+
+
+@pytest.mark.parametrize('hostvars, host, result', [
+    ({"example.com": {"openshift_release": "-3.11.0"}}, "example.com", None),
+    ({"example.com": {"openshift_release": "-3.7.0-0.126.0.git.0.9351aae.el7"}}, "example.com", None),
+    ({"example.com": {"openshift_release": "3.1.2.3"}}, "example.com", None),
+])
+def test_invalid_check_release_format(hostvars, host, result):
+    with pytest.raises(errors.AnsibleModuleError):
+        task = FakeTask('sanity_checks', {'checks': []})
+        plugin = ActionModule(task, None, PlayContext(), None, Templar(None, None, None), None)
+        plugin.check_release_format(hostvars, host)
+
+
+def fake_execute_module(*args):
+    raise AssertionError('this function should not be called')
+
+
+class FakeTask(object):
+    def __init__(self, action, args):
+        self.action = action
+        self.args = args
+        self.async = 0