tests/test_audit.py

import os
import sys

sys.path.append(os.path.abspath(os.path.dirname(__file__)))

from database import init_db, get_pg_engine, AuditLog
from sqlalchemy.orm import sessionmaker

def test_audit():
    init_db()
    engine = get_pg_engine()
    Session = sessionmaker(bind=engine)
    session = Session()
    
    log = AuditLog(
        user_id="test_key",
        endpoint="/test",
        request_hash="abcd",
        request_body={"test": 123},
        response_weights={"SPY": 1.0},
        ip_address="127.0.0.1"
    )
    
    session.add(log)
    session.commit()
    
    count = session.query(AuditLog).count()
    print(f"Audit log count: {count}")
    
if __name__ == "__main__":
    test_audit()