Local changes: Updated model training, removed debug instrumentation, and configuration improvements

tests/conftest.py

@@ -0,0 +1,86 @@
+"""Pytest configuration and fixtures."""
+
+import pytest
+import asyncio
+from unittest.mock import Mock, AsyncMock, PropertyMock
+from decimal import Decimal
+from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
+from src.core.database import Base, Database, get_database
+
+from sqlalchemy.pool import StaticPool
+
+@pytest.fixture(scope="session")
+def event_loop():
+    """Create an instance of the default event loop for each test case."""
+    loop = asyncio.get_event_loop_policy().new_event_loop()
+    yield loop
+    loop.close()
+
+@pytest.fixture(scope="session")
+async def db_engine():
+    """Create async database engine."""
+    engine = create_async_engine(
+        "sqlite+aiosqlite:///:memory:", 
+        connect_args={"check_same_thread": False},
+        poolclass=StaticPool,
+        echo=False
+    )
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all)
+    yield engine
+    await engine.dispose()
+
+@pytest.fixture
+async def db_session(db_engine):
+    """Create async database session."""
+    async_session = async_sessionmaker(bind=db_engine, expire_on_commit=False)
+    async with async_session() as session:
+        yield session
+
+@pytest.fixture(autouse=True)
+def override_get_database(db_engine, monkeypatch):
+    """Override get_database to use test engine."""
+    test_db = Database()
+    # We mock the internal attributes to return our test engine/session
+    test_db.engine = db_engine
+    test_db.SessionLocal = async_sessionmaker(bind=db_engine, class_=AsyncSession, expire_on_commit=False)
+    
+    # Patch the global get_database
+    monkeypatch.setattr("src.core.database._db_instance", test_db)
+    return test_db
+
+@pytest.fixture
+def mock_exchange_adapter():
+    """Mock exchange adapter."""
+    from src.exchanges.base import BaseExchangeAdapter
+    adapter = AsyncMock(spec=BaseExchangeAdapter)
+    adapter.get_ticker.return_value = {'last': Decimal("50000")}
+    adapter.place_order.return_value = {'id': 'test_order_123', 'status': 'open'}
+    adapter.get_balance.return_value = {'USD': Decimal("10000")}
+    # Helper methods should be sync mocks
+    # Note: If extract_fee... is not part of BaseExchangeAdapter spec, we have to attach it manually
+    # But checking base.py, it likely IS or isn't.
+    # Safe to attach it manually even with spec if we traverse __dict__ or simply assign.
+    # However, standard mock might block unknown attribs.
+    # Actually BaseExchangeAdapter is abstract.
+    
+    # Let's inspect BaseExchangeAdapter structure if needed. 
+    # For now, let's assume usage of spec is the right direction. 
+    # But if extract_fee... is NOT in BaseExchangeAdapter, we might need to mock a Concrete class like Coinbase
+    pass
+    
+    # Better approach: Just Delete get_fee_structure from the mock to ensure AttributeError
+    adapter = AsyncMock()
+    del adapter.get_fee_structure 
+    # Wait, AsyncMock creates attrs on access. del might not work if not existing.
+    # We can se side_effect to raise AttributeError
+    
+    adapter.get_ticker.return_value = {'last': Decimal("50000")}
+    adapter.place_order.return_value = {'id': 'test_order_123', 'status': 'open'}
+    adapter.get_balance.return_value = {'USD': Decimal("10000")}
+    adapter.extract_fee_from_order_response = Mock(return_value=Decimal("1.0"))
+    adapter.name = "coinbase" # FeeCalculator accesses .name
+    type(adapter).get_fee_structure = PropertyMock(side_effect=AttributeError) 
+    # Accessing methods on AsyncMock...
+    
+    return adapter