# (c) 2012-2014, Michael DeHaan # (c) 2015, Toshio Kuraotmi # # This file is part of Ansible # # Ansible is free software: you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # Ansible is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with Ansible. If not, see . # Make coding more python3-ish from __future__ import (absolute_import, division, print_function) __metaclass__ = type from collections import defaultdict from unittest import mock from units.compat import unittest from ansible.errors import AnsibleError from ansible.utils.vars import combine_vars, merge_hash class TestVariableUtils(unittest.TestCase): def setUp(self): pass def tearDown(self): pass combine_vars_merge_data = ( dict( a=dict(a=1), b=dict(b=2), result=dict(a=1, b=2), ), dict( a=dict(a=1, c=dict(foo='bar')), b=dict(b=2, c=dict(baz='bam')), result=dict(a=1, b=2, c=dict(foo='bar', baz='bam')) ), dict( a=defaultdict(a=1, c=defaultdict(foo='bar')), b=dict(b=2, c=dict(baz='bam')), result=defaultdict(a=1, b=2, c=defaultdict(foo='bar', baz='bam')) ), ) combine_vars_replace_data = ( dict( a=dict(a=1), b=dict(b=2), result=dict(a=1, b=2) ), dict( a=dict(a=1, c=dict(foo='bar')), b=dict(b=2, c=dict(baz='bam')), result=dict(a=1, b=2, c=dict(baz='bam')) ), dict( a=defaultdict(a=1, c=dict(foo='bar')), b=dict(b=2, c=defaultdict(baz='bam')), result=defaultdict(a=1, b=2, c=defaultdict(baz='bam')) ), ) def test_combine_vars_improper_args(self): with mock.patch('ansible.constants.DEFAULT_HASH_BEHAVIOUR', 'replace'): with self.assertRaises(AnsibleError): combine_vars([1, 2, 3], dict(a=1)) with self.assertRaises(AnsibleError): combine_vars(dict(a=1), [1, 2, 3]) with mock.patch('ansible.constants.DEFAULT_HASH_BEHAVIOUR', 'merge'): with self.assertRaises(AnsibleError): combine_vars([1, 2, 3], dict(a=1)) with self.assertRaises(AnsibleError): combine_vars(dict(a=1), [1, 2, 3]) def test_combine_vars_replace(self): with mock.patch('ansible.constants.DEFAULT_HASH_BEHAVIOUR', 'replace'): for test in self.combine_vars_replace_data: self.assertEqual(combine_vars(test['a'], test['b']), test['result']) def test_combine_vars_merge(self): with mock.patch('ansible.constants.DEFAULT_HASH_BEHAVIOUR', 'merge'): for test in self.combine_vars_merge_data: self.assertEqual(combine_vars(test['a'], test['b']), test['result']) merge_hash_data = { "low_prio": { "a": { "a'": { "x": "low_value", "y": "low_value", "list": ["low_value"] } }, "b": [1, 1, 2, 3] }, "high_prio": { "a": { "a'": { "y": "high_value", "z": "high_value", "list": ["high_value"] } }, "b": [3, 4, 4, {"5": "value"}] } } def test_merge_hash_simple(self): for test in self.combine_vars_merge_data: self.assertEqual(merge_hash(test['a'], test['b']), test['result']) low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": { "a'": { "x": "low_value", "y": "high_value", "z": "high_value", "list": ["high_value"] } }, "b": high['b'] } self.assertEqual(merge_hash(low, high), expected) def test_merge_hash_non_recursive_and_list_replace(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = high self.assertEqual(merge_hash(low, high, False, 'replace'), expected) def test_merge_hash_non_recursive_and_list_keep(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": high['a'], "b": low['b'] } self.assertEqual(merge_hash(low, high, False, 'keep'), expected) def test_merge_hash_non_recursive_and_list_append(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": high['a'], "b": low['b'] + high['b'] } self.assertEqual(merge_hash(low, high, False, 'append'), expected) def test_merge_hash_non_recursive_and_list_prepend(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": high['a'], "b": high['b'] + low['b'] } self.assertEqual(merge_hash(low, high, False, 'prepend'), expected) def test_merge_hash_non_recursive_and_list_append_rp(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": high['a'], "b": [1, 1, 2] + high['b'] } self.assertEqual(merge_hash(low, high, False, 'append_rp'), expected) def test_merge_hash_non_recursive_and_list_prepend_rp(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": high['a'], "b": high['b'] + [1, 1, 2] } self.assertEqual(merge_hash(low, high, False, 'prepend_rp'), expected) def test_merge_hash_recursive_and_list_replace(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": { "a'": { "x": "low_value", "y": "high_value", "z": "high_value", "list": ["high_value"] } }, "b": high['b'] } self.assertEqual(merge_hash(low, high, True, 'replace'), expected) def test_merge_hash_recursive_and_list_keep(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": { "a'": { "x": "low_value", "y": "high_value", "z": "high_value", "list": ["low_value"] } }, "b": low['b'] } self.assertEqual(merge_hash(low, high, True, 'keep'), expected) def test_merge_hash_recursive_and_list_append(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": { "a'": { "x": "low_value", "y": "high_value", "z": "high_value", "list": ["low_value", "high_value"] } }, "b": low['b'] + high['b'] } self.assertEqual(merge_hash(low, high, True, 'append'), expected) def test_merge_hash_recursive_and_list_prepend(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": { "a'": { "x": "low_value", "y": "high_value", "z": "high_value", "list": ["high_value", "low_value"] } }, "b": high['b'] + low['b'] } self.assertEqual(merge_hash(low, high, True, 'prepend'), expected) def test_merge_hash_recursive_and_list_append_rp(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": { "a'": { "x": "low_value", "y": "high_value", "z": "high_value", "list": ["low_value", "high_value"] } }, "b": [1, 1, 2] + high['b'] } self.assertEqual(merge_hash(low, high, True, 'append_rp'), expected) def test_merge_hash_recursive_and_list_prepend_rp(self): low = self.merge_hash_data['low_prio'] high = self.merge_hash_data['high_prio'] expected = { "a": { "a'": { "x": "low_value", "y": "high_value", "z": "high_value", "list": ["high_value", "low_value"] } }, "b": high['b'] + [1, 1, 2] } self.assertEqual(merge_hash(low, high, True, 'prepend_rp'), expected)