"""Conftest."""

from unittest.mock import MagicMock

import pytest


@pytest.fixture
def mock_file_handler():
    """Fixture for file handler."""
    mock_file_handler = MagicMock()
    mock_file_handler.write.return_value = None

    return mock_file_handler


@pytest.fixture
def mock_smart_open_obj(mock_file_handler):
    """Fixture for smart_open object."""
    so_obj = MagicMock()
    so_obj.__enter__.return_value = mock_file_handler
    so_obj.__exit__.return_value = None

    return so_obj