"""Cable Calculation queries function unit tests."""

from flows.cable_calculation import queries


def test_get_delete_from_cable_revenue_sql():
    """Test get_delete_from_cable_revenue function."""
    upcs = ['123', '234', '345', '456', '567']
    upcs_batches = [['123', '234', '345'], ['456', '567']]

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

    assert len(results) == len(upcs_batches)
    for batch, upcs_batch in enumerate(upcs_batches):
        for upc in upcs_batch:
            assert upc in results[batch]