From 3aff591350d81f9ec632aabb55148457e0b2f6cb Mon Sep 17 00:00:00 2001 From: Angela Tran Date: Fri, 31 Jan 2025 17:56:47 +0000 Subject: [PATCH] test: unit test shows staff member cannot add EnrollmentFlow with agency --- tests/pytest/core/admin/test_enrollment.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tests/pytest/core/admin/test_enrollment.py b/tests/pytest/core/admin/test_enrollment.py index d30ed80481..d702697e9d 100644 --- a/tests/pytest/core/admin/test_enrollment.py +++ b/tests/pytest/core/admin/test_enrollment.py @@ -165,3 +165,24 @@ def test_has_add_permission( request = admin_user_request(user_type) assert flow_admin_model.has_add_permission(request) == expected + + @pytest.mark.parametrize("include_transit_agency", [True, False]) + def test_EnrollmentFlowForm_staff_member( + self, admin_user_request, flow_admin_model, model_TransitAgency, include_transit_agency + ): + request = admin_user_request() + + # get the Form class that's used in the admin add view as the user would see it + form_class = flow_admin_model.get_form(request) + + request.POST = dict( + system_name="testflow", + supported_enrollment_methods=[models.EnrollmentMethods.DIGITAL, models.EnrollmentMethods.IN_PERSON], + ) + + if include_transit_agency: + request.POST.update({"transit_agency": model_TransitAgency.id}) + + form = form_class(request.POST) + assert not form.errors + assert form.is_valid()