File: test_security_scheme.py

package info (click to toggle)
openapi-pydantic 0.5.1-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 744 kB
  • sloc: python: 4,392; makefile: 4
file content (46 lines) | stat: -rw-r--r-- 1,926 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from openapi_pydantic import SecurityScheme
from openapi_pydantic.compat import PYDANTIC_V2


def test_oidc_parsing() -> None:
    security_scheme_1 = SecurityScheme(
        type="openIdConnect", openIdConnectUrl="https://example.com/openIdConnect"
    )
    assert isinstance(security_scheme_1.openIdConnectUrl, str)
    dump1 = getattr(security_scheme_1, "model_dump_json" if PYDANTIC_V2 else "json")
    if PYDANTIC_V2:
        assert dump1(by_alias=True, exclude_none=True) == (
            '{"type":"openIdConnect","openIdConnectUrl":"https://example.com/openIdConnect"}'
        )
    else:
        assert dump1(by_alias=True, exclude_none=True) == (
            '{"type": "openIdConnect", "openIdConnectUrl": "https://example.com/openIdConnect"}'
        )

    security_scheme_2 = SecurityScheme(
        type="openIdConnect", openIdConnectUrl="/openIdConnect"
    )
    assert isinstance(security_scheme_2.openIdConnectUrl, str)
    dump2 = getattr(security_scheme_2, "model_dump_json" if PYDANTIC_V2 else "json")
    if PYDANTIC_V2:
        assert dump2(by_alias=True, exclude_none=True) == (
            '{"type":"openIdConnect","openIdConnectUrl":"/openIdConnect"}'
        )
    else:
        assert dump2(by_alias=True, exclude_none=True) == (
            '{"type": "openIdConnect", "openIdConnectUrl": "/openIdConnect"}'
        )

    security_scheme_3 = SecurityScheme(
        type="openIdConnect", openIdConnectUrl="openIdConnect"
    )
    assert isinstance(security_scheme_3.openIdConnectUrl, str)
    dump3 = getattr(security_scheme_3, "model_dump_json" if PYDANTIC_V2 else "json")
    if PYDANTIC_V2:
        assert dump3(by_alias=True, exclude_none=True) == (
            '{"type":"openIdConnect","openIdConnectUrl":"openIdConnect"}'
        )
    else:
        assert dump3(by_alias=True, exclude_none=True) == (
            '{"type": "openIdConnect", "openIdConnectUrl": "openIdConnect"}'
        )