add SerializerMethodField self-reference hint test

tfranzel · Oct 1, 2024 · 96a4f38 · 96a4f38
1 parent 9f0d166
commit 96a4f38
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/tests/test_regressions.py b/tests/test_regressions.py
@@ -3421,3 +3421,26 @@ class XViewSet(viewsets.ReadOnlyModelViewSet):
     schema = generate_schema('/x', XViewSet)
     assert schema['paths']['/x/']['get']["operationId"] == 'list_x'
     assert schema['paths']['/x/{id}/']['get']["operationId"] == 'retrieve_x'
+
+
+class SelfReferentialSerializer(serializers.Serializer):
+    field_int = serializers.IntegerField()
+    field_self = serializers.SerializerMethodField()
+
+    def get_field_self(self) -> "SelfReferentialSerializer":
+        return SelfReferentialSerializer()  # pragma: no cover
+
+
+def test_self_referential_serializer_method_field(no_warnings):
+    class XViewset(viewsets.ModelViewSet):
+        serializer_class = SelfReferentialSerializer
+        queryset = SimpleModel.objects.all()
+
+    schema = generate_schema('/x', XViewset)
+    assert schema['components']['schemas']['SelfReferential']['properties'] == {
+        'field_int': {'type': 'integer'},
+        'field_self': {
+            'allOf': [{'$ref': '#/components/schemas/SelfReferential'}],
+            'readOnly': True
+        }
+    }