"""Shared test fixtures."""

from unittest.mock import MagicMock


def make_mock_engine(select_rows: list[tuple] | None = None):
    """Return a mock SQLAlchemy engine."""
    select_rows = select_rows or []

    engine = MagicMock()

    write_conn = MagicMock()
    engine.begin.return_value.__enter__ = MagicMock(return_value=write_conn)
    engine.begin.return_value.__exit__ = MagicMock(return_value=False)

    read_conn = MagicMock()
    read_conn.execute.return_value = iter(select_rows)
    engine.connect.return_value.__enter__ = MagicMock(return_value=read_conn)
    engine.connect.return_value.__exit__ = MagicMock(return_value=False)

    return engine, write_conn, read_conn


def make_mock_listing(data: dict):
    """Return a mock pyfunda Listing-like object."""
    listing = MagicMock()
    listing.to_dict.return_value = data
    return listing