diff --git a/tests/pytest/core/test_views.py b/tests/pytest/core/test_views.py index e096edfb5..ef55b1a2b 100644 --- a/tests/pytest/core/test_views.py +++ b/tests/pytest/core/test_views.py @@ -11,6 +11,8 @@ TEMPLATE_AGENCY, bad_request, csrf_failure, + page_not_found, + server_error, ) from benefits.core.middleware import ROUTE_INDEX @@ -177,6 +179,20 @@ def test_not_found_no_active_agency(mocker, client, mocked_session_update): assert mocked_session_update.call_args.kwargs["origin"] == reverse(ROUTE_INDEX) +@pytest.mark.django_db +def test_page_not_found(app_request): + response = page_not_found(app_request, Exception()) + + assert response.status_code == 404 + + +@pytest.mark.django_db +def test_server_error(app_request): + response = server_error(app_request) + + assert response.status_code == 500 + + @pytest.mark.django_db def test_logged_out(client): path = reverse(ROUTE_LOGGED_OUT)