Add skip to management commands which depend on coreapi

Author: math-a3k

tests/schemas/test_managementcommand.py

@@ -8,7 +8,7 @@
 from django.test.utils import override_settings
 from django.urls import path
 
-from rest_framework.compat import uritemplate, yaml
+from rest_framework.compat import coreapi, uritemplate, yaml
 from rest_framework.management.commands import generateschema
 from rest_framework.utils import formatting, json
 from rest_framework.views import APIView
@@ -35,20 +35,23 @@ def get_schema(self, **kwargs):
 
 
 @override_settings(ROOT_URLCONF=__name__)
+@pytest.mark.skipif(not coreapi, reason='coreapi is not installed')
 @pytest.mark.skipif(not uritemplate, reason='uritemplate is not installed')
 class GenerateSchemaTests(TestCase):
     """Tests for management command generateschema."""
 
     def setUp(self):
         self.out = io.StringIO()
 
+    @pytest.mark.skipif(not coreapi, reason='coreapi is not installed')
     def test_command_detects_schema_generation_mode(self):
         """Switching between CoreAPI & OpenAPI"""
         command = generateschema.Command()
         assert command.get_mode() == generateschema.OPENAPI_MODE
         with override_settings(REST_FRAMEWORK={'DEFAULT_SCHEMA_CLASS': 'rest_framework.schemas.AutoSchema'}):
             assert command.get_mode() == generateschema.COREAPI_MODE
 
+    @pytest.mark.skipif(not coreapi, reason='coreapi is not installed')
     @pytest.mark.skipif(yaml is None, reason='PyYAML is required.')
     def test_renders_default_schema_with_custom_title_url_and_description(self):
         call_command('generateschema',
@@ -91,6 +94,7 @@ def test_writes_schema_to_file_on_parameter(self):
             os.remove(path)
 
     @pytest.mark.skipif(yaml is None, reason='PyYAML is required.')
+    @pytest.mark.skipif(not coreapi, reason='coreapi is not installed')
     @override_settings(REST_FRAMEWORK={'DEFAULT_SCHEMA_CLASS': 'rest_framework.schemas.AutoSchema'})
     def test_coreapi_renders_default_schema_with_custom_title_url_and_description(self):
         expected_out = """info:
@@ -113,6 +117,7 @@ def test_coreapi_renders_default_schema_with_custom_title_url_and_description(se
 
         self.assertIn(formatting.dedent(expected_out), self.out.getvalue())
 
+    @pytest.mark.skipif(not coreapi, reason='coreapi is not installed')
     @override_settings(REST_FRAMEWORK={'DEFAULT_SCHEMA_CLASS': 'rest_framework.schemas.AutoSchema'})
     def test_coreapi_renders_openapi_json_schema(self):
         expected_out = {
@@ -142,6 +147,7 @@ def test_coreapi_renders_openapi_json_schema(self):
 
         self.assertDictEqual(out_json, expected_out)
 
+    @pytest.mark.skipif(not coreapi, reason='coreapi is not installed')
     @override_settings(REST_FRAMEWORK={'DEFAULT_SCHEMA_CLASS': 'rest_framework.schemas.AutoSchema'})
     def test_renders_corejson_schema(self):
         expected_out = """{"_type":"document","":{"list":{"_type":"link","url":"/","action":"get"}}}"""