"""Test configuration and fixtures for the backend tests.""" import logging from collections.abc import AsyncGenerator from collections.abc import Generator from datetime import datetime from datetime import timedelta from datetime import timezone from unittest.mock import AsyncMock from unittest.mock import MagicMock from unittest.mock import patch import pytest import pytest_asyncio from httpx import ASGITransport from httpx import AsyncClient from sqlalchemy.ext.asyncio import AsyncSession from backend.main import app from backend.models import Booking from backend.models import BookingList from backend.models import Invitee from backend.models import Room from backend.models import RoomList from backend.models import User from backend.models import UserList @pytest.fixture def async_session() -> AsyncSession: """Fixture to provide a mocked AsyncSession for database interactions. Returns: A mocked AsyncSession object for database interactions. """ return AsyncMock(spec=AsyncSession) @pytest_asyncio.fixture async def client() -> AsyncGenerator[AsyncClient, None]: """Fixture to provide an AsyncClient for async endpoint testing. Yields: An HTTPX AsyncClient instance for making async HTTP requests. """ transport = ASGITransport(app=app) async with AsyncClient(transport=transport, base_url="http://test") as ac: yield ac @pytest.fixture def mock_logger(request: pytest.FixtureRequest) -> Generator[MagicMock, None, None]: """Fixture to mock a logger for a specified module path. Args: request: Pytest request object, used to pass the module path for patching. Yields: A mocked logger instance for the specified module path. Usage: To use this fixture in a test module, parameterize it with the module path where the logger is defined. """ # Default path if none provided, or use the parameterized path module_path = getattr(request, "param", "backend") + ".logger" mock_logger_instance = MagicMock(spec=logging.Logger) with patch(module_path, mock_logger_instance): yield mock_logger_instance @pytest.fixture def event_publisher() -> AsyncMock: """Fixture for an AsyncMock event publisher for event publishing tests. Returns: An async mock event publisher callable. """ return AsyncMock() @pytest.fixture def sample_users() -> UserList: """Fixture to provide sample User objects for testing. Returns: A list of sample User objects. """ return [ User(email="user1@example.com", name="User One"), User(email="user2@example.com", name="User Two"), ] @pytest.fixture def sample_rooms() -> RoomList: """Fixture to provide sample Room objects for testing. Returns: A list of sample Room objects. """ return [ Room( id=1, name="Room 1", location="Building A", equipment="Projector", capacity=10, ), Room( id=2, name="Room 2", location="Building B", equipment="Whiteboard", capacity=15, ), ] @pytest.fixture def sample_room(sample_rooms: RoomList) -> Room: """Fixture to provide a sample Room object for testing. Args: sample_rooms: The list of sample Room objects. Returns: A single sample Room object. """ return sample_rooms[0] @pytest.fixture def sample_bookings(sample_room: Room) -> BookingList: """Fixture to provide sample Booking objects for testing. Args: sample_room: The sample Room object to attach to bookings. Returns: A list of sample Booking objects with attached room relationships. """ booking1 = Booking( room_id=1, start_time=datetime.now(timezone.utc) + timedelta(hours=3), end_time=datetime.now(timezone.utc) + timedelta(hours=4), title="Sample Booking 1", ) booking1.id = 1 # Manually set ID for testing purposes booking1.room = sample_room # Attach room relationship booking2 = Booking( room_id=1, start_time=datetime.now(timezone.utc) + timedelta(hours=12), end_time=datetime.now(timezone.utc) + timedelta(hours=13), title="Sample Booking 2", ) booking2.id = 2 # Manually set ID for testing purposes booking2.room = sample_room # Attach room relationship return [booking1, booking2] @pytest.fixture def sample_booking(sample_bookings: BookingList) -> Booking: """Fixture to provide a sample Booking object for testing. Args: sample_bookings: The list of sample Booking objects. Returns: A single sample Booking object. """ return sample_bookings[0] @pytest.fixture def sample_invitees(sample_booking: Booking, sample_users: UserList) -> list[Invitee]: """Fixture to provide sample Invitee objects for testing. Args: sample_booking: The sample Booking object to attach to invitees. sample_users: The list of sample User objects to attach to invitees. Returns: A list of sample Invitee objects with attached booking and user relationships. """ invitee1 = Invitee(booking_id=1, user_email="user1@example.com") invitee1.id = 1 invitee1.booking = sample_booking invitee1.user = sample_users[0] invitee2 = Invitee(booking_id=1, user_email="user2@example.com") invitee2.id = 2 invitee2.booking = sample_booking invitee2.user = sample_users[1] return [invitee1, invitee2] @pytest.fixture def sample_invitee(sample_invitees: list[Invitee]) -> Invitee: """Fixture to provide a sample Invitee object for testing. Args: sample_invitees: The list of sample Invitee objects. Returns: A single sample Invitee object. """ return sample_invitees[0]