"""Tests for utils module."""

import pytest

from reserves_take.utils import chunk_post_payload


@pytest.mark.parametrize('chunk_size', [1, 3, 5, 10, 100])
def test_chunk_post_payload(chunk_size):
    """Test for chunk_past_payload method."""
    initial_payload = [f'item {ind}' for ind in range(1, 21)]
    chunked_payload = []
    for chunk in chunk_post_payload(initial_payload, chunk_size=chunk_size):
        chunked_payload += chunk
    assert initial_payload == chunked_payload