reformat code

This commit is contained in:
dongfangtianyu 2019-04-06 22:17:42 +08:00
parent 630cebe057
commit fcc59bcb64
2 changed files with 7 additions and 8 deletions

View File

@ -168,18 +168,18 @@ class TestGenerator(TestCase):
assert 'paths' in schema assert 'paths' in schema
def test_serializer_datefield(self): def test_serializer_datefield(self):
patterns = [ patterns = [
url(r'^example/?$', views.ExampleGenericViewSet.as_view({"get":"get"})), url(r'^example/?$', views.ExampleGenericViewSet.as_view({"get": "get"})),
] ]
generator = SchemaGenerator(patterns=patterns) generator = SchemaGenerator(patterns=patterns)
request = create_request('/') request = create_request('/')
schema = generator.get_schema(request=request) schema = generator.get_schema(request=request)
response_schema = schema['paths']['/example/']['get']['responses']['200']['content']['application/json']['schema']['properties'] response = schema['paths']['/example/']['get']['responses']
response_schema = response['200']['content']['application/json']['schema']['properties']
assert response_schema['date']['type'] == response_schema['datetime']['type']== 'string' assert response_schema['date']['type'] == response_schema['datetime']['type'] == 'string'
assert response_schema['date']['format'] == 'date' assert response_schema['date']['format'] == 'date'
assert response_schema['datetime']['format'] == 'date-time' assert response_schema['datetime']['format'] == 'date-time'

View File

@ -23,7 +23,6 @@ class ExampleDetailView(APIView):
class ExampleGenericViewSet(GenericViewSet): class ExampleGenericViewSet(GenericViewSet):
class ExampleSerializer(serializers.Serializer): class ExampleSerializer(serializers.Serializer):
date = serializers.DateField() date = serializers.DateField()
datetime = serializers.DateTimeField() datetime = serializers.DateTimeField()
@ -34,5 +33,5 @@ class ExampleGenericViewSet(GenericViewSet):
from datetime import datetime from datetime import datetime
now = datetime.now() now = datetime.now()
serializer = self.get_serializer(data=now.date(),datetime=now) serializer = self.get_serializer(data=now.date(), datetime=now)
return Response(serializer.data) return Response(serializer.data)