diff --git a/tests/pass_through_unit_tests/test_pass_through_unit_tests.py b/tests/pass_through_unit_tests/test_pass_through_unit_tests.py index 0e681cb1e0..f529fe85ab 100644 --- a/tests/pass_through_unit_tests/test_pass_through_unit_tests.py +++ b/tests/pass_through_unit_tests/test_pass_through_unit_tests.py @@ -68,6 +68,8 @@ def mock_request(): self.request_body = request_body or {} # Add url attribute that the actual code expects self.url = "http://localhost:8000/test" + # Add state attribute that FastAPI requests have + self.state = type("State", (), {})() async def body(self) -> bytes: return bytes(json.dumps(self.request_body), "utf-8")