thrillwiki_django_no_react/backend/tests/e2e/conftest.py

import contextlib
import tempfile
from pathlib import Path

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(scope="session")
def test_images():
    """Generate temporary test images for upload tests.

    Creates simple valid JPEG images using raw bytes.
    Images are cleaned up after the test session.
    """
    # Minimal valid JPEG image (1x1 red pixel)
    # This is a valid JPEG that any image library will accept
    jpeg_bytes = bytes([
        0xFF, 0xD8, 0xFF, 0xE0, 0x00, 0x10, 0x4A, 0x46, 0x49, 0x46, 0x00, 0x01,
        0x01, 0x00, 0x00, 0x01, 0x00, 0x01, 0x00, 0x00, 0xFF, 0xDB, 0x00, 0x43,
        0x00, 0x08, 0x06, 0x06, 0x07, 0x06, 0x05, 0x08, 0x07, 0x07, 0x07, 0x09,
        0x09, 0x08, 0x0A, 0x0C, 0x14, 0x0D, 0x0C, 0x0B, 0x0B, 0x0C, 0x19, 0x12,
        0x13, 0x0F, 0x14, 0x1D, 0x1A, 0x1F, 0x1E, 0x1D, 0x1A, 0x1C, 0x1C, 0x20,
        0x24, 0x2E, 0x27, 0x20, 0x22, 0x2C, 0x23, 0x1C, 0x1C, 0x28, 0x37, 0x29,
        0x2C, 0x30, 0x31, 0x34, 0x34, 0x34, 0x1F, 0x27, 0x39, 0x3D, 0x38, 0x32,
        0x3C, 0x2E, 0x33, 0x34, 0x32, 0xFF, 0xC0, 0x00, 0x0B, 0x08, 0x00, 0x01,
        0x00, 0x01, 0x01, 0x01, 0x11, 0x00, 0xFF, 0xC4, 0x00, 0x1F, 0x00, 0x00,
        0x01, 0x05, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x00, 0x00, 0x00, 0x00,
        0x00, 0x00, 0x00, 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
        0x09, 0x0A, 0x0B, 0xFF, 0xC4, 0x00, 0xB5, 0x10, 0x00, 0x02, 0x01, 0x03,
        0x03, 0x02, 0x04, 0x03, 0x05, 0x05, 0x04, 0x04, 0x00, 0x00, 0x01, 0x7D,
        0x01, 0x02, 0x03, 0x00, 0x04, 0x11, 0x05, 0x12, 0x21, 0x31, 0x41, 0x06,
        0x13, 0x51, 0x61, 0x07, 0x22, 0x71, 0x14, 0x32, 0x81, 0x91, 0xA1, 0x08,
        0x23, 0x42, 0xB1, 0xC1, 0x15, 0x52, 0xD1, 0xF0, 0x24, 0x33, 0x62, 0x72,
        0x82, 0x09, 0x0A, 0x16, 0x17, 0x18, 0x19, 0x1A, 0x25, 0x26, 0x27, 0x28,
        0x29, 0x2A, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3A, 0x43, 0x44, 0x45,
        0x46, 0x47, 0x48, 0x49, 0x4A, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59,
        0x5A, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68, 0x69, 0x6A, 0x73, 0x74, 0x75,
        0x76, 0x77, 0x78, 0x79, 0x7A, 0x83, 0x84, 0x85, 0x86, 0x87, 0x88, 0x89,
        0x8A, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99, 0x9A, 0xA2, 0xA3,
        0xA4, 0xA5, 0xA6, 0xA7, 0xA8, 0xA9, 0xAA, 0xB2, 0xB3, 0xB4, 0xB5, 0xB6,
        0xB7, 0xB8, 0xB9, 0xBA, 0xC2, 0xC3, 0xC4, 0xC5, 0xC6, 0xC7, 0xC8, 0xC9,
        0xCA, 0xD2, 0xD3, 0xD4, 0xD5, 0xD6, 0xD7, 0xD8, 0xD9, 0xDA, 0xE1, 0xE2,
        0xE3, 0xE4, 0xE5, 0xE6, 0xE7, 0xE8, 0xE9, 0xEA, 0xF1, 0xF2, 0xF3, 0xF4,
        0xF5, 0xF6, 0xF7, 0xF8, 0xF9, 0xFA, 0xFF, 0xDA, 0x00, 0x08, 0x01, 0x01,
        0x00, 0x00, 0x3F, 0x00, 0xFB, 0xD5, 0xDB, 0x20, 0xA8, 0xBA, 0xAE, 0xAF,
        0xDA, 0xAD, 0x28, 0xA6, 0x02, 0x8A, 0x28, 0x03, 0xFF, 0xD9
    ])

    temp_dir = tempfile.mkdtemp(prefix="thrillwiki_test_")
    temp_path = Path(temp_dir)

    test_photo_path = temp_path / "test_photo.jpg"
    test_avatar_path = temp_path / "test_avatar.jpg"

    test_photo_path.write_bytes(jpeg_bytes)
    test_avatar_path.write_bytes(jpeg_bytes)

    yield {
        "test_photo": str(test_photo_path),
        "test_avatar": str(test_avatar_path),
    }

    # Cleanup
    import shutil
    shutil.rmtree(temp_dir, ignore_errors=True)


@pytest.fixture
def test_park(auth_page: Page, live_server, test_images):
    """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(test_images["test_photo"])
    auth_page.get_by_role("button", name="Create Park").click()

    yield auth_page


@pytest.fixture
def test_ride(test_park: Page, live_server, test_images):
    """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(test_images["test_photo"])
    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 django.contrib.contenttypes.models import ContentType

    from apps.moderation.models import EditSubmission
    from apps.parks.models import Park

    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
    with contextlib.suppress(Exception):
        submission.delete()


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

    from apps.moderation.models import EditSubmission
    from apps.parks.models import Park

    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

    with contextlib.suppress(Exception):
        submission.delete()


@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 datetime import date, timedelta

    from tests.factories import ParkFactory

    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 datetime import date, timedelta

    from tests.factories import RideFactory

    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

    with contextlib.suppress(Exception):
        queue_item.delete()


@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

    with contextlib.suppress(Exception):
        operation.delete()


# =============================================================================
# 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