Coverage for tests / test_tutorial / test_relationship_attributes / test_read_relationships / test_tutorial001.py: 100%
15 statements
« prev ^ index » next coverage.py v7.13.0, created at 2026-01-06 21:09 +0000
1import importlib 1abcdefghi
2from types import ModuleType 1abcdefghi
4import pytest 1abcdefghi
5from sqlmodel import create_engine 1abcdefghi
7from ....conftest import PrintMock, needs_py310 1abcdefghi
10@pytest.fixture( 1abcdefghi
11 name="mod",
12 params=[
13 pytest.param("tutorial001_py39"),
14 pytest.param("tutorial001_py310", marks=needs_py310),
15 ],
16)
17def get_module(request: pytest.FixtureRequest) -> ModuleType: 1abcdefghi
18 mod = importlib.import_module( 1abcdefghi
19 f"docs_src.tutorial.relationship_attributes.read_relationships.{request.param}"
20 )
21 mod.sqlite_url = "sqlite://" 1abcdefghi
22 mod.engine = create_engine(mod.sqlite_url) 1abcdefghi
23 return mod 1abcdefghi
26expected_calls = [ 1abcdefghi
27 [
28 "Created hero:",
29 {
30 "age": None,
31 "id": 1,
32 "secret_name": "Dive Wilson",
33 "team_id": 1,
34 "name": "Deadpond",
35 },
36 ],
37 [
38 "Created hero:",
39 {
40 "age": 48,
41 "id": 2,
42 "secret_name": "Tommy Sharp",
43 "team_id": 2,
44 "name": "Rusty-Man",
45 },
46 ],
47 [
48 "Created hero:",
49 {
50 "age": None,
51 "id": 3,
52 "secret_name": "Pedro Parqueador",
53 "team_id": None,
54 "name": "Spider-Boy",
55 },
56 ],
57 [
58 "Updated hero:",
59 {
60 "age": None,
61 "id": 3,
62 "secret_name": "Pedro Parqueador",
63 "team_id": 2,
64 "name": "Spider-Boy",
65 },
66 ],
67 [
68 "Team Wakaland:",
69 {"headquarters": "Wakaland Capital City", "id": 3, "name": "Wakaland"},
70 ],
71 [
72 "Preventers new hero:",
73 {
74 "age": 32,
75 "id": 6,
76 "secret_name": "Natalia Roman-on",
77 "team_id": 2,
78 "name": "Tarantula",
79 },
80 ],
81 [
82 "Preventers new hero:",
83 {
84 "age": 36,
85 "id": 7,
86 "secret_name": "Steve Weird",
87 "team_id": 2,
88 "name": "Dr. Weird",
89 },
90 ],
91 [
92 "Preventers new hero:",
93 {
94 "age": 93,
95 "id": 8,
96 "secret_name": "Esteban Rogelios",
97 "team_id": 2,
98 "name": "Captain North America",
99 },
100 ],
101 [
102 "Spider-Boy's team:",
103 {"headquarters": "Sharp Tower", "id": 2, "name": "Preventers"},
104 ],
105 [
106 "Spider-Boy's team again:",
107 {"headquarters": "Sharp Tower", "id": 2, "name": "Preventers"},
108 ],
109]
112def test_tutorial(print_mock: PrintMock, mod: ModuleType): 1abcdefghi
113 mod.main() 1jklmnopqr
114 assert print_mock.calls == expected_calls 1jklmnopqr