workload-automation/tests/test_config.py

import unittest
from nose.tools import assert_equal

from wa.utils.misc import merge_config_values


class TestConfigUtils(unittest.TestCase):

    def test_merge_values(self):
        test_cases = [
            # base, other, expected_result
            ('a', 3, 3),
            ('a', [1, 2], ['a', 1, 2]),
            ({1: 2}, [3, 4], [{1: 2}, 3, 4]),
            (set([2]), [1, 2, 3], [2, 1, 3]),
            ([1, 2, 3], set([2]), set([1, 2, 3])),
            ([1, 2], None, [1, 2]),
            (None, 'a', 'a'),
        ]
        for v1, v2, expected in test_cases:
            result = merge_config_values(v1, v2)
            assert_equal(result, expected)
            if v2 is not None:
                assert_equal(type(result), type(v2))