thrillwiki_django_no_react/backend/apps/rides/tests/test_admin.py

"""
Tests for rides admin interfaces.

These tests verify the functionality of ride, model, stats, company,
review, and ranking admin classes including query optimization and custom actions.
"""

import pytest
from django.contrib.admin.sites import AdminSite
from django.contrib.auth import get_user_model
from django.test import RequestFactory, TestCase

from apps.rides.admin import (
    CompanyAdmin,
    RankingSnapshotAdmin,
    RideAdmin,
    RideLocationAdmin,
    RideModelAdmin,
    RidePairComparisonAdmin,
    RideRankingAdmin,
    RideReviewAdmin,
    RollerCoasterStatsAdmin,
)
from apps.rides.models.company import Company
from apps.rides.models.location import RideLocation
from apps.rides.models.rankings import RankingSnapshot, RidePairComparison, RideRanking
from apps.rides.models.reviews import RideReview
from apps.rides.models.rides import Ride, RideModel, RollerCoasterStats

User = get_user_model()


class TestRideAdmin(TestCase):
    """Tests for RideAdmin class."""

    def setUp(self):
        self.factory = RequestFactory()
        self.site = AdminSite()
        self.admin = RideAdmin(model=Ride, admin_site=self.site)

    def test_list_display_fields(self):
        """Verify all required fields are in list_display."""
        required_fields = [
            "name",
            "park_link",
            "category_badge",
            "manufacturer_link",
            "status_badge",
        ]
        for field in required_fields:
            assert field in self.admin.list_display

    def test_list_select_related(self):
        """Verify select_related is configured for ForeignKeys."""
        assert "park" in self.admin.list_select_related
        assert "manufacturer" in self.admin.list_select_related
        assert "designer" in self.admin.list_select_related
        assert "ride_model" in self.admin.list_select_related

    def test_list_prefetch_related(self):
        """Verify prefetch_related is configured for reverse relations."""
        assert "reviews" in self.admin.list_prefetch_related

    def test_export_fields_configured(self):
        """Verify export fields are configured."""
        assert hasattr(self.admin, "export_fields")
        assert "id" in self.admin.export_fields
        assert "name" in self.admin.export_fields
        assert "category" in self.admin.export_fields

    def test_status_actions_registered(self):
        """Verify status change actions are registered."""
        request = self.factory.get("/admin/")
        request.user = User(is_superuser=True)

        actions = self.admin.get_actions(request)
        assert "bulk_set_operating" in actions
        assert "bulk_set_closed" in actions
        assert "bulk_set_sbno" in actions


class TestRideModelAdmin(TestCase):
    """Tests for RideModelAdmin class."""

    def setUp(self):
        self.factory = RequestFactory()
        self.site = AdminSite()
        self.admin = RideModelAdmin(model=RideModel, admin_site=self.site)

    def test_list_select_related(self):
        """Verify select_related for manufacturer."""
        assert "manufacturer" in self.admin.list_select_related

    def test_list_prefetch_related(self):
        """Verify prefetch_related for rides."""
        assert "rides" in self.admin.list_prefetch_related


class TestRollerCoasterStatsAdmin(TestCase):
    """Tests for RollerCoasterStatsAdmin class."""

    def setUp(self):
        self.factory = RequestFactory()
        self.site = AdminSite()
        self.admin = RollerCoasterStatsAdmin(model=RollerCoasterStats, admin_site=self.site)

    def test_list_select_related(self):
        """Verify select_related for ride and park."""
        assert "ride" in self.admin.list_select_related
        assert "ride__park" in self.admin.list_select_related
        assert "ride__manufacturer" in self.admin.list_select_related


class TestRideReviewAdmin(TestCase):
    """Tests for RideReviewAdmin class."""

    def setUp(self):
        self.factory = RequestFactory()
        self.site = AdminSite()
        self.admin = RideReviewAdmin(model=RideReview, admin_site=self.site)

    def test_list_select_related(self):
        """Verify select_related for ride, park, and user."""
        assert "ride" in self.admin.list_select_related
        assert "ride__park" in self.admin.list_select_related
        assert "user" in self.admin.list_select_related
        assert "moderated_by" in self.admin.list_select_related

    def test_moderation_actions_registered(self):
        """Verify moderation actions are registered."""
        request = self.factory.get("/admin/")
        request.user = User(is_superuser=True)

        actions = self.admin.get_actions(request)
        assert "bulk_approve" in actions
        assert "bulk_reject" in actions
        assert "flag_for_review" in actions


class TestRideRankingAdmin(TestCase):
    """Tests for RideRankingAdmin class."""

    def setUp(self):
        self.factory = RequestFactory()
        self.site = AdminSite()
        self.admin = RideRankingAdmin(model=RideRanking, admin_site=self.site)

    def test_readonly_permissions(self):
        """Verify read-only permissions are set."""
        request = self.factory.get("/admin/")
        request.user = User(is_superuser=False)

        assert self.admin.has_add_permission(request) is False
        assert self.admin.has_change_permission(request) is False

    def test_list_select_related(self):
        """Verify select_related for ride and park."""
        assert "ride" in self.admin.list_select_related
        assert "ride__park" in self.admin.list_select_related


class TestRidePairComparisonAdmin(TestCase):
    """Tests for RidePairComparisonAdmin class."""

    def setUp(self):
        self.factory = RequestFactory()
        self.site = AdminSite()
        self.admin = RidePairComparisonAdmin(model=RidePairComparison, admin_site=self.site)

    def test_readonly_permissions(self):
        """Verify read-only permissions are set."""
        request = self.factory.get("/admin/")
        request.user = User(is_superuser=False)

        assert self.admin.has_add_permission(request) is False
        assert self.admin.has_change_permission(request) is False

    def test_list_select_related(self):
        """Verify select_related for both rides."""
        assert "ride_a" in self.admin.list_select_related
        assert "ride_b" in self.admin.list_select_related


class TestRankingSnapshotAdmin(TestCase):
    """Tests for RankingSnapshotAdmin class."""

    def setUp(self):
        self.factory = RequestFactory()
        self.site = AdminSite()
        self.admin = RankingSnapshotAdmin(model=RankingSnapshot, admin_site=self.site)

    def test_readonly_permissions(self):
        """Verify read-only permissions are set."""
        request = self.factory.get("/admin/")
        request.user = User(is_superuser=False)

        assert self.admin.has_add_permission(request) is False
        assert self.admin.has_change_permission(request) is False


class TestCompanyAdmin(TestCase):
    """Tests for rides CompanyAdmin class."""

    def setUp(self):
        self.factory = RequestFactory()
        self.site = AdminSite()
        self.admin = CompanyAdmin(model=Company, admin_site=self.site)

    def test_list_prefetch_related(self):
        """Verify prefetch_related for manufactured rides."""
        assert "manufactured_rides" in self.admin.list_prefetch_related
        assert "designed_rides" in self.admin.list_prefetch_related