"""Tests for response_params utils."""

import requests
from pytest_mock.plugin import MockerFixture

from notifications.utils.response_params import get_filename_from_content_disposition


def test_get_filename_from_content_disposition(mocker: MockerFixture) -> None:
    """Test extracting filename from Content-Disposition header."""
    filename = 'test_file.xlsx'
    mock_response = mocker.Mock(spec=requests.Response)
    mock_response.headers = {'Content-Disposition': f'attachment; filename="{filename}"'}

    assert get_filename_from_content_disposition(mock_response) == filename


def test_get_filename_from_content_disposition_missing_header(mocker: MockerFixture) -> None:
    """Test extracting filename from Content-Disposition header when missing."""
    mock_response = mocker.Mock(spec=requests.Response)
    mock_response.headers = {}

    assert get_filename_from_content_disposition(mock_response) is None