unit tests for playbook/conditional.py (#18586)

Start of unit tests for playbook/conditional.py

test/units/playbook/test_conditional.py

@@ -0,0 +1,223 @@
+
+from ansible.compat.tests import unittest
+from units.mock.loader import DictDataLoader
+
+from ansible.plugins.strategy import SharedPluginLoaderObj
+from ansible.template import Templar
+from ansible import errors
+
+from ansible.playbook import conditional
+
+
+class TestConditional(unittest.TestCase):
+    def setUp(self):
+        self.loader = DictDataLoader({})
+        self.cond = conditional.Conditional(loader=self.loader)
+        self.shared_loader = SharedPluginLoaderObj()
+        self.templar = Templar(loader=self.loader, variables={})
+
+    def _eval_con(self, when=None, variables=None):
+        when = when or []
+        variables = variables or {}
+        self.cond.when = when
+        ret = self.cond.evaluate_conditional(self.templar, variables)
+        return ret
+
+    def test_false(self):
+        when = [u"False"]
+        ret = self._eval_con(when, {})
+        self.assertFalse(ret)
+
+    def test_true(self):
+        when = [u"True"]
+        ret = self._eval_con(when, {})
+        self.assertTrue(ret)
+
+    def test_undefined(self):
+        when = [u"{{ some_undefined_thing }}"]
+        self.assertRaisesRegexp(errors.AnsibleError, "The conditional check '{{ some_undefined_thing }}' failed",
+                                self._eval_con, when, {})
+
+    def test_defined(self):
+        variables = {'some_defined_thing': True}
+        when = [u"{{ some_defined_thing }}"]
+        ret = self._eval_con(when, variables)
+        self.assertTrue(ret)
+
+    def test_dict_defined_values(self):
+        variables = {'dict_value': 1,
+                     'some_defined_dict': {'key1': 'value1',
+                                           'key2': '{{ dict_value }}'}}
+
+        when = [u"some_defined_dict"]
+        ret = self._eval_con(when, variables)
+        self.assertTrue(ret)
+
+    def test_dict_defined_values_is_defined(self):
+        variables = {'dict_value': 1,
+                     'some_defined_dict': {'key1': 'value1',
+                                           'key2': '{{ dict_value }}'}}
+
+        when = [u"some_defined_dict.key1 is defined"]
+        ret = self._eval_con(when, variables)
+        self.assertTrue(ret)
+
+    def test_dict_defined_multiple_values_is_defined(self):
+        variables = {'dict_value': 1,
+                     'some_defined_dict': {'key1': 'value1',
+                                           'key2': '{{ dict_value }}'}}
+
+        when = [u"some_defined_dict.key1 is defined",
+                u"some_defined_dict.key2 is not undefined"]
+        ret = self._eval_con(when, variables)
+        self.assertTrue(ret)
+
+    def test_dict_undefined_values(self):
+        variables = {'dict_value': 1,
+                     'some_defined_dict_with_undefined_values': {'key1': 'value1',
+                                                                 'key2': '{{ dict_value }}',
+                                                                 'key3': '{{ undefined_dict_value }}'
+                                                                 }}
+
+        when = [u"some_defined_dict_with_undefined_values is defined"]
+        self.assertRaisesRegexp(errors.AnsibleError,
+                                "The conditional check 'some_defined_dict_with_undefined_values is defined' failed.",
+                                self._eval_con,
+                                when, variables)
+
+    def test_nested_hostvars_undefined_values(self):
+        variables = {'dict_value': 1,
+                     'hostvars': {'host1': {'key1': 'value1',
+                                            'key2': '{{ dict_value }}'},
+                                  'host2': '{{ dict_value }}',
+                                  'host3': '{{ undefined_dict_value }}',
+                                  # no host4
+                                  },
+                     'some_dict': {'some_dict_key1': '{{ hostvars["host3"] }}'}
+                     }
+
+        when = [u"some_dict.some_dict_key1 == hostvars['host3']"]
+        #self._eval_con(when, variables)
+        self.assertRaisesRegexp(errors.AnsibleError,
+                                "The conditional check 'some_dict.some_dict_key1 == hostvars\['host3'\]' failed",
+                                #"The conditional check 'some_dict.some_dict_key1 == hostvars['host3']' failed",
+                                #"The conditional check 'some_dict.some_dict_key1 == hostvars['host3']' failed.",
+                                self._eval_con,
+                                when, variables)
+
+    def test_dict_undefined_values_bare(self):
+        variables = {'dict_value': 1,
+                     'some_defined_dict_with_undefined_values': {'key1': 'value1',
+                                                                 'key2': '{{ dict_value }}',
+                                                                 'key3': '{{ undefined_dict_value }}'
+                                                                 }}
+
+        # raises an exception when a non-string conditional is passed to extract_defined_undefined()
+        when = [u"some_defined_dict_with_undefined_values"]
+        self.assertRaisesRegexp(errors.AnsibleError,
+                                "The conditional check 'some_defined_dict_with_undefined_values' failed.",
+                                self._eval_con,
+                                when, variables)
+
+    def test_dict_undefined_values_is_defined(self):
+        variables = {'dict_value': 1,
+                     'some_defined_dict_with_undefined_values': {'key1': 'value1',
+                                                                 'key2': '{{ dict_value }}',
+                                                                 'key3': '{{ undefined_dict_value }}'
+                                                                 }}
+
+        when = [u"some_defined_dict_with_undefined_values is defined"]
+        self.assertRaisesRegexp(errors.AnsibleError,
+                                "The conditional check 'some_defined_dict_with_undefined_values is defined' failed.",
+                                self._eval_con,
+                                when, variables)
+
+    def test_is_defined(self):
+        variables = {'some_defined_thing': True}
+        when = [u"some_defined_thing is defined"]
+        ret = self._eval_con(when, variables)
+        self.assertTrue(ret)
+
+    def test_is_undefined(self):
+        variables = {'some_defined_thing': True}
+        when = [u"some_defined_thing is undefined"]
+        ret = self._eval_con(when, variables)
+        self.assertFalse(ret)
+
+    def test_is_undefined_and_defined(self):
+        variables = {'some_defined_thing': True}
+        when = [u"some_defined_thing is undefined", u"some_defined_thing is defined"]
+        ret = self._eval_con(when, variables)
+        self.assertFalse(ret)
+
+    def test_is_undefined_and_defined_reversed(self):
+        variables = {'some_defined_thing': True}
+        when = [u"some_defined_thing is defined", u"some_defined_thing is undefined"]
+        ret = self._eval_con(when, variables)
+        self.assertFalse(ret)
+
+    def test_is_not_undefined(self):
+        variables = {'some_defined_thing': True}
+        when = [u"some_defined_thing is not undefined"]
+        ret = self._eval_con(when, variables)
+        self.assertTrue(ret)
+
+    def test_is_not_defined(self):
+        variables = {'some_defined_thing': True}
+        when = [u"some_undefined_thing is not defined"]
+        ret = self._eval_con(when, variables)
+        self.assertTrue(ret)
+
+    def test_is_hostvars_quotes_is_defined(self):
+        variables = {'hostvars': {'some_host': {}},
+                     'compare_targets_single': "hostvars['some_host']",
+                     'compare_targets_double': 'hostvars["some_host"]',
+                     'compare_targets': {'double': '{{ compare_targets_double }}',
+                                         'single': "{{ compare_targets_single }}"},
+                     }
+        when = [u"hostvars['some_host'] is defined",
+                u'hostvars["some_host"] is defined',
+                u"{{ compare_targets.double }} is defined",
+                u"{{ compare_targets.single }} is defined"]
+        ret = self._eval_con(when, variables)
+        self.assertTrue(ret)
+
+    def test_is_hostvars_quotes_is_defined_but_is_not_defined(self):
+        variables = {'hostvars': {'some_host': {}},
+                     'compare_targets_single': "hostvars['some_host']",
+                     'compare_targets_double': 'hostvars["some_host"]',
+                     'compare_targets': {'double': '{{ compare_targets_double }}',
+                                         'single': "{{ compare_targets_single }}"},
+                     }
+        when = [u"hostvars['some_host'] is defined",
+                u'hostvars["some_host"] is defined',
+                u"{{ compare_targets.triple }} is defined",
+                u"{{ compare_targets.quadruple }} is defined"]
+        self.assertRaisesRegexp(errors.AnsibleError,
+                                "The conditional check '{{ compare_targets.triple }} is defined' failed",
+                                self._eval_con,
+                                when, variables)
+
+    def test_is_hostvars_host_is_defined(self):
+        variables = {'hostvars': {'some_host': {}, }}
+        when = [u"hostvars['some_host'] is defined"]
+        ret = self._eval_con(when, variables)
+        self.assertTrue(ret)
+
+    def test_is_hostvars_host_undefined_is_defined(self):
+        variables = {'hostvars': {'some_host': {}, }}
+        when = [u"hostvars['some_undefined_host'] is defined"]
+        ret = self._eval_con(when, variables)
+        self.assertFalse(ret)
+
+    def test_is_hostvars_host_undefined_is_undefined(self):
+        variables = {'hostvars': {'some_host': {}, }}
+        when = [u"hostvars['some_undefined_host'] is undefined"]
+        ret = self._eval_con(when, variables)
+        self.assertTrue(ret)
+
+    def test_is_hostvars_host_undefined_is_not_defined(self):
+        variables = {'hostvars': {'some_host': {}, }}
+        when = [u"hostvars['some_undefined_host'] is not defined"]
+        ret = self._eval_con(when, variables)
+        self.assertTrue(ret)