import asyncio
from common.pool_wrapper import pool_wrapper
from worker.stylist_engine import StylistEngine
import sys

async def run():
    await pool_wrapper.init_all()
    engine = StylistEngine()
    db_pool = pool_wrapper
    print("Testing 6OT25S007...")
    res1 = engine.generate_matches(db_pool, "6OT25S007")
    print(f"Matches for 6OT25S007: {len(res1)}")

    print("\nTesting 2LA26S003-SL388...")
    res2 = engine.generate_matches(db_pool, "2LA26S003-SL388")
    print(f"Matches for 2LA26S003-SL388: {len(res2)}")
    
    await pool_wrapper.close_all()

asyncio.run(run())
