Skip to content
Snippets Groups Projects
Commit 9cd98c18 authored by Erik Reid's avatar Erik Reid
Browse files

use pydantic for version, added a test

parent eb4f99c7
Branches
Tags
No related merge requests found
from importlib.metadata import PackageNotFoundError, version from importlib.metadata import version
from fastapi import APIRouter from fastapi import APIRouter
from pydantic import BaseModel
router = APIRouter() router = APIRouter()
class Version(BaseModel):
module: str
api: str
@router.get("/version") @router.get("/version")
def get_version() -> dict[str, str]: def get_version() -> Version:
"""Get the version of the package.""" """
try: handler for /version
return {"version": version("mapping_provider")} """
except PackageNotFoundError: return Version(
return {"version": "unknown"} module = version('mapping_provider'),
api = '0.1'
)
from fastapi.testclient import TestClient
import pytest
from mapping_provider import create_app
@pytest.fixture
def client():
return TestClient(create_app())
from mapping_provider.api.common import Version
def test_version(client):
rv = client.get("/version")
assert rv.status_code == 200
assert rv.json()
Version.model_validate(rv.json())
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment