Phil Skentelbery
bebd47955f
Implements Phase 1 Foundation with all core services: Core Components: - Configuration management with GONDULF_ environment variables - Database layer with SQLAlchemy and migration system - In-memory code storage with TTL support - Email service with SMTP and TLS support (STARTTLS + implicit TLS) - DNS service with TXT record verification - Structured logging with Python standard logging - FastAPI application with health check endpoint Database Schema: - authorization_codes table for OAuth 2.0 authorization codes - domains table for domain verification - migrations table for tracking schema versions - Simple sequential migration system (001_initial_schema.sql) Configuration: - Environment-based configuration with validation - .env.example template with all GONDULF_ variables - Fail-fast validation on startup - Sensible defaults for optional settings Testing: - 96 comprehensive tests (77 unit, 5 integration) - 94.16% code coverage (exceeds 80% requirement) - All tests passing - Test coverage includes: - Configuration loading and validation - Database migrations and health checks - In-memory storage with expiration - Email service (STARTTLS, implicit TLS, authentication) - DNS service (TXT records, domain verification) - Health check endpoint integration Documentation: - Implementation report with test results - Phase 1 clarifications document - ADRs for key decisions (config, database, email, logging) Technical Details: - Python 3.10+ with type hints - SQLite with configurable database URL - System DNS with public DNS fallback - Port-based TLS detection (465=SSL, 587=STARTTLS) - Lazy configuration loading for testability Exit Criteria Met: ✓ All foundation services implemented ✓ Application starts without errors ✓ Health check endpoint operational ✓ Database migrations working ✓ Test coverage exceeds 80% ✓ All tests passing Ready for Architect review and Phase 2 development. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
275 lines
9.3 KiB
Python
275 lines
9.3 KiB
Python
"""
|
|
Unit tests for database connection and migrations.
|
|
|
|
Tests database initialization, migration running, and health checks.
|
|
"""
|
|
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
from sqlalchemy import text
|
|
|
|
from gondulf.database.connection import Database, DatabaseError
|
|
|
|
|
|
class TestDatabaseInit:
|
|
"""Tests for Database initialization."""
|
|
|
|
def test_init_with_valid_url(self):
|
|
"""Test Database can be initialized with valid URL."""
|
|
db = Database("sqlite:///:memory:")
|
|
assert db.database_url == "sqlite:///:memory:"
|
|
|
|
def test_init_with_file_url(self):
|
|
"""Test Database can be initialized with file URL."""
|
|
db = Database("sqlite:///./test.db")
|
|
assert db.database_url == "sqlite:///./test.db"
|
|
|
|
|
|
class TestDatabaseDirectory:
|
|
"""Tests for database directory creation."""
|
|
|
|
def test_ensure_directory_creates_parent(self):
|
|
"""Test ensure_database_directory creates parent directories."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "subdir" / "nested" / "test.db"
|
|
db_url = f"sqlite:///{db_path}"
|
|
|
|
db = Database(db_url)
|
|
db.ensure_database_directory()
|
|
|
|
assert db_path.parent.exists()
|
|
|
|
def test_ensure_directory_relative_path(self):
|
|
"""Test ensure_database_directory works with relative paths."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
# Change to temp dir temporarily to test relative paths
|
|
import os
|
|
|
|
original_cwd = os.getcwd()
|
|
try:
|
|
os.chdir(tmpdir)
|
|
|
|
db = Database("sqlite:///./data/test.db")
|
|
db.ensure_database_directory()
|
|
|
|
assert Path("data").exists()
|
|
finally:
|
|
os.chdir(original_cwd)
|
|
|
|
def test_ensure_directory_does_not_fail_if_exists(self):
|
|
"""Test ensure_database_directory doesn't fail if directory exists."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "test.db"
|
|
db_url = f"sqlite:///{db_path}"
|
|
|
|
db = Database(db_url)
|
|
db.ensure_database_directory()
|
|
# Call again - should not raise
|
|
db.ensure_database_directory()
|
|
|
|
|
|
class TestDatabaseEngine:
|
|
"""Tests for database engine creation."""
|
|
|
|
def test_get_engine_creates_engine(self):
|
|
"""Test get_engine creates SQLAlchemy engine."""
|
|
db = Database("sqlite:///:memory:")
|
|
engine = db.get_engine()
|
|
|
|
assert engine is not None
|
|
assert engine.url.drivername == "sqlite"
|
|
|
|
def test_get_engine_returns_same_instance(self):
|
|
"""Test get_engine returns same engine instance."""
|
|
db = Database("sqlite:///:memory:")
|
|
engine1 = db.get_engine()
|
|
engine2 = db.get_engine()
|
|
|
|
assert engine1 is engine2
|
|
|
|
def test_get_engine_with_invalid_url_raises_error(self):
|
|
"""Test get_engine raises DatabaseError with invalid URL."""
|
|
db = Database("invalid://bad_url")
|
|
|
|
with pytest.raises(DatabaseError, match="Failed to create database engine"):
|
|
db.get_engine()
|
|
|
|
|
|
class TestDatabaseHealth:
|
|
"""Tests for database health checks."""
|
|
|
|
def test_check_health_success(self):
|
|
"""Test health check passes for healthy database."""
|
|
db = Database("sqlite:///:memory:")
|
|
db.get_engine() # Initialize engine
|
|
|
|
assert db.check_health() is True
|
|
|
|
def test_check_health_failure(self):
|
|
"""Test health check fails for inaccessible database."""
|
|
db = Database("sqlite:////nonexistent/path/db.db")
|
|
|
|
# Trying to check health on non-existent DB should fail gracefully
|
|
assert db.check_health() is False
|
|
|
|
|
|
class TestDatabaseMigrations:
|
|
"""Tests for database migrations."""
|
|
|
|
def test_get_applied_migrations_empty(self):
|
|
"""Test get_applied_migrations returns empty set for new database."""
|
|
db = Database("sqlite:///:memory:")
|
|
db.get_engine() # Initialize engine
|
|
|
|
migrations = db.get_applied_migrations()
|
|
|
|
assert migrations == set()
|
|
|
|
def test_get_applied_migrations_after_running(self):
|
|
"""Test get_applied_migrations returns versions after running migrations."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "test.db"
|
|
db = Database(f"sqlite:///{db_path}")
|
|
|
|
# Initialize will run migrations
|
|
db.initialize()
|
|
|
|
migrations = db.get_applied_migrations()
|
|
|
|
# Migration 001 should be applied
|
|
assert 1 in migrations
|
|
|
|
def test_run_migrations_creates_tables(self):
|
|
"""Test run_migrations creates expected tables."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "test.db"
|
|
db = Database(f"sqlite:///{db_path}")
|
|
|
|
db.ensure_database_directory()
|
|
db.run_migrations()
|
|
|
|
# Check that tables were created
|
|
engine = db.get_engine()
|
|
with engine.connect() as conn:
|
|
# Check migrations table
|
|
result = conn.execute(text("SELECT name FROM sqlite_master WHERE type='table'"))
|
|
tables = {row[0] for row in result}
|
|
|
|
assert "migrations" in tables
|
|
assert "authorization_codes" in tables
|
|
assert "domains" in tables
|
|
|
|
def test_run_migrations_idempotent(self):
|
|
"""Test run_migrations can be run multiple times safely."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "test.db"
|
|
db = Database(f"sqlite:///{db_path}")
|
|
|
|
db.ensure_database_directory()
|
|
db.run_migrations()
|
|
# Run again - should not raise or duplicate
|
|
db.run_migrations()
|
|
|
|
engine = db.get_engine()
|
|
with engine.connect() as conn:
|
|
# Check migration was recorded only once
|
|
result = conn.execute(text("SELECT COUNT(*) FROM migrations"))
|
|
count = result.fetchone()[0]
|
|
assert count == 1
|
|
|
|
def test_initialize_full_setup(self):
|
|
"""Test initialize performs full database setup."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "test.db"
|
|
db = Database(f"sqlite:///{db_path}")
|
|
|
|
db.initialize()
|
|
|
|
# Verify database is healthy
|
|
assert db.check_health() is True
|
|
|
|
# Verify migrations ran
|
|
migrations = db.get_applied_migrations()
|
|
assert 1 in migrations
|
|
|
|
# Verify tables exist
|
|
engine = db.get_engine()
|
|
with engine.connect() as conn:
|
|
result = conn.execute(text("SELECT name FROM sqlite_master WHERE type='table'"))
|
|
tables = {row[0] for row in result}
|
|
|
|
assert "migrations" in tables
|
|
assert "authorization_codes" in tables
|
|
assert "domains" in tables
|
|
|
|
|
|
class TestMigrationSchemaCorrectness:
|
|
"""Tests for correctness of migration schema."""
|
|
|
|
def test_authorization_codes_schema(self):
|
|
"""Test authorization_codes table has correct columns."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "test.db"
|
|
db = Database(f"sqlite:///{db_path}")
|
|
db.initialize()
|
|
|
|
engine = db.get_engine()
|
|
with engine.connect() as conn:
|
|
result = conn.execute(text("PRAGMA table_info(authorization_codes)"))
|
|
columns = {row[1] for row in result} # row[1] is column name
|
|
|
|
expected_columns = {
|
|
"code",
|
|
"client_id",
|
|
"redirect_uri",
|
|
"state",
|
|
"code_challenge",
|
|
"code_challenge_method",
|
|
"scope",
|
|
"me",
|
|
"created_at",
|
|
}
|
|
|
|
assert columns == expected_columns
|
|
|
|
def test_domains_schema(self):
|
|
"""Test domains table has correct columns."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "test.db"
|
|
db = Database(f"sqlite:///{db_path}")
|
|
db.initialize()
|
|
|
|
engine = db.get_engine()
|
|
with engine.connect() as conn:
|
|
result = conn.execute(text("PRAGMA table_info(domains)"))
|
|
columns = {row[1] for row in result}
|
|
|
|
expected_columns = {
|
|
"domain",
|
|
"email",
|
|
"verification_code",
|
|
"verified",
|
|
"created_at",
|
|
"verified_at",
|
|
}
|
|
|
|
assert columns == expected_columns
|
|
|
|
def test_migrations_schema(self):
|
|
"""Test migrations table has correct columns."""
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
db_path = Path(tmpdir) / "test.db"
|
|
db = Database(f"sqlite:///{db_path}")
|
|
db.initialize()
|
|
|
|
engine = db.get_engine()
|
|
with engine.connect() as conn:
|
|
result = conn.execute(text("PRAGMA table_info(migrations)"))
|
|
columns = {row[1] for row in result}
|
|
|
|
expected_columns = {"version", "description", "applied_at"}
|
|
|
|
assert columns == expected_columns
|