import pytest
from tortoise import Tortoise

@pytest.fixture(scope="session")
async def initialize_tests(request):
    """Initialize test database"""
    db_config = {
        'connections': {
            'default': {
                'engine': 'tortoise.backends.sqlite',
                'credentials': {
                    'file_path': ':memory:',
                }
            },
        },
        'apps': {
            'models': {
                'models': [
                    'App.routers.stocks.models',
                    'App.routers.tasks.models',
                    'App.routers.utt.models',
                    'App.routers.users.models',
                    'App.routers.portfolio.models',
                    'App.routers.bonds.models'
                ],
                'default_connection': 'default',
            }
        }
    }
    
    await Tortoise.init(config=db_config)
    await Tortoise.generate_schemas()
    
    yield
    
    await Tortoise.close_connections()

@pytest.fixture
async def client():
    """Create a test client"""
    from httpx import AsyncClient
    from main import app
    
    async with AsyncClient(app=app, base_url="http://test") as client:
        yield client