import unittest

from parameterized import parameterized

from apps_etl_utils.utils import get_param


class UtilsTestCase(unittest.TestCase):
    @parameterized.expand(
        [
            ('1', [{'1': 2}, {'1': 3}, ], 4, 2),
            ('1', [{'2': 2}, {'1': 3}, ], 4, 3),
            ('3', [{'1': 2}, {'1': 3}, ], None, None),
            ('3', [{'1': 2}, {'1': 3}, ], 4, 4),
            ('3', [{}, {}, {}, {}, {}, ], 4, 4),
            ('3', [], 4, 4),
        ]
    )  # yapf: disable
    def test_get_param(self, param, sources, default, expected):
        actual = get_param(param, *sources, default=default)
        self.assertEqual(actual, expected)