"""Functional tests for fetching statement periods."""

import json

from tests.testutils import db, mock_auth


@db.test_schema_default_seed
def test_post_product_counts_dataloader(auth_client, mocker):
    """Test post statement periods dataloader."""
    vendor_id = 24601

    mock_auth(mocker, vendor_id)

    result = auth_client.post(
        "/products/counts-dataloader",
        data=json.dumps({"product_ids": [2, 6, 10050, 100]}),
        content_type="application/json",
    )

    assert result.json == [
        {
            "data": {
                "product_id": 2,
                "splits_count": 1,
                "tracks_count": 1,
                "collaborators_count": 1,
                "tracks_with_splits_count": 1,
            }
        },
        {"data": None},
        {
            "data": {
                "product_id": 10050,
                "splits_count": 0,
                "tracks_count": 0,
                "collaborators_count": 0,
                "tracks_with_splits_count": 0,
            }
        },
        {"data": None},
    ]

    assert result.status_code == 200