"""Unit tests for digital queries functions."""

from flows.digital import config
from flows.digital import queries
from flows.queries import sql_upcs_condition_in


def test_get_unload_from_snowflake_sql():
    """Test get_unload_from_snowflake_sql generator."""
    bucket = 'test-bucket'
    correlation_id = '1234-3456-5678'
    date_end = '2012-12-20'
    date_start = '2000-01-01'
    upcs = ['123', '234', '345', '456', '567']
    upcs_batches = [['123', '234', '345'], ['456', '567']]
    destination = config.UNLOAD_DESTINATION.format(
        bucket=bucket, correlation_id=correlation_id)

    queries.get_unload_from_snowflake_sql.batch_size = 3
    results_raw = queries.get_unload_from_snowflake_sql(
        bucket, correlation_id, date_end, date_start, upcs)
    results = list(results_raw)

    assert len(results) == len(upcs_batches)
    for i, result in enumerate(results):
        assert destination in result
        assert sql_upcs_condition_in(upcs_batches[i], 'fa.releaseid') in result


def test_get_delete_from_digital_revenue_sql():
    """Test get_delete_from_digital_revenue_sql generator."""
    upcs = ['123', '234', '345', '456', '567']
    upcs_batches = [['123', '234', '345'], ['456', '567']]

    queries.get_delete_from_digital_revenue_sql.batch_size = 3
    results_raw = queries.get_delete_from_digital_revenue_sql(upcs)
    results = list(results_raw)

    assert len(results) == len(upcs_batches)
    for i, result in enumerate(results):
        assert sql_upcs_condition_in(upcs_batches[i]) in result