salt/tests/unit/utils/test_listdiffer.py
2018-01-12 08:42:39 -08:00

90 lines
3.7 KiB
Python

# -*- coding: utf-8 -*-
# Import python libs
from __future__ import absolute_import, unicode_literals
# Import Salt Testing libs
from tests.support.unit import TestCase
# Import Salt libs
from salt.utils.listdiffer import list_diff
from salt.utils import dictdiffer
NONE = dictdiffer.RecursiveDictDiffer.NONE_VALUE
class ListDictDifferTestCase(TestCase):
def setUp(self):
old_list = [{'key': 1, 'value': 'foo1', 'int_value': 101},
{'key': 2, 'value': 'foo2', 'int_value': 102},
{'key': 3, 'value': 'foo3', 'int_value': 103}]
new_list = [{'key': 1, 'value': 'foo1', 'int_value': 101},
{'key': 2, 'value': 'foo2', 'int_value': 112},
{'key': 5, 'value': 'foo5', 'int_value': 105}]
self.list_diff = list_diff(old_list, new_list, key='key')
def tearDown(self):
for attrname in ('list_diff',):
try:
delattr(self, attrname)
except AttributeError:
continue
def test_added(self):
self.assertEqual(len(self.list_diff.added), 1)
self.assertDictEqual(self.list_diff.added[0],
{'key': 5, 'value': 'foo5', 'int_value': 105})
def test_removed(self):
self.assertEqual(len(self.list_diff.removed), 1)
self.assertDictEqual(self.list_diff.removed[0],
{'key': 3, 'value': 'foo3', 'int_value': 103})
def test_diffs(self):
self.assertEqual(len(self.list_diff.diffs), 3)
self.assertDictEqual(self.list_diff.diffs[0],
{2: {'int_value': {'new': 112, 'old': 102}}})
self.assertDictEqual(self.list_diff.diffs[1],
# Added items
{5: {'int_value': {'new': 105, 'old': NONE},
'key': {'new': 5, 'old': NONE},
'value': {'new': 'foo5', 'old': NONE}}})
self.assertDictEqual(self.list_diff.diffs[2],
# Removed items
{3: {'int_value': {'new': NONE, 'old': 103},
'key': {'new': NONE, 'old': 3},
'value': {'new': NONE, 'old': 'foo3'}}})
def test_new_values(self):
self.assertEqual(len(self.list_diff.new_values), 2)
self.assertDictEqual(self.list_diff.new_values[0],
{'key': 2, 'int_value': 112})
self.assertDictEqual(self.list_diff.new_values[1],
{'key': 5, 'value': 'foo5', 'int_value': 105})
def test_old_values(self):
self.assertEqual(len(self.list_diff.old_values), 2)
self.assertDictEqual(self.list_diff.old_values[0],
{'key': 2, 'int_value': 102})
self.assertDictEqual(self.list_diff.old_values[1],
{'key': 3, 'value': 'foo3', 'int_value': 103})
def test_changed_all(self):
self.assertEqual(self.list_diff.changed(selection='all'),
['key.2.int_value', 'key.5.int_value', 'key.5.value',
'key.3.int_value', 'key.3.value'])
def test_changed_intersect(self):
self.assertEqual(self.list_diff.changed(selection='intersect'),
['key.2.int_value'])
def test_changes_str(self):
self.assertEqual(self.list_diff.changes_str,
'\tidentified by key 2:\n'
'\tint_value from 102 to 112\n'
'\tidentified by key 3:\n'
'\twill be removed\n'
'\tidentified by key 5:\n'
'\twill be added\n')