Coverage for tests / test_tutorial / test_connect / test_select / test_tutorial003.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
9expected_calls = [ 1abcdefghi
10 [
11 "Created hero:",
12 {
13 "age": None,
14 "id": 1,
15 "secret_name": "Dive Wilson",
16 "team_id": 2,
17 "name": "Deadpond",
18 },
19 ],
20 [
21 "Created hero:",
22 {
23 "age": 48,
24 "id": 2,
25 "secret_name": "Tommy Sharp",
26 "team_id": 1,
27 "name": "Rusty-Man",
28 },
29 ],
30 [
31 "Created hero:",
32 {
33 "age": None,
34 "id": 3,
35 "secret_name": "Pedro Parqueador",
36 "team_id": None,
37 "name": "Spider-Boy",
38 },
39 ],
40 [
41 "Hero:",
42 {
43 "age": None,
44 "id": 1,
45 "secret_name": "Dive Wilson",
46 "team_id": 2,
47 "name": "Deadpond",
48 },
49 "Team:",
50 {"id": 2, "name": "Z-Force", "headquarters": "Sister Margaret's Bar"},
51 ],
52 [
53 "Hero:",
54 {
55 "age": 48,
56 "id": 2,
57 "secret_name": "Tommy Sharp",
58 "team_id": 1,
59 "name": "Rusty-Man",
60 },
61 "Team:",
62 {"id": 1, "name": "Preventers", "headquarters": "Sharp Tower"},
63 ],
64 [
65 "Hero:",
66 {
67 "age": None,
68 "id": 3,
69 "secret_name": "Pedro Parqueador",
70 "team_id": None,
71 "name": "Spider-Boy",
72 },
73 "Team:",
74 None,
75 ],
76]
79@pytest.fixture( 1abcdefghi
80 name="module",
81 params=[
82 "tutorial003_py39",
83 pytest.param("tutorial003_py310", marks=needs_py310),
84 ],
85)
86def get_module(request: pytest.FixtureRequest) -> ModuleType: 1abcdefghi
87 module = importlib.import_module( 1abcdefghi
88 f"docs_src.tutorial.connect.select.{request.param}"
89 )
90 module.sqlite_url = "sqlite://" 1abcdefghi
91 module.engine = create_engine(module.sqlite_url) 1abcdefghi
92 return module 1abcdefghi
95def test_tutorial(print_mock: PrintMock, module: ModuleType): 1abcdefghi
96 module.main() 1jklmnopqr
97 assert print_mock.calls == expected_calls 1jklmnopqr