Custom exception handler with TestClient cannot access request data

[LaundroMat]

I have this custom exception handler for PermissionDenied errors:

@api.exception_handler(PermissionDenied)
def permission_error(request, exc):
    if 'team_id' in request.resolver_match.captured_kwargs.keys():
        message = "You do not have access to this team."
    else:
        message = "Forbidden."

    return api.create_response(
        request,
        {"detail": message},
        status=403,
    )

I'd like to customize the error messages based on information in the request. But using the TestClient, the request is a Mock instance, so I get an TypeError: argument of type 'Mock' is not iterable when trying to access the request's captured kwargs. Is there a way around this?

[vitalik]

@LaundroMat

I guess you can override the TestClient like this:

class MyTestClient(TestClient):
    def _call(self, func, request, kwargs):
        request.resolver_match.captured_kwargs.keys = Mock(return_value=['team_id'])
        return super()._call(func, request, kwargs)

[LaundroMat]

Thanks for the pointer. In the end I'm returning custom error messages from within the views, it makes things a bit more manageable :)