thrillwiki_django_no_react/backend/tests/e2e/conftest.py

import pytest
from playwright.sync_api import Page


@pytest.fixture(scope="session")
def setup_test_data(django_db_setup, django_db_blocker):
    """
    Setup test data before the test session using factories.

    This fixture:
    - Uses factories instead of shelling out to management commands
    - Is scoped to session (not autouse per test) to reduce overhead
    - Uses django_db_blocker to allow database access in session-scoped fixture
    """
    with django_db_blocker.unblock():
        from django.contrib.auth import get_user_model

        User = get_user_model()

        # Create test users if they don't exist
        test_users = [
            {"username": "testuser", "email": "testuser@example.com", "password": "testpass123"},
            {"username": "moderator", "email": "moderator@example.com", "password": "modpass123", "is_staff": True},
            {"username": "admin", "email": "admin@example.com", "password": "adminpass123", "is_staff": True, "is_superuser": True},
        ]

        for user_data in test_users:
            password = user_data.pop("password")
            user, created = User.objects.get_or_create(
                username=user_data["username"],
                defaults=user_data
            )
            if created:
                user.set_password(password)
                user.save()

    yield

    # Cleanup is handled automatically by pytest-django's transactional database


@pytest.fixture(autouse=True)
def setup_page(page: Page):
    """Configure page for tests"""
    # Set viewport size
    page.set_viewport_size({"width": 1280, "height": 720})

    # Set default navigation timeout
    page.set_default_timeout(5000)

    # Listen for console errors
    page.on(
        "console",
        lambda msg: (
            print(f"Browser console {msg.type}: {msg.text}")
            if msg.type == "error"
            else None
        ),
    )

    yield page


@pytest.fixture
def auth_page(page: Page, live_server, setup_test_data):
    """Fixture for authenticated page"""
    # Login using live_server URL
    page.goto(f"{live_server.url}/accounts/login/")
    page.get_by_label("Username").fill("testuser")
    page.get_by_label("Password").fill("testpass123")
    page.get_by_role("button", name="Sign In").click()

    yield page


@pytest.fixture
def mod_page(page: Page, live_server, setup_test_data):
    """Fixture for moderator page"""
    # Login as moderator using live_server URL
    page.goto(f"{live_server.url}/accounts/login/")
    page.get_by_label("Username").fill("moderator")
    page.get_by_label("Password").fill("modpass123")
    page.get_by_role("button", name="Sign In").click()

    yield page


@pytest.fixture
def test_park(auth_page: Page, live_server):
    """Fixture for test park"""
    # Create test park using live_server URL
    auth_page.goto(f"{live_server.url}/parks/create/")
    auth_page.get_by_label("Name").fill("Test Park")
    auth_page.get_by_label("Location").fill("Orlando, FL")
    auth_page.get_by_label("Description").fill("A test theme park")
    auth_page.get_by_label("Photo").set_input_files("tests/fixtures/test_photo.jpg")
    auth_page.get_by_role("button", name="Create Park").click()

    yield auth_page


@pytest.fixture
def test_ride(test_park: Page, live_server):
    """Fixture for test ride"""
    # Create test ride using live_server URL
    test_park.goto(f"{live_server.url}/rides/create/")
    test_park.get_by_label("Name").fill("Test Ride")
    test_park.get_by_label("Park").select_option("Test Park")
    test_park.get_by_label("Type").select_option("Roller Coaster")
    test_park.get_by_label("Description").fill("A test ride")
    test_park.get_by_label("Photo").set_input_files("tests/fixtures/test_photo.jpg")
    test_park.get_by_role("button", name="Create Ride").click()

    yield test_park


@pytest.fixture
def test_review(test_park: Page, live_server):
    """Fixture for test review"""
    # Create test review using live_server URL
    test_park.goto(f"{live_server.url}/parks/test-park/")
    test_park.get_by_role("tab", name="Reviews").click()
    test_park.get_by_role("button", name="Write Review").click()
    test_park.get_by_label("Rating").select_option("5")
    test_park.get_by_label("Title").fill("Test Review")
    test_park.get_by_label("Review").fill("This is a test review")
    test_park.get_by_role("button", name="Submit Review").click()

    yield test_park


