"""Tests for analytics_aggregation.generators.sos_generators module."""

from unittest.mock import patch

from analytics_aggregation.generators import sos_generators


@patch('analytics_aggregation.generators.sos_generators.sos_config')
def test_granular_copy_sos_data_params(sos_config):
    """Test granular_copy_sos_data_params generator."""
    params_mock = [1, 2, 3, 4]
    sos_config.GRANULAR_COPY_PARAMS = params_mock
    context = {}
    generator = sos_generators.granular_copy_sos_data_params(context)
    assert list(generator) == params_mock


@patch('analytics_aggregation.generators.sos_generators.sos_config')
def test_granular_copy_sos_data_params_skip_placements(sos_config):
    """Test granular_copy_sos_data_params generator."""
    params_mock = [1, 2, 3, 4]
    sos_config.GRANULAR_COPY_TOTALS_PARAMS = params_mock
    context = {'skip_placements': True}
    generator = sos_generators.granular_copy_sos_data_params(context)
    assert list(generator) == params_mock