"""Test common util functions"""
import pytest

from switchboard_consumer.utils.common import filter_keys_from_list_of_dict


@pytest.mark.parametrize('test_input, filter_keys, expected', [
    ([{'A': None, 'B': None}], ['B'], [{'B': None}]),
    ([{'A': None, 'B': None}], ['A', 'B'], [{'A': None, 'B': None}]),
    ([{'A': None, 'B': None, 'C': 'Something'},
      {'C': 'Something'}
      ], ['C'], [{'C': 'Something'}, {'C': 'Something'}]),
])
def test_filter_keys_from_list_of_dict(test_input, filter_keys, expected):
    """Test create track mapping with missing data."""
    assert filter_keys_from_list_of_dict(test_input, filter_keys) == expected