test customezed_schema

bojiang · Feb 25, 2021 · d1abb1b · d1abb1b
1 parent 3d910e5
commit d1abb1b
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/tests/integration/projects/general/service.py b/tests/integration/projects/general/service.py
@@ -70,6 +70,26 @@ def predict_json(self, input_datas):
     def customezed_route(self, input_datas):
         return input_datas
 
+    @bentoml.api(
+        input=JsonInput(
+            request_schema={
+                "application/json": {
+                    "schema": {
+                        "type": "object",
+                        "required": ["field1", "field2"],
+                        "properties": {
+                            "field1": {"type": "string"},
+                            "field2": {"type": "uuid"},
+                        },
+                    },
+                }
+            }
+        ),
+        batch=True,
+    )
+    def customezed_schema(self, input_datas):
+        return input_datas
+
     @bentoml.api(input=JsonInput(), batch=True)
     def predict_strict_json(self, input_datas, tasks: Sequence[InferenceTask] = None):
         filtered_jsons = []

diff --git a/tests/integration/projects/general_non_batch/tests/test_meta.py b/tests/integration/projects/general_non_batch/tests/test_meta.py
@@ -7,3 +7,13 @@ async def test_api_server_meta(host):
     await pytest.assert_request("GET", f"http://{host}/")
     await pytest.assert_request("GET", f"http://{host}/healthz")
     await pytest.assert_request("GET", f"http://{host}/docs.json")
+
+
+@pytest.mark.asyncio
+async def test_customized_request_schema(host):
+    await pytest.assert_request(
+        "GET",
+        f"http://{host}/docs.json",
+        headers=(("Content-Type", "application/json"),),
+        assert_data="",
+    )