mirror of
https://github.com/Xe138/AI-Trader.git
synced 2026-04-18 15:57:24 -04:00
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:
@@ -13,7 +13,7 @@ from unittest.mock import patch, Mock
|
||||
from datetime import datetime
|
||||
|
||||
from api.price_data_manager import PriceDataManager, RateLimitError, DownloadError
|
||||
from api.database import initialize_database, get_db_connection
|
||||
from api.database import initialize_database, get_db_connection, db_connection
|
||||
from api.date_utils import expand_date_range
|
||||
|
||||
|
||||
@@ -130,12 +130,11 @@ class TestEndToEndDownload:
|
||||
assert available_dates == ["2025-01-20", "2025-01-21"]
|
||||
|
||||
# Verify coverage tracking
|
||||
conn = get_db_connection(manager.db_path)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT COUNT(*) FROM price_data_coverage")
|
||||
coverage_count = cursor.fetchone()[0]
|
||||
assert coverage_count == 5 # One record per symbol
|
||||
conn.close()
|
||||
with db_connection(manager.db_path) as conn:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT COUNT(*) FROM price_data_coverage")
|
||||
coverage_count = cursor.fetchone()[0]
|
||||
assert coverage_count == 5 # One record per symbol
|
||||
|
||||
@patch('api.price_data_manager.requests.get')
|
||||
def test_download_with_partial_existing_data(self, mock_get, manager, mock_alpha_vantage_response):
|
||||
@@ -340,15 +339,14 @@ class TestCoverageTracking:
|
||||
manager._update_coverage("AAPL", dates[0], dates[1])
|
||||
|
||||
# Verify coverage was recorded
|
||||
conn = get_db_connection(manager.db_path)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("""
|
||||
SELECT symbol, start_date, end_date, source
|
||||
FROM price_data_coverage
|
||||
WHERE symbol = 'AAPL'
|
||||
""")
|
||||
row = cursor.fetchone()
|
||||
conn.close()
|
||||
with db_connection(manager.db_path) as conn:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("""
|
||||
SELECT symbol, start_date, end_date, source
|
||||
FROM price_data_coverage
|
||||
WHERE symbol = 'AAPL'
|
||||
""")
|
||||
row = cursor.fetchone()
|
||||
|
||||
assert row is not None
|
||||
assert row[0] == "AAPL"
|
||||
@@ -444,10 +442,9 @@ class TestDataValidation:
|
||||
assert set(stored_dates) == requested_dates
|
||||
|
||||
# Verify in database
|
||||
conn = get_db_connection(manager.db_path)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT date FROM price_data WHERE symbol = 'AAPL' ORDER BY date")
|
||||
db_dates = [row[0] for row in cursor.fetchall()]
|
||||
conn.close()
|
||||
with db_connection(manager.db_path) as conn:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute("SELECT date FROM price_data WHERE symbol = 'AAPL' ORDER BY date")
|
||||
db_dates = [row[0] for row in cursor.fetchall()]
|
||||
|
||||
assert db_dates == ["2025-01-20", "2025-01-21"]
|
||||
|
||||
Reference in New Issue
Block a user