✅ Add tests

tests/test_tutorial/test_one/test_tutorial001.py

@@ -0,0 +1,29 @@
+from typing import Any, Dict, List, Union
+from unittest.mock import patch
+
+from sqlmodel import create_engine
+from ...conftest import get_testing_print_function
+
+
+def test_tutorial(clear_sqlmodel):
+    from docs_src.tutorial.one 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 == [
+        [
+            "Hero:",
+            {
+                "name": "Tarantula",
+                "secret_name": "Natalia Roman-on",
+                "age": 32,
+                "id": 4,
+            },
+        ]
+    ]

tests/test_tutorial/test_one/test_tutorial002.py

@@ -0,0 +1,19 @@
+from typing import Any, Dict, List, Union
+from unittest.mock import patch
+
+from sqlmodel import create_engine
+from ...conftest import get_testing_print_function
+
+
+def test_tutorial(clear_sqlmodel):
+    from docs_src.tutorial.one 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 == [["Hero:", None]]

tests/test_tutorial/test_one/test_tutorial003.py

@@ -0,0 +1,24 @@
+from typing import Any, Dict, List, Union
+from unittest.mock import patch
+
+from sqlmodel import create_engine
+from ...conftest import get_testing_print_function
+
+
+def test_tutorial(clear_sqlmodel):
+    from docs_src.tutorial.one 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 == [
+        [
+            "Hero:",
+            {"name": "Deadpond", "secret_name": "Dive Wilson", "age": None, "id": 1},
+        ]
+    ]

tests/test_tutorial/test_one/test_tutorial004.py

@@ -0,0 +1,42 @@
+from typing import Any, Dict, List, Union
+from unittest.mock import patch
+
+import pytest
+from sqlalchemy.sql.expression import delete
+
+from sqlmodel import create_engine, Session, delete, select
+from sqlalchemy.exc import MultipleResultsFound
+from ...conftest import get_testing_print_function
+
+
+def test_tutorial(clear_sqlmodel):
+    from docs_src.tutorial.one import tutorial004 as mod
+
+    mod.sqlite_url = "sqlite://"
+    mod.engine = create_engine(mod.sqlite_url)
+    with pytest.raises(MultipleResultsFound):
+        mod.main()
+    with Session(mod.engine) as session:
+        # TODO: create delete() function
+        # TODO: add overloads for .exec() with delete object
+        session.exec(delete(mod.Hero))
+        session.add(mod.Hero(name="Test Hero", secret_name="Secret Test Hero", age=24))
+        session.commit()
+
+    calls = []
+
+    new_print = get_testing_print_function(calls)
+
+    with patch("builtins.print", new=new_print):
+        mod.select_heroes()
+    assert calls == [
+        [
+            "Hero:",
+            {
+                "id": 1,
+                "name": "Test Hero",
+                "secret_name": "Secret Test Hero",
+                "age": 24,
+            },
+        ]
+    ]

tests/test_tutorial/test_one/test_tutorial005.py

@@ -0,0 +1,42 @@
+from typing import Any, Dict, List, Union
+from unittest.mock import patch
+
+import pytest
+from sqlalchemy.sql.expression import delete
+
+from sqlmodel import create_engine, Session, delete, select
+from sqlalchemy.exc import NoResultFound
+from ...conftest import get_testing_print_function
+
+
+def test_tutorial(clear_sqlmodel):
+    from docs_src.tutorial.one import tutorial005 as mod
+
+    mod.sqlite_url = "sqlite://"
+    mod.engine = create_engine(mod.sqlite_url)
+    with pytest.raises(NoResultFound):
+        mod.main()
+    with Session(mod.engine) as session:
+        # TODO: create delete() function
+        # TODO: add overloads for .exec() with delete object
+        session.exec(delete(mod.Hero))
+        session.add(mod.Hero(name="Test Hero", secret_name="Secret Test Hero", age=24))
+        session.commit()
+
+    calls = []
+
+    new_print = get_testing_print_function(calls)
+
+    with patch("builtins.print", new=new_print):
+        mod.select_heroes()
+    assert calls == [
+        [
+            "Hero:",
+            {
+                "id": 1,
+                "name": "Test Hero",
+                "secret_name": "Secret Test Hero",
+                "age": 24,
+            },
+        ]
+    ]

tests/test_tutorial/test_one/test_tutorial006.py

@@ -0,0 +1,24 @@
+from typing import Any, Dict, List, Union
+from unittest.mock import patch
+
+from sqlmodel import create_engine
+from ...conftest import get_testing_print_function
+
+
+def test_tutorial(clear_sqlmodel):
+    from docs_src.tutorial.one import tutorial006 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 == [
+        [
+            "Hero:",
+            {"name": "Deadpond", "secret_name": "Dive Wilson", "age": None, "id": 1},
+        ]
+    ]

tests/test_tutorial/test_one/test_tutorial007.py

@@ -0,0 +1,24 @@
+from typing import Any, Dict, List, Union
+from unittest.mock import patch
+
+from sqlmodel import create_engine
+from ...conftest import get_testing_print_function
+
+
+def test_tutorial(clear_sqlmodel):
+    from docs_src.tutorial.one import tutorial007 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 == [
+        [
+            "Hero:",
+            {"name": "Deadpond", "secret_name": "Dive Wilson", "age": None, "id": 1},
+        ]
+    ]

tests/test_tutorial/test_one/test_tutorial008.py

@@ -0,0 +1,24 @@
+from typing import Any, Dict, List, Union
+from unittest.mock import patch
+
+from sqlmodel import create_engine
+from ...conftest import get_testing_print_function
+
+
+def test_tutorial(clear_sqlmodel):
+    from docs_src.tutorial.one import tutorial008 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 == [
+        [
+            "Hero:",
+            {"name": "Deadpond", "secret_name": "Dive Wilson", "age": None, "id": 1},
+        ]
+    ]

tests/test_tutorial/test_one/test_tutorial009.py

@@ -0,0 +1,19 @@
+from typing import Any, Dict, List, Union
+from unittest.mock import patch
+
+from sqlmodel import create_engine
+from ...conftest import get_testing_print_function
+
+
+def test_tutorial(clear_sqlmodel):
+    from docs_src.tutorial.one import tutorial009 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 == [["Hero:", None]]