# =============================================================================
# FSM Testing Fixtures
# =============================================================================


@pytest.fixture
def admin_page(page: Page, live_server, setup_test_data):
    """Fixture for admin/superuser page"""
    # Login as admin using live_server URL
    page.goto(f"{live_server.url}/accounts/login/")
    page.get_by_label("Username").fill("admin")
    page.get_by_label("Password").fill("adminpass123")
    page.get_by_role("button", name="Sign In").click()

    yield page


@pytest.fixture
def submission_pending(db):
    """Create a pending EditSubmission for FSM testing."""
    from django.contrib.auth import get_user_model
    from apps.moderation.models import EditSubmission
    from apps.parks.models import Park
    from django.contrib.contenttypes.models import ContentType

    User = get_user_model()

    # Get or create test user
    user, _ = User.objects.get_or_create(
        username="fsm_test_submitter",
        defaults={"email": "fsm_test@example.com"}
    )
    user.set_password("testpass123")
    user.save()

    # Get a park
    park = Park.objects.first()
    if not park:
        pytest.skip("No parks available for testing")

    content_type = ContentType.objects.get_for_model(Park)

    submission = EditSubmission.objects.create(
        user=user,
        content_type=content_type,
        object_id=park.pk,
        submission_type="EDIT",
        changes={"description": "FSM test submission"},
        reason="FSM e2e test",
        status="PENDING"
    )

    yield submission

    # Cleanup
    try:
        submission.delete()
    except Exception:
        pass


@pytest.fixture
def submission_approved(db):
    """Create an approved EditSubmission for FSM testing."""
    from django.contrib.auth import get_user_model
    from apps.moderation.models import EditSubmission
    from apps.parks.models import Park
    from django.contrib.contenttypes.models import ContentType

    User = get_user_model()

    user, _ = User.objects.get_or_create(
        username="fsm_test_submitter_approved",
        defaults={"email": "fsm_approved@example.com"}
    )

    park = Park.objects.first()
    if not park:
        pytest.skip("No parks available for testing")

    content_type = ContentType.objects.get_for_model(Park)

    submission = EditSubmission.objects.create(
        user=user,
        content_type=content_type,
        object_id=park.pk,
        submission_type="EDIT",
        changes={"description": "Already approved"},
        reason="FSM approved test",
        status="APPROVED"
    )

    yield submission

    try:
        submission.delete()
    except Exception:
        pass


@pytest.fixture
def park_operating(db):
    """Create an operating Park for FSM testing."""
    from tests.factories import ParkFactory

    park = ParkFactory(
        name="FSM Test Park Operating",
        slug="fsm-test-park-operating",
        status="OPERATING"
    )

    yield park


@pytest.fixture
def park_closed_temp(db):
    """Create a temporarily closed Park for FSM testing."""
    from tests.factories import ParkFactory

    park = ParkFactory(
        name="FSM Test Park Closed Temp",
        slug="fsm-test-park-closed-temp",
        status="CLOSED_TEMP"
    )

    yield park


@pytest.fixture
def park_closed_perm(db):
    """Create a permanently closed Park for FSM testing."""
    from tests.factories import ParkFactory
    from datetime import date, timedelta

    park = ParkFactory(
        name="FSM Test Park Closed Perm",
        slug="fsm-test-park-closed-perm",
        status="CLOSED_PERM",
        closing_date=date.today() - timedelta(days=365)
    )

    yield park


@pytest.fixture
def ride_operating(db, park_operating):
    """Create an operating Ride for FSM testing."""
    from tests.factories import RideFactory

    ride = RideFactory(
        name="FSM Test Ride Operating",
        slug="fsm-test-ride-operating",
        park=park_operating,
        status="OPERATING"
    )

    yield ride


@pytest.fixture
def ride_sbno(db, park_operating):
    """Create an SBNO Ride for FSM testing."""
    from tests.factories import RideFactory

    ride = RideFactory(
        name="FSM Test Ride SBNO",
        slug="fsm-test-ride-sbno",
        park=park_operating,
        status="SBNO"
    )

    yield ride


@pytest.fixture
def ride_closed_perm(db, park_operating):
    """Create a permanently closed Ride for FSM testing."""
    from tests.factories import RideFactory
    from datetime import date, timedelta

    ride = RideFactory(
        name="FSM Test Ride Closed Perm",
        slug="fsm-test-ride-closed-perm",
        park=park_operating,
        status="CLOSED_PERM",
        closing_date=date.today() - timedelta(days=365)
    )

    yield ride


@pytest.fixture
def queue_item_pending(db):
    """Create a pending ModerationQueue item for FSM testing."""
    from django.contrib.auth import get_user_model
    from apps.moderation.models import ModerationQueue

    User = get_user_model()

    user, _ = User.objects.get_or_create(
        username="fsm_queue_flagger",
        defaults={"email": "fsm_queue@example.com"}
    )

    queue_item = ModerationQueue.objects.create(
        item_type="CONTENT_REVIEW",
        status="PENDING",
        priority="MEDIUM",
        title="FSM Test Queue Item",
        description="Queue item for FSM e2e testing",
        flagged_by=user
    )

    yield queue_item

    try:
        queue_item.delete()
    except Exception:
        pass


@pytest.fixture
def bulk_operation_pending(db):
    """Create a pending BulkOperation for FSM testing."""
    from django.contrib.auth import get_user_model
    from apps.moderation.models import BulkOperation

    User = get_user_model()

    user, _ = User.objects.get_or_create(
        username="fsm_bulk_creator",
        defaults={"email": "fsm_bulk@example.com", "is_staff": True}
    )

    operation = BulkOperation.objects.create(
        operation_type="IMPORT",
        status="PENDING",
        priority="MEDIUM",
        description="FSM Test Bulk Operation",
        parameters={"test": True},
        created_by=user,
        total_items=10
    )

    yield operation

    try:
        operation.delete()
    except Exception:
        pass


# =============================================================================
# Helper Fixtures
# =============================================================================


@pytest.fixture
def live_server(live_server_url):
    """Provide the live server URL for tests.

    Note: This fixture is provided by pytest-django. The live_server_url
    fixture provides the URL as a string.
    """
    class LiveServer:
        url = live_server_url

    return LiveServer()


@pytest.fixture
def moderator_user(db):
    """Get or create a moderator user for testing."""
    from django.contrib.auth import get_user_model

    User = get_user_model()

    user, _ = User.objects.get_or_create(
        username="moderator",
        defaults={
            "email": "moderator@example.com",
            "is_staff": True
        }
    )
    user.set_password("modpass123")
    user.save()

    return user


@pytest.fixture
def regular_user(db):
    """Get or create a regular user for testing."""
    from django.contrib.auth import get_user_model

    User = get_user_model()

    user, _ = User.objects.get_or_create(
        username="testuser",
        defaults={"email": "testuser@example.com"}
    )
    user.set_password("testpass123")
    user.save()

    return user


@pytest.fixture
def parks_data(db):
    """Create test parks for E2E testing."""
    from tests.factories import ParkFactory

    parks = [
        ParkFactory(
            name=f"E2E Test Park {i}",
            slug=f"e2e-test-park-{i}",
            status="OPERATING"
        )
        for i in range(3)
    ]

    return parks


@pytest.fixture
def rides_data(db, parks_data):
    """Create test rides for E2E testing."""
    from tests.factories import RideFactory

    rides = []
    for park in parks_data:
        for i in range(2):
            ride = RideFactory(
                name=f"E2E Test Ride {park.name} {i}",
                slug=f"e2e-test-ride-{park.slug}-{i}",
                park=park,
                status="OPERATING"
            )
            rides.append(ride)

    return rides