Skip to content

Commit 659469c

Browse files
committed
Update red test to use field_to_swagger_object
1 parent b064a64 commit 659469c

File tree

2 files changed

+32
-15
lines changed

2 files changed

+32
-15
lines changed

tests/test_annotations.py

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
from typing import TYPE_CHECKING
2+
3+
from rest_framework import serializers
4+
5+
from drf_yasg import openapi
6+
from drf_yasg.inspectors.field import SerializerMethodFieldInspector
7+
from drf_yasg.openapi import ReferenceResolver
8+
9+
if TYPE_CHECKING:
10+
from uuid import UUID
11+
12+
13+
def test_missing_runtime_annotations():
14+
class UserSerializer(serializers.Serializer):
15+
uuid = serializers.SerializerMethodField()
16+
17+
def get_uuid(self, obj) -> "UUID":
18+
return obj.uuid
19+
20+
field = UserSerializer().fields["uuid"]
21+
components = ReferenceResolver("definitions", "parameters", force_init=True)
22+
23+
inspector = SerializerMethodFieldInspector(
24+
view=None,
25+
path="/",
26+
method="GET",
27+
components=components,
28+
request=None,
29+
field_inspectors=[],
30+
)
31+
32+
inspector.field_to_swagger_object(field, openapi.Schema, True)

tests/test_type_checking_issue.py

Lines changed: 0 additions & 15 deletions
This file was deleted.

0 commit comments

Comments
 (0)