Add test

Author: marvinbuss

code/function/api/v1/api_v1.py

@@ -2,5 +2,5 @@
 from function.api.v1.endpoints import heartbeat, sample
 
 api_v1_router = APIRouter()
-api_v1_router.include_router(sample.router, prefix="/landingZone", tags=["sample"])
+api_v1_router.include_router(sample.router, prefix="/sample", tags=["sample"])
 api_v1_router.include_router(heartbeat.router, prefix="/health", tags=["health"])

code/function/api/v1/endpoints/sample.py

@@ -9,7 +9,7 @@
 router = APIRouter()
 
 
-@router.post("/create", response_model=SampleResponse, name="create")
+@router.post("/sample", response_model=SampleResponse, name="sample")
 async def post_predict(
     data: SampleRequest,
 ) -> SampleResponse:

tests/test_main.py

@@ -20,3 +20,16 @@ def test_get_heartbeat(client, version):
     # assert
     assert response.status_code == 200
     assert response.json() == {"isAlive": True}
+
+
+@pytest.mark.parametrize("version", ("v1",))
+def test_post_sample(client, version):
+    # arrange
+    path = f"/{version}/sample/sample"
+
+    # action
+    response = client.post(path, json={"input": "Test"})
+
+    # assert
+    assert response.status_code == 200
+    assert response.json() == {"output": "Hello Test"}