"""Tests for product submission API endpoint validation functions."""

import pytest

from availability.validation import product_submission


@pytest.mark.parametrize(
    'input_dict, expected_dict', (
        ({'store_id': 1, 'upc': 2}, {'store_id': 1, 'upc': 2}),
        ({'countries': ['a', 'b'], 'upc': 2}, {'upc': 2}),
        ({'countries': ['a', 'b']}, {}),
        ({}, {}),
    ))
def test_filter_initial_product_fields(input_dict, expected_dict):
    """Test filter_initial_product_fields for various input values."""
    assert (
        product_submission.filter_initial_product_fields(input_dict) ==
        expected_dict
    )