📝 Add source examples for Python 3.9 and 3.10 (#715)
* 📝 Add source examples for Python 3.9 and 3.10 * ✅ Add tests for new source examples for Python 3.9 and 3.10, still needs pytest markers * ✅ Add tests for fastapi examples * ✅ Update tests for FastAPI app testing, for Python 3.9 and 3.10, fixing multi-app testing conflicts * ✅ Require Python 3.9 and 3.10 for tests * ✅ Update tests with missing markers
This commit is contained in:
committed by
GitHub
parent
cce30d7546
commit
d8effcbc5c
46
tests/test_tutorial/test_indexes/test_tutorial001_py310.py
Normal file
46
tests/test_tutorial/test_indexes/test_tutorial001_py310.py
Normal file
@@ -0,0 +1,46 @@
|
||||
from unittest.mock import patch
|
||||
|
||||
from sqlalchemy import inspect
|
||||
from sqlalchemy.engine.reflection import Inspector
|
||||
from sqlmodel import create_engine
|
||||
|
||||
from ...conftest import get_testing_print_function, needs_py310
|
||||
|
||||
|
||||
@needs_py310
|
||||
def test_tutorial(clear_sqlmodel):
|
||||
from docs_src.tutorial.indexes import tutorial001_py310 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 == [
|
||||
[{"secret_name": "Dive Wilson", "age": None, "id": 1, "name": "Deadpond"}]
|
||||
]
|
||||
|
||||
insp: Inspector = inspect(mod.engine)
|
||||
indexes = insp.get_indexes(str(mod.Hero.__tablename__))
|
||||
expected_indexes = [
|
||||
{
|
||||
"name": "ix_hero_name",
|
||||
"dialect_options": {},
|
||||
"column_names": ["name"],
|
||||
"unique": 0,
|
||||
},
|
||||
{
|
||||
"name": "ix_hero_age",
|
||||
"dialect_options": {},
|
||||
"column_names": ["age"],
|
||||
"unique": 0,
|
||||
},
|
||||
]
|
||||
for index in expected_indexes:
|
||||
assert index in indexes, "This expected index should be in the indexes in DB"
|
||||
# Now that this index was checked, remove it from the list of indexes
|
||||
indexes.pop(indexes.index(index))
|
||||
assert len(indexes) == 0, "The database should only have the expected indexes"
|
||||
47
tests/test_tutorial/test_indexes/test_tutorial002_py310.py
Normal file
47
tests/test_tutorial/test_indexes/test_tutorial002_py310.py
Normal file
@@ -0,0 +1,47 @@
|
||||
from unittest.mock import patch
|
||||
|
||||
from sqlalchemy import inspect
|
||||
from sqlalchemy.engine.reflection import Inspector
|
||||
from sqlmodel import create_engine
|
||||
|
||||
from ...conftest import get_testing_print_function, needs_py310
|
||||
|
||||
|
||||
@needs_py310
|
||||
def test_tutorial(clear_sqlmodel):
|
||||
from docs_src.tutorial.indexes import tutorial002_py310 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 == [
|
||||
[{"name": "Tarantula", "secret_name": "Natalia Roman-on", "age": 32, "id": 4}],
|
||||
[{"name": "Black Lion", "secret_name": "Trevor Challa", "age": 35, "id": 5}],
|
||||
]
|
||||
|
||||
insp: Inspector = inspect(mod.engine)
|
||||
indexes = insp.get_indexes(str(mod.Hero.__tablename__))
|
||||
expected_indexes = [
|
||||
{
|
||||
"name": "ix_hero_name",
|
||||
"dialect_options": {},
|
||||
"column_names": ["name"],
|
||||
"unique": 0,
|
||||
},
|
||||
{
|
||||
"name": "ix_hero_age",
|
||||
"dialect_options": {},
|
||||
"column_names": ["age"],
|
||||
"unique": 0,
|
||||
},
|
||||
]
|
||||
for index in expected_indexes:
|
||||
assert index in indexes, "This expected index should be in the indexes in DB"
|
||||
# Now that this index was checked, remove it from the list of indexes
|
||||
indexes.pop(indexes.index(index))
|
||||
assert len(indexes) == 0, "The database should only have the expected indexes"
|
||||
Reference in New Issue
Block a user