diff --git a/InvenTree/InvenTree/unit_test.py b/InvenTree/InvenTree/unit_test.py index 05e5bbe1d25..45c8d40aa37 100644 --- a/InvenTree/InvenTree/unit_test.py +++ b/InvenTree/InvenTree/unit_test.py @@ -6,9 +6,11 @@ import re from contextlib import contextmanager from pathlib import Path +from typing import Any from django.contrib.auth import get_user_model from django.contrib.auth.models import Group, Permission +from django.core.handlers.wsgi import WSGIRequest from django.db import connections from django.http.response import StreamingHttpResponse from django.test import TestCase @@ -20,6 +22,8 @@ from plugin import registry from plugin.models import PluginConfig +MyResponse = Any | WSGIRequest + def addUserPermission(user, permission): """Shortcut function for adding a certain permission to a user.""" @@ -302,7 +306,7 @@ def get(self, url, data=None, expected_code=200, format='json', **kwargs): if data is None: data = {} - response = self.client.get(url, data, format=format, **kwargs) + response: MyResponse = self.client.get(url, data, format=format, **kwargs) self.checkResponse(url, 'GET', expected_code, response) @@ -314,7 +318,7 @@ def post(self, url, data=None, expected_code=None, format='json', **kwargs): if data is None: data = {} - response = self.client.post(url, data=data, format=format, **kwargs) + response: MyResponse = self.client.post(url, data=data, format=format, **kwargs) self.checkResponse(url, 'POST', expected_code, response) @@ -325,7 +329,9 @@ def delete(self, url, data=None, expected_code=None, format='json', **kwargs): if data is None: data = {} - response = self.client.delete(url, data=data, format=format, **kwargs) + response: MyResponse = self.client.delete( + url, data=data, format=format, **kwargs + ) self.checkResponse(url, 'DELETE', expected_code, response) @@ -333,7 +339,9 @@ def delete(self, url, data=None, expected_code=None, format='json', **kwargs): def patch(self, url, data, expected_code=None, format='json', **kwargs): """Issue a PATCH request.""" - response = self.client.patch(url, data=data, format=format, **kwargs) + response: MyResponse = self.client.patch( + url, data=data, format=format, **kwargs + ) self.checkResponse(url, 'PATCH', expected_code, response) @@ -341,7 +349,7 @@ def patch(self, url, data, expected_code=None, format='json', **kwargs): def put(self, url, data, expected_code=None, format='json', **kwargs): """Issue a PUT request.""" - response = self.client.put(url, data=data, format=format, **kwargs) + response: MyResponse = self.client.put(url, data=data, format=format, **kwargs) self.checkResponse(url, 'PUT', expected_code, response) @@ -359,7 +367,7 @@ def download_file( self, url, data, expected_code=None, expected_fn=None, decode=True ): """Download a file from the server, and return an in-memory file.""" - response = self.client.get(url, data=data, format='json') + response: MyResponse = self.client.get(url, data=data, format='json') self.checkResponse(url, 'DOWNLOAD_FILE', expected_code, response)