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

import json

from tests.testutils import db, mock_auth


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

    mock_auth(mocker, vendor_id)

    result = auth_client.post(
        "/statement-period/dataloader",
        data=json.dumps({"statement_period_ids": [1, 2, 3, 4]}),
        content_type="application/json",
    )

    assert result.json == [
        {
            "data": {
                "id": 1,
                "name": "SP 1",
                "status": "CLOSED",
                "created_date": "2022-04-05T06:07:08",
                "updated_date": "2022-04-05T06:07:08",
                "vendor_id": vendor_id,
            }
        },
        {
            "data": {
                "id": 2,
                "name": "SP 2",
                "status": "OPEN",
                "created_date": "2022-04-06T06:07:08",
                "updated_date": "2022-04-06T06:07:08",
                "vendor_id": vendor_id,
            }
        },
        {"data": None},
        {"data": None},
    ]

    assert result.status_code == 200