benefits/tests/pytest/core/admin/test_transit.py at d832b6557b97a22431123db3e26ad7317f311a01 · cal-itp/benefits · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
import pytest
from django.contrib import admin
from django.contrib.auth.models import Group

from benefits.core import models
from benefits.core.admin.mixins import StaffPermissionMixin
from benefits.core.admin.transit import TransitAgencyAdmin


@pytest.mark.django_db
class TestTransitAgencyAdmin:
    @pytest.fixture(autouse=True)
    def init(self):
        self.model_admin = TransitAgencyAdmin(models.TransitAgency, admin.site)

    def test_permissions_mixin(self):
        assert isinstance(self.model_admin, StaffPermissionMixin)

    @pytest.mark.parametrize(
        "user_type,expected",
        [
            (
                "staff",
                ["sso_domain", "customer_service_group"],
            ),
            ("super", ["customer_service_group"]),
        ],
    )
    def test_get_exclude(self, admin_user_request, user_type, expected):
        if expected:
            model_fields = [f.name for f in self.model_admin.model._meta.get_fields()]
            assert all(field in model_fields for field in expected)

        request = admin_user_request(user_type)

        excluded = self.model_admin.get_exclude(request)

        if expected:
            assert set(excluded) == set(expected)
        else:
            assert excluded is None

    @pytest.mark.parametrize(
        "user_type,expected",
        [
            ("staff", ["sso_domain"]),
            ("super", []),
        ],
    )
    def test_get_exclude_modify_agency(self, admin_user_request, user_type, expected, model_TransitAgency):
        obj = model_TransitAgency
        request = admin_user_request(user_type)
        excluded = self.model_admin.get_exclude(request, obj)

        if expected:
            assert set(excluded) == set(expected)
        else:
            assert excluded is None

    def test_save_model_create_agency(self, admin_user_request, model_TransitAgency):
        request = admin_user_request("super")
        obj = model_TransitAgency
        form = None
        change = False  # Simulate creating a new agency
        initial_group_count = Group.objects.count()

        self.model_admin.save_model(request, obj, form, change)

        assert obj.customer_service_group.name == f"{obj.short_name} Customer Service"
        assert Group.objects.count() == initial_group_count + 1

    def test_save_model_modify_agency_not_short_name(self, mocker, admin_user_request, model_TransitAgency):
        request = admin_user_request("super")
        model_TransitAgency.customer_service_group = Group.objects.create(name="Existing Customer Service Group")
        obj = model_TransitAgency
        mock_form = mocker.Mock()
        mock_form.changed_data = ["info_url"]
        change = True  # Simulate modifying a field that is not the short_name of an existing agency
        initial_group_count = Group.objects.count()

        self.model_admin.save_model(request, obj, mock_form, change)

        assert obj.customer_service_group.name == "Existing Customer Service Group"
        assert Group.objects.count() == initial_group_count

    def test_save_model_modify_agency_short_name(self, mocker, admin_user_request, model_TransitAgency):
        request = admin_user_request("super")
        model_TransitAgency.customer_service_group = Group.objects.create(name="Existing Customer Service")
        obj = model_TransitAgency
        mock_form = mocker.Mock()
        mock_form.changed_data = ["short_name"]
        change = True  # Simulate modifying the short_name of an existing agency
        initial_group_count = Group.objects.count()

        self.model_admin.save_model(request, obj, mock_form, change)

        assert obj.customer_service_group.name == f"{obj.short_name} Customer Service"
        assert Group.objects.count() == initial_group_count