added support for custom json content types

manchenkoff · Sep 14, 2023 · 9786060 · 9786060
1 parent aa3cd66
commit 9786060
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/src/openapi_parser/enumeration.py b/src/openapi_parser/enumeration.py
@@ -97,6 +97,8 @@ class CookieParameterStyle(Enum):
 @unique
 class ContentType(Enum):
     JSON = 'application/json'
+    JSON_TEXT = 'text/json'
+    JSON_ANY = 'application/*+json'
     XML = 'application/xml'
     FORM = 'application/x-www-form-urlencoded'
     MULTIPART_FORM = 'multipart/form-data'

diff --git a/tests/builders/test_content_builder.py b/tests/builders/test_content_builder.py
@@ -31,6 +31,19 @@ def _get_schema_factory_mock(expected_value: Schema) -> SchemaFactory:
         ],
         _get_schema_factory_mock(string_schema)
     ),
+    (
+        {
+            "text/json": {
+                "schema": {
+                    "type": "string"
+                }
+            }
+        },
+        [
+            Content(type=ContentType.JSON_TEXT, schema=string_schema)
+        ],
+        _get_schema_factory_mock(string_schema)
+    ),
 )
 
 

diff --git a/tests/test_enumeration.py b/tests/test_enumeration.py
@@ -139,6 +139,8 @@ def test_parameter_location_error() -> None:
 
 media_type_provider = (
     ("application/json", ContentType.JSON),
+    ("application/*+json", ContentType.JSON_ANY),
+    ("text/json", ContentType.JSON_TEXT),
     ("application/xml", ContentType.XML),
     ("application/x-www-form-urlencoded", ContentType.FORM),
     ("multipart/form-data", ContentType.MULTIPART_FORM),