updated tests

tests/unit/test_database.py

@@ -0,0 +1,138 @@
+"""Unit tests for database module."""
+
+from pathlib import Path
+from sqlalchemy.orm import Session
+
+from baby_monitor.repositories.dependencies.get_database import get_database, init_db
+
+
+def test_get_db_yields_session() -> None:
+    """Test that get_db yields a valid database session."""
+    init_db()
+
+    db_generator = get_database()
+    db = next(db_generator)
+
+    assert isinstance(db, Session)
+    assert db.is_active
+
+    # Close the session
+    try:
+        next(db_generator)
+    except StopIteration:
+        pass
+
+
+def test_get_db_closes_session() -> None:
+    """Test that get_db properly closes the session."""
+    from unittest.mock import patch
+    
+    init_db()
+
+    db_generator = get_database()
+    db = next(db_generator)
+
+    assert db.is_active
+
+    # Patch the close method to verify it gets called
+    with patch.object(db, 'close', wraps=db.close) as mock_close:
+        # Complete the generator to trigger finally block
+        try:
+            next(db_generator)
+        except StopIteration:
+            pass
+        
+        # Verify that close() was called
+        mock_close.assert_called_once()
+
+
+def test_get_db_closes_on_exception() -> None:
+    """Test that get_db closes session when used in context manager."""
+    from unittest.mock import patch
+    
+    init_db()
+
+    # Simulate how FastAPI actually uses the dependency
+    db = None
+    db_generator = get_database()
+    try:
+        db = next(db_generator)
+        assert db.is_active
+
+        # Patch close to verify it gets called even with an exception
+        with patch.object(db, 'close', wraps=db.close) as mock_close:
+            try:
+                # Simulate an exception occurring during request processing
+                raise ValueError("Simulated error during request")
+            except ValueError:
+                pass
+            finally:
+                # FastAPI ensures the generator completes
+                try:
+                    next(db_generator)
+                except StopIteration:
+                    pass
+            
+            # Verify close() was called despite the exception
+            mock_close.assert_called_once()
+    except Exception:
+        # Ensure we have a reference to db for the final assertion
+        pass
+
+    # Verify db was created
+    assert db is not None
+
+# def test_init_db_creates_directory(tmp_path: Path) -> None:
+#     """Test that init_db creates the data directory."""
+#     import os
+
+#     # Use a fresh temp directory for this test
+#     test_dir = tmp_path / "test_data"
+
+#     # Reinitialize database module with new path
+#     from baby_monitor.repositories.dependencies.get_database import (
+#         DATA_DIR,
+#         DATABASE_URL,
+#         create_engine,
+#         sessionmaker,
+#     )
+#     DATABASE_URL = f"sqlite:///{test_dir}/baby_monitor.db"
+#     engine = create_engine(
+#         DATABASE_URL,
+#         connect_args={"check_same_thread": False},
+#         echo=False,
+#     )
+#     SessionLocal = sessionmaker(
+#         autocommit=False,
+#         autoflush=False,
+#         bind=engine
+#     )
+
+#     init_db()
+
+#     assert isinstance(SessionLocal, sessionmaker)
+#     assert test_dir.exists()
+#     assert test_dir.is_dir()
+
+
+def test_init_db_creates_tables(tmp_path: Path) -> None:
+    """Test that init_db creates database tables."""
+    init_db()
+
+    # Check that database file exists
+    import os
+
+    data_dir = Path(os.environ["DATA_DIR"])
+    db_path = data_dir / "baby_monitor.db"
+    assert db_path.exists()
+
+    # Verify tables were created by checking metadata
+    from baby_monitor.repositories.dependencies.get_database import engine
+    from sqlalchemy import inspect
+
+    inspector = inspect(engine)
+    tables = inspector.get_table_names()
+
+    # Should have at least some tables (depends on models)
+    # For now, just verify the function runs without error
+    assert isinstance(tables, list)