import json
from unittest.mock import Mock, AsyncMock

import aiohttp


def get_api_data(fixture: str, plain=False) -> str | dict:
    with open(f'tests/fixtures/{fixture}', 'r') as file:
        if plain:
            return file.read()
        return json.load(file)


def get_mock_session(json_file=None, text_file=None):
    # Create the mock response
    mock_response = Mock()
    if json_file is not None:
        mock_response.json = AsyncMock(return_value=get_api_data(json_file))
    if text_file is not None:
        mock_response.text = AsyncMock(return_value=get_api_data(text_file, plain=True))

    mock_response.status = 200
    mock_response.headers = dict()

    # Create the mock session
    mock_session = Mock(aiohttp.ClientSession)
    mock_session.request = AsyncMock(return_value=mock_response)
    return mock_session