"""Utils for tests."""

import csv
from datetime import date
from io import StringIO
import typing

import pytest

from src.utils import parse_date


def create_sample_csv_buffer(
    rows: typing.Iterable[typing.Iterable[typing.Any]],
) -> StringIO:
    """
    Create temporary csv file with data.

    returns filename.
    """
    csv_buffer = StringIO()
    writer = csv.writer(csv_buffer)
    writer.writerows(rows)
    csv_buffer.seek(0)
    return csv_buffer


@pytest.mark.parametrize(
    'date_value', ('2015-11-05', '11/5/2015', date(2015, 11, 5), None)
)
def test_parse_date(date_value: str | date | None) -> None:
    """Test parse_date."""
    result = parse_date(date_value)
    if date_value is None:
        assert result is None
    else:
        assert result == date(2015, 11, 5)