test em

tests/test_hero.py

@@ -2,6 +2,9 @@ import tempfile
 
 from fastapi.testclient import TestClient
 import pytest
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from sqlmodel import SQLModel
 from typer.testing import CliRunner
 
 from learn_sql_model.api.app import app
@@ -18,11 +21,31 @@ client = TestClient(app)
 def config() -> Config:
     tmp_db = tempfile.NamedTemporaryFile(suffix=".db")
     config = get_config({"database_url": f"sqlite:///{tmp_db.name}"})
-    return config
+
+    engine = create_engine(
+        config.database_url, connect_args={"check_same_thread": False}
+    )
+    TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+    # breakpoint()
+    SQLModel.metadata.create_all(config.database.engine)
+
+    # def override_get_db():
+    #     try:
+    #         db = TestingSessionLocal()
+    #         yield db
+    #     finally:
+    #         db.close()
+    def override_get_config():
+        return config
+
+    app.dependency_overrides[get_config] = override_get_config
+
+    yield config
+    # tmp_db automatically deletes here
 
 
 def test_post_hero(config: Config) -> None:
-    config.init()  # required for python api, and no existing db
     hero = HeroFactory().build(name="Batman", age=50, id=1)
     hero = hero.post(config=config)
     db_hero = Hero().get(id=1, config=config)
@@ -31,48 +54,105 @@ def test_post_hero(config: Config) -> None:
 
 
 def test_update_hero(config: Config) -> None:
-    config.init()  # required for python api, and no existing db
     hero = HeroFactory().build(name="Batman", age=50, id=1)
     hero = hero.post(config=config)
     db_hero = Hero().get(id=1, config=config)
-    db_hero.name = "Superman"
+    db_hero.name = "Superbman"
     hero = db_hero.post(config=config)
     db_hero = Hero().get(id=1, config=config)
     assert db_hero.age == 50
-    assert db_hero.name == "Superman"
+    assert db_hero.name == "Superbman"
+
+
+def test_cli_get(config):
+    hero = HeroFactory().build(name="Steelman", age=25, id=99)
+    hero.post(config=config)
+    result = runner.invoke(
+        hero_app,
+        ["get", "--id", 99, "--database-url", config.database_url],
+    )
+    assert result.exit_code == 0
+    db_hero = Hero().get(id=99, config=config)
+    assert db_hero.age == 25
+    assert db_hero.name == "Steelman"
 
 
 def test_cli_create(config):
+    hero = HeroFactory().build(name="Steelman", age=25, id=99)
     result = runner.invoke(
         hero_app,
         [
             "create",
-            "--name",
-            "Darth Vader",
-            "--secret-name",
-            "Anakin",
-            "--id",
-            "2",
-            "--age",
-            "100",
+            *hero.flags(config=config),
             "--database-url",
             config.database_url,
         ],
     )
     assert result.exit_code == 0
-    db_hero = Hero().get(id=2, config=config)
-    assert db_hero.age == 100
-    assert db_hero.name == "Darth Vader"
+    db_hero = Hero().get(id=99, config=config)
+    assert db_hero.age == 25
+    assert db_hero.name == "Steelman"
 
 
-def test_read_main(config):
-    config.init()
-    hero = HeroFactory().build(name="Ironman", age=25, id=99)
+def test_cli_populate(config):
+    result = runner.invoke(
+        hero_app,
+        [
+            "populate",
+            "--n",
+            10,
+            "--database-url",
+            config.database_url,
+        ],
+    )
+    assert result.exit_code == 0
+    db_hero = Hero().get(config=config)
+    assert len(db_hero) == 10
+
+
+def test_cli_populate_fails_prod(config):
+    result = runner.invoke(
+        hero_app,
+        ["populate", "--n", 10, "--database-url", config.database_url, "--env", "prod"],
+    )
+    assert result.exit_code == 1
+    assert result.output.strip() == "populate is not supported in production"
+
+
+def test_api_read(config):
+    hero = HeroFactory().build(name="Steelman", age=25, id=99)
     hero_id = hero.id
     hero = hero.post(config=config)
     response = client.get(f"/hero/{hero_id}")
     assert response.status_code == 200
     reponse_hero = Hero.parse_obj(response.json())
     assert reponse_hero.id == hero_id
-    assert reponse_hero.name == "Ironman"
+    assert reponse_hero.name == "Steelman"
     assert reponse_hero.age == 25
+
+
+def test_api_post(config):
+    hero = HeroFactory().build(name="Steelman", age=25)
+    hero_dict = hero.dict()
+    response = client.post("/hero/", json={"hero": hero_dict})
+    assert response.status_code == 200
+
+    response_hero = Hero.parse_obj(response.json())
+    db_hero = Hero().get(id=response_hero.id, config=config)
+
+    assert db_hero.name == "Steelman"
+    assert db_hero.age == 25
+
+
+def test_api_read_all(config):
+    hero = HeroFactory().build(name="Mothman", age=25, id=99)
+    hero_id = hero.id
+    hero = hero.post(config=config)
+    response = client.get("/heros/")
+    assert response.status_code == 200
+    heros = response.json()
+    response_hero_json = [hero for hero in heros if hero["id"] == hero_id][0]
+    response_hero = Hero.parse_obj(response_hero_json)
+    assert response_hero.id == hero_id
+    assert response_hero.name == "Mothman"
+    assert response_hero.age == 25