fixed failing test

eadwinCode · Nov 8, 2023 · 8187b5a · 8187b5a
1 parent 36ef232
commit 8187b5a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/ninja_jwt/schema.py b/ninja_jwt/schema.py
@@ -135,11 +135,14 @@ def post_validate_schema(cls, values: Dict) -> dict:
 
         return values
 
+    def get_response_schema_init_kwargs(self) -> dict:
+        return dict(
+            self.dict(exclude={"password"}), **self.__dict__.get("token_data", {})
+        )
+
     def to_response_schema(self):
         _schema_type = self.get_response_schema()
-        return _schema_type(
-            **self.dict(exclude={"password"}), **self.__dict__.get("token_data", {})
-        )
+        return _schema_type(**self.get_response_schema_init_kwargs())
 
 
 class TokenObtainPairOutputSchema(AuthUserSchema):

diff --git a/tests/test_custom_schema.py b/tests/test_custom_schema.py
@@ -72,7 +72,7 @@ def to_response_schema(self):
         return MyNewObtainTokenSlidingSchemaOutput(
             first_name=self._user.first_name,
             last_name=self._user.last_name,
-            **self.dict(exclude={"password"}),
+            **self.get_response_schema_init_kwargs(),
         )