feat: auto-initialize trading_days schema on database creation

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

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
2025-11-04 07:09:09 -05:00
parent 93ba9deebb
commit a673fc5008
2 changed files with 59 additions and 0 deletions

View File

@@ -0,0 +1,30 @@
import pytest
from api.database import Database
class TestDatabaseInitialization:
def test_database_creates_new_schema_on_init(self, tmp_path):
"""Test database automatically creates trading_days schema."""
db_path = tmp_path / "new.db"
# Create database (should auto-initialize schema)
db = Database(str(db_path))
# Verify trading_days table exists
cursor = db.connection.execute(
"SELECT name FROM sqlite_master WHERE type='table' AND name='trading_days'"
)
assert cursor.fetchone() is not None
# Verify holdings table exists
cursor = db.connection.execute(
"SELECT name FROM sqlite_master WHERE type='table' AND name='holdings'"
)
assert cursor.fetchone() is not None
# Verify actions table exists
cursor = db.connection.execute(
"SELECT name FROM sqlite_master WHERE type='table' AND name='actions'"
)
assert cursor.fetchone() is not None