import datetime

from bson import ObjectId, DatetimeMS


class TestCreator:
    def __init__(self, db):
        self.db = db
        self.test_obj = None

    def create(self, record_ids: list[str]) -> ObjectId:
        record_results = {}
        for batch_id in record_ids:
            record_results[batch_id] = {"status": "pending"}

        test_obj = {
            "status": "pending",
            "created_at": datetime.datetime.now(datetime.UTC),
            "pipeline_version": "1.0.0",
            "record_results": record_results
        }
        insert_result = self.db.test_evaluation.insert_one(test_obj)
        return insert_result.inserted_id