encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-04 12:30:11 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add test for new setting

Browse Source
This commit is contained in:
Carlton Gibson 2017-12-05 09:09:07 +01:00
parent 4a200d5e66
commit 4b09e82402
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/test_schemas.py
View File

@ -516,6 +516,11 @@ class Test4605Regression(TestCase):
assert prefix == '/' assert prefix == '/'
class CustomViewInspector(AutoSchema):
"""A dummy AutoSchema subclass"""
pass
class TestAutoSchema(TestCase): class TestAutoSchema(TestCase):
def test_apiview_schema_descriptor(self): def test_apiview_schema_descriptor(self):
@ -523,6 +528,18 @@ class TestAutoSchema(TestCase):
assert hasattr(view, 'schema') assert hasattr(view, 'schema')
assert isinstance(view.schema, AutoSchema) assert isinstance(view.schema, AutoSchema)
def test_set_custom_inspector_class_on_view(self):
class CustomView(APIView):
schema = CustomViewInspector()
view = CustomView()
assert isinstance(view.schema, CustomViewInspector)
def test_set_custom_inspector_class_via_settings(self):
with override_settings(REST_FRAMEWORK={'DEFAULT_SCHEMA_CLASS':'tests.test_schemas.CustomViewInspector'}):
view = APIView()
assert isinstance(view.schema, CustomViewInspector)
def test_get_link_requires_instance(self): def test_get_link_requires_instance(self):
descriptor = APIView.schema # Accessed from class descriptor = APIView.schema # Accessed from class
with pytest.raises(AssertionError): with pytest.raises(AssertionError):