mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-28 02:23:53 +02:00
a4dc84d1ab
- Introduced should_summarize parameter in TaskDispatcher and CeleryTaskDispatcher to control summary generation. - Updated InlineTaskDispatcher to support the new parameter for document processing.
104 lines
3.3 KiB
Python
104 lines
3.3 KiB
Python
"""Shared pytest fixtures for SurfSense backend tests."""
|
|
|
|
import pytest
|
|
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
from app.db import Base
|
|
|
|
|
|
@pytest.fixture
|
|
async def async_session():
|
|
"""Create an async database session for testing."""
|
|
from sqlalchemy import JSON, ARRAY, event
|
|
from sqlalchemy.dialects.postgresql import JSONB
|
|
|
|
# Use in-memory SQLite for tests
|
|
engine = create_async_engine(
|
|
"sqlite+aiosqlite:///:memory:",
|
|
echo=False,
|
|
)
|
|
|
|
# Replace JSONB and ARRAY with JSON for SQLite compatibility
|
|
@event.listens_for(Base.metadata, "before_create")
|
|
def _set_json_type(target, connection, **kw):
|
|
for table in Base.metadata.tables.values():
|
|
for column in table.columns:
|
|
# Convert JSONB to JSON
|
|
if isinstance(column.type, type(JSONB())):
|
|
column.type = JSON()
|
|
# Convert ARRAY to JSON (SQLite doesn't support ARRAY)
|
|
elif isinstance(column.type, ARRAY):
|
|
column.type = JSON()
|
|
|
|
# Create tables
|
|
async with engine.begin() as conn:
|
|
await conn.run_sync(Base.metadata.create_all)
|
|
|
|
# Create session
|
|
async_session_maker = sessionmaker(
|
|
engine, class_=AsyncSession, expire_on_commit=False
|
|
)
|
|
|
|
async with async_session_maker() as session:
|
|
yield session
|
|
|
|
# Cleanup
|
|
await engine.dispose()
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_connector_config():
|
|
"""Mock connector configuration."""
|
|
return {
|
|
"tokens": [
|
|
{
|
|
"chain": "ethereum",
|
|
"address": "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2",
|
|
"name": "WETH",
|
|
},
|
|
{
|
|
"chain": "solana",
|
|
"address": "So11111111111111111111111111111111111111112",
|
|
"name": "SOL",
|
|
},
|
|
]
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_pair_data():
|
|
"""Mock DexScreener API response data."""
|
|
return {
|
|
"pairs": [
|
|
{
|
|
"chainId": "ethereum",
|
|
"dexId": "uniswap",
|
|
"url": "https://dexscreener.com/ethereum/0x123",
|
|
"pairAddress": "0x123",
|
|
"baseToken": {
|
|
"address": "0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2",
|
|
"name": "Wrapped Ether",
|
|
"symbol": "WETH",
|
|
},
|
|
"quoteToken": {
|
|
"address": "0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48",
|
|
"name": "USD Coin",
|
|
"symbol": "USDC",
|
|
},
|
|
"priceNative": "1.0",
|
|
"priceUsd": "2500.00",
|
|
"txns": {
|
|
"m5": {"buys": 10, "sells": 5},
|
|
"h1": {"buys": 100, "sells": 50},
|
|
"h6": {"buys": 500, "sells": 250},
|
|
"h24": {"buys": 2000, "sells": 1000},
|
|
},
|
|
"volume": {"h24": 1000000.0, "h6": 250000.0, "h1": 50000.0, "m5": 5000.0},
|
|
"priceChange": {"m5": 0.5, "h1": 1.2, "h6": 2.5, "h24": 5.0},
|
|
"liquidity": {"usd": 5000000.0, "base": 2000.0, "quote": 5000000.0},
|
|
"fdv": 10000000.0,
|
|
"pairCreatedAt": 1609459200000,
|
|
}
|
|
]
|
|
}
|