Home / File/ test_union_body_discriminator.py — fastapi Source File

test_union_body_discriminator.py — fastapi Source File

Architecture documentation for test_union_body_discriminator.py, a python file in the fastapi codebase. 6 imports, 0 dependents.

File python FastAPI Responses 6 imports 1 functions 2 classes

Entity Profile

Dependency Diagram

graph LR
  b7547d18_a665_12ea_3e49_b878c6e6d305["test_union_body_discriminator.py"]
  0dda2280_3359_8460_301c_e98c77e78185["typing"]
  b7547d18_a665_12ea_3e49_b878c6e6d305 --> 0dda2280_3359_8460_301c_e98c77e78185
  534f6e44_61b8_3c38_8b89_6934a6df9802["__init__.py"]
  b7547d18_a665_12ea_3e49_b878c6e6d305 --> 534f6e44_61b8_3c38_8b89_6934a6df9802
  a7c04dee_ee23_5891_b185_47ff6bed036d["testclient.py"]
  b7547d18_a665_12ea_3e49_b878c6e6d305 --> a7c04dee_ee23_5891_b185_47ff6bed036d
  a7f4e7b0_9725_db90_5cbc_7ca8211b323a["inline_snapshot"]
  b7547d18_a665_12ea_3e49_b878c6e6d305 --> a7f4e7b0_9725_db90_5cbc_7ca8211b323a
  6913fbd4_39df_d14b_44bb_522e99b65b90["pydantic"]
  b7547d18_a665_12ea_3e49_b878c6e6d305 --> 6913fbd4_39df_d14b_44bb_522e99b65b90
  87f0eda4_1c7f_c164_42ba_f715b8cf0a6b["typing_extensions"]
  b7547d18_a665_12ea_3e49_b878c6e6d305 --> 87f0eda4_1c7f_c164_42ba_f715b8cf0a6b
  style b7547d18_a665_12ea_3e49_b878c6e6d305 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

from typing import Annotated, Any, Union

from fastapi import FastAPI
from fastapi.testclient import TestClient
from inline_snapshot import snapshot
from pydantic import BaseModel, Field
from typing_extensions import Literal


def test_discriminator_pydantic_v2() -> None:
    from pydantic import Tag

    app = FastAPI()

    class FirstItem(BaseModel):
        value: Literal["first"]
        price: int

    class OtherItem(BaseModel):
        value: Literal["other"]
        price: float

    Item = Annotated[
        Union[Annotated[FirstItem, Tag("first")], Annotated[OtherItem, Tag("other")]],
        Field(discriminator="value"),
    ]

    @app.post("/items/")
    def save_union_body_discriminator(
        item: Item, q: Annotated[str, Field(description="Query string")]
    ) -> dict[str, Any]:
        return {"item": item}

    client = TestClient(app)
    response = client.post("/items/?q=first", json={"value": "first", "price": 100})
    assert response.status_code == 200, response.text
    assert response.json() == {"item": {"value": "first", "price": 100}}

    response = client.post("/items/?q=other", json={"value": "other", "price": 100.5})
    assert response.status_code == 200, response.text
    assert response.json() == {"item": {"value": "other", "price": 100.5}}

    response = client.get("/openapi.json")
    assert response.status_code == 200, response.text
    assert response.json() == snapshot(
        {
            "openapi": "3.1.0",
            "info": {"title": "FastAPI", "version": "0.1.0"},
            "paths": {
                "/items/": {
                    "post": {
                        "summary": "Save Union Body Discriminator",
                        "operationId": "save_union_body_discriminator_items__post",
                        "parameters": [
                            {
                                "name": "q",
                                "in": "query",
                                "required": True,
                                "schema": {
                                    "type": "string",
// ... (117 more lines)

Domain

Subdomains

Dependencies

Frequently Asked Questions

What does test_union_body_discriminator.py do?
test_union_body_discriminator.py is a source file in the fastapi codebase, written in python. It belongs to the FastAPI domain, Responses subdomain.
What functions are defined in test_union_body_discriminator.py?
test_union_body_discriminator.py defines 1 function(s): test_discriminator_pydantic_v2.
What does test_union_body_discriminator.py depend on?
test_union_body_discriminator.py imports 6 module(s): __init__.py, inline_snapshot, pydantic, testclient.py, typing, typing_extensions.
Where is test_union_body_discriminator.py in the architecture?
test_union_body_discriminator.py is located at tests/test_union_body_discriminator.py (domain: FastAPI, subdomain: Responses, directory: tests).

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free