"""Delivery report handler tests."""

from deliveryhistory import handlers  # noqa (handlers are imported for test client)


EXPECTED_RESULT = {
    'items': [
        {'product_id': 111, 'display_upc': '1110', 'stores_count': 2},
        {'product_id': 222, 'display_upc': '2220', 'stores_count': 3}
    ]
}


def test_get_delivery_stores_count(
        fixture_client, ar_db_fixture, header_fixture):
    """Test handler for get delivered stores count."""
    result = fixture_client.get(
        '/delivered_stores_count?upcs=1110,2220', headers=header_fixture)
    assert result.status_code == 200
    assert result.json == EXPECTED_RESULT