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