GitHub-Proxy/sqlmodel-fix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

✅ Add tests

Browse Source
This commit is contained in:
Sebastián Ramírez
2021-08-24 15:15:26 +02:00
parent 362f2e2e17
commit 79089f7a6c
114 changed files with 6742 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
tests/test_tutorial/test_many_to_many/__init__.py Normal file
View File

53
tests/test_tutorial/test_many_to_many/test_tutorial001.py Normal file
View File

@@ -0,0 +1,53 @@
from typing import Any, Dict, List, Union
from unittest.mock import patch
from sqlalchemy import inspect
from sqlalchemy.engine.reflection import Inspector
from sqlmodel import create_engine
from sqlmodel.pool import StaticPool
from ...conftest import get_testing_print_function
expected_calls = [
[
"Deadpond:",
{"id": 1, "secret_name": "Dive Wilson", "age": None, "name": "Deadpond"},
],
[
"Deadpond teams:",
[
{"id": 1, "name": "Z-Force", "headquarters": "Sister Margaret’s Bar"},
{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"},
],
],
[
"Rusty-Man:",
{"id": 2, "secret_name": "Tommy Sharp", "age": 48, "name": "Rusty-Man"},
],
[
"Rusty-Man Teams:",
[{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"}],
],
[
"Spider-Boy:",
{"id": 3, "secret_name": "Pedro Parqueador", "age": None, "name": "Spider-Boy"},
],
[
"Spider-Boy Teams:",
[{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"}],
],
]
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.many_to_many import tutorial001 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

80
tests/test_tutorial/test_many_to_many/test_tutorial002.py Normal file
View File

@@ -0,0 +1,80 @@
from typing import Any, Dict, List, Union
from unittest.mock import patch
from sqlalchemy import inspect
from sqlalchemy.engine.reflection import Inspector
from sqlmodel import create_engine
from sqlmodel.pool import StaticPool
from ...conftest import get_testing_print_function
expected_calls = [
[
"Deadpond:",
{"id": 1, "secret_name": "Dive Wilson", "age": None, "name": "Deadpond"},
],
[
"Deadpond teams:",
[
{"id": 1, "name": "Z-Force", "headquarters": "Sister Margaret’s Bar"},
{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"},
],
],
[
"Rusty-Man:",
{"id": 2, "secret_name": "Tommy Sharp", "age": 48, "name": "Rusty-Man"},
],
[
"Rusty-Man Teams:",
[{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"}],
],
[
"Spider-Boy:",
{"id": 3, "secret_name": "Pedro Parqueador", "age": None, "name": "Spider-Boy"},
],
[
"Spider-Boy Teams:",
[{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"}],
],
[
"Updated Spider-Boy's Teams:",
[
{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"},
{"id": 1, "name": "Z-Force", "headquarters": "Sister Margaret’s Bar"},
],
],
[
"Z-Force heroes:",
[
{"id": 1, "secret_name": "Dive Wilson", "age": None, "name": "Deadpond"},
{
"id": 3,
"secret_name": "Pedro Parqueador",
"age": None,
"name": "Spider-Boy",
},
],
],
[
"Reverted Z-Force's heroes:",
[{"id": 1, "secret_name": "Dive Wilson", "age": None, "name": "Deadpond"}],
],
[
"Reverted Spider-Boy's teams:",
[{"id": 2, "name": "Preventers", "headquarters": "Sharp Tower"}],
],
]
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.many_to_many import tutorial002 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls

76
tests/test_tutorial/test_many_to_many/test_tutorial003.py Normal file
View File

@@ -0,0 +1,76 @@
from typing import Any, Dict, List, Union
from unittest.mock import patch
from sqlalchemy import inspect
from sqlalchemy.engine.reflection import Inspector
from sqlmodel import create_engine
from sqlmodel.pool import StaticPool
from ...conftest import get_testing_print_function
expected_calls = [
[
"Z-Force hero:",
{"name": "Deadpond", "secret_name": "Dive Wilson", "id": 1, "age": None},
"is training:",
False,
],
[
"Preventers hero:",
{"name": "Deadpond", "secret_name": "Dive Wilson", "id": 1, "age": None},
"is training:",
True,
],
[
"Preventers hero:",
{"name": "Spider-Boy", "secret_name": "Pedro Parqueador", "id": 2, "age": None},
"is training:",
True,
],
[
"Preventers hero:",
{"name": "Rusty-Man", "secret_name": "Tommy Sharp", "id": 3, "age": 48},
"is training:",
False,
],
[
"Updated Spider-Boy's Teams:",
[
{"team_id": 2, "is_training": True, "hero_id": 2},
{"team_id": 1, "is_training": True, "hero_id": 2},
],
],
[
"Z-Force heroes:",
[
{"team_id": 1, "is_training": False, "hero_id": 1},
{"team_id": 1, "is_training": True, "hero_id": 2},
],
],
[
"Spider-Boy team:",
{"headquarters": "Sharp Tower", "id": 2, "name": "Preventers"},
"is training:",
False,
],
[
"Spider-Boy team:",
{"headquarters": "Sister Margaret’s Bar", "id": 1, "name": "Z-Force"},
"is training:",
True,
],
]
def test_tutorial(clear_sqlmodel):
from docs_src.tutorial.many_to_many import tutorial003 as mod
mod.sqlite_url = "sqlite://"
mod.engine = create_engine(mod.sqlite_url)
calls = []
new_print = get_testing_print_function(calls)
with patch("builtins.print", new=new_print):
mod.main()
assert calls == expected_calls