diff --git a/papi/main.py b/papi/main.py index 23920b7..028ad74 100644 --- a/papi/main.py +++ b/papi/main.py @@ -1,10 +1,16 @@ from typing import Optional +from pydantic import BaseModel from fastapi import FastAPI app = FastAPI() +class Sonde(BaseModel): + identifiant: str + nom: str + + @app.get("/") def read_root(): return {"msg": "Hello World"} @@ -13,3 +19,9 @@ def read_root(): @app.get("/items/{item_id}") def read_item(item_id: int, q: Optional[str] = None): return {"item_id": item_id, "q": q} + + +@app.post("/sonde/") +def post_sonde(sonde: Sonde): + print(sonde) + return diff --git a/tests/test_papi.py b/tests/test_papi.py index 22a79eb..255afd5 100644 --- a/tests/test_papi.py +++ b/tests/test_papi.py @@ -16,3 +16,9 @@ def test_read_main(): response = client.get("/") assert response.status_code == 200 assert response.json() == {"msg": "Hello World"} + + +def test_creer_sonde(): + sonde1 = {"identifiant": "masonde_001", "nom": "client1"} + response = client.post("/sonde/", json=sonde1) + assert response.status_code == 200