62 lines
1.8 KiB
Python
Executable File
62 lines
1.8 KiB
Python
Executable File
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
import unittest
|
|
import mozunit
|
|
from taskgraph.util.templates import (
|
|
merge_to,
|
|
merge
|
|
)
|
|
|
|
|
|
class MergeTest(unittest.TestCase):
|
|
|
|
def test_merge_to_dicts(self):
|
|
source = {'a': 1, 'b': 2}
|
|
dest = {'b': '20', 'c': 30}
|
|
expected = {
|
|
'a': 1, # source only
|
|
'b': 2, # source overrides dest
|
|
'c': 30, # dest only
|
|
}
|
|
self.assertEqual(merge_to(source, dest), expected)
|
|
self.assertEqual(dest, expected)
|
|
|
|
def test_merge_to_lists(self):
|
|
source = {'x': [3, 4]}
|
|
dest = {'x': [1, 2]}
|
|
expected = {'x': [1, 2, 3, 4]} # dest first
|
|
self.assertEqual(merge_to(source, dest), expected)
|
|
self.assertEqual(dest, expected)
|
|
|
|
def test_merge_diff_types(self):
|
|
source = {'x': [1, 2]}
|
|
dest = {'x': 'abc'}
|
|
expected = {'x': [1, 2]} # source wins
|
|
self.assertEqual(merge_to(source, dest), expected)
|
|
self.assertEqual(dest, expected)
|
|
|
|
def test_merge(self):
|
|
first = {'a': 1, 'b': 2, 'd': 11}
|
|
second = {'b': 20, 'c': 30}
|
|
third = {'c': 300, 'd': 400}
|
|
expected = {
|
|
'a': 1,
|
|
'b': 20,
|
|
'c': 300,
|
|
'd': 400,
|
|
}
|
|
self.assertEqual(merge(first, second, third), expected)
|
|
|
|
# inputs haven't changed..
|
|
self.assertEqual(first, {'a': 1, 'b': 2, 'd': 11})
|
|
self.assertEqual(second, {'b': 20, 'c': 30})
|
|
self.assertEqual(third, {'c': 300, 'd': 400})
|
|
|
|
|
|
if __name__ == '__main__':
|
|
mozunit.main()
|