test: improve test coverage from 61% to 84.81%

Major improvements:
- Fixed all 42 broken tests (database connection leaks)
- Added db_connection() context manager for proper cleanup
- Created comprehensive test suites for undertested modules

New test coverage:
- tools/general_tools.py: 26 tests (97% coverage)
- tools/price_tools.py: 11 tests (validates NASDAQ symbols, date handling)
- api/price_data_manager.py: 12 tests (85% coverage)
- api/routes/results_v2.py: 3 tests (98% coverage)
- agent/reasoning_summarizer.py: 2 tests (87% coverage)
- api/routes/period_metrics.py: 2 edge case tests (100% coverage)
- agent/mock_provider: 1 test (100% coverage)

Database fixes:
- Added db_connection() context manager to prevent leaks
- Updated 16+ test files to use context managers
- Fixed drop_all_tables() to match new schema
- Added CHECK constraint for action_type
- Added ON DELETE CASCADE to trading_days foreign key

Test improvements:
- Updated SQL INSERT statements with all required fields
- Fixed date parameter handling in API integration tests
- Added edge case tests for validation functions
- Fixed import errors across test suite

Results:
- Total coverage: 84.81% (was 61%)
- Tests passing: 406 (was 364 with 42 failures)
- Total lines covered: 6364 of 7504

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
2025-11-07 21:02:38 -05:00
parent 61baf3f90f
commit 14cf88f642
30 changed files with 1840 additions and 1013 deletions

View File

@@ -11,7 +11,7 @@ import pytest
import tempfile
import os
from pathlib import Path
from api.database import initialize_database, get_db_connection
from api.database import initialize_database, get_db_connection, db_connection
@pytest.fixture(scope="session")
@@ -52,39 +52,38 @@ def clean_db(test_db_path):
db = Database(test_db_path)
db.connection.close()
# Clear all tables
conn = get_db_connection(test_db_path)
cursor = conn.cursor()
# Clear all tables using context manager for guaranteed cleanup
with db_connection(test_db_path) as conn:
cursor = conn.cursor()
# Get list of tables that exist
cursor.execute("""
SELECT name FROM sqlite_master
WHERE type='table' AND name NOT LIKE 'sqlite_%'
""")
tables = [row[0] for row in cursor.fetchall()]
# Get list of tables that exist
cursor.execute("""
SELECT name FROM sqlite_master
WHERE type='table' AND name NOT LIKE 'sqlite_%'
""")
tables = [row[0] for row in cursor.fetchall()]
# Delete in correct order (respecting foreign keys), only if table exists
if 'tool_usage' in tables:
cursor.execute("DELETE FROM tool_usage")
if 'actions' in tables:
cursor.execute("DELETE FROM actions")
if 'holdings' in tables:
cursor.execute("DELETE FROM holdings")
if 'trading_days' in tables:
cursor.execute("DELETE FROM trading_days")
if 'simulation_runs' in tables:
cursor.execute("DELETE FROM simulation_runs")
if 'job_details' in tables:
cursor.execute("DELETE FROM job_details")
if 'jobs' in tables:
cursor.execute("DELETE FROM jobs")
if 'price_data_coverage' in tables:
cursor.execute("DELETE FROM price_data_coverage")
if 'price_data' in tables:
cursor.execute("DELETE FROM price_data")
# Delete in correct order (respecting foreign keys), only if table exists
if 'tool_usage' in tables:
cursor.execute("DELETE FROM tool_usage")
if 'actions' in tables:
cursor.execute("DELETE FROM actions")
if 'holdings' in tables:
cursor.execute("DELETE FROM holdings")
if 'trading_days' in tables:
cursor.execute("DELETE FROM trading_days")
if 'simulation_runs' in tables:
cursor.execute("DELETE FROM simulation_runs")
if 'job_details' in tables:
cursor.execute("DELETE FROM job_details")
if 'jobs' in tables:
cursor.execute("DELETE FROM jobs")
if 'price_data_coverage' in tables:
cursor.execute("DELETE FROM price_data_coverage")
if 'price_data' in tables:
cursor.execute("DELETE FROM price_data")
conn.commit()
conn.close()
conn.commit()
return test_db_path