from sqlalchemy.orm import Session
from . import models, schemas

def get_test(db: Session, test_id: int):
    return db.query(models.Test).filter(models.Test.id == test_id).first()

def get_tests(db: Session, skip: int = 0, limit: int = 100):
    return db.query(models.Test).offset(skip).limit(limit).all()

def create_test(db: Session, test: schemas.TestCreate):
    db_test = models.Test(name=test.name, description=test.description)
    db.add(db_test)
    db.commit()
    db.refresh(db_test)
    return db_test

def update_test(db: Session, db_test: models.Test, updates: schemas.TestUpdate):
    if updates.name is not None:
        db_test.name = updates.name
    if updates.description is not None:
        db_test.description = updates.description
    db.commit()
    db.refresh(db_test)
    return db_test

def delete_test(db: Session, db_test: models.Test):
    db.delete(db_test)
    db.commit()