update mongodb config

This commit is contained in:
Laura Abro
2025-04-24 12:28:54 -03:00
parent 5e6911d4f8
commit 0b0298e400
7 changed files with 10 additions and 68 deletions

View File

@ -4,7 +4,6 @@ from pathlib import Path
from prometheus_test import TestRunner
import dotenv
import argparse
import uuid
dotenv.load_dotenv()
@ -20,26 +19,6 @@ def parse_args():
return parser.parse_args()
def add_uuids(db):
"""Post-load callback to process MongoDB data after JSON import"""
# Process docs collection
docs = list(db.docs.find({"taskId": runner.config.task_id}))
for doc in docs:
if "uuid" not in doc:
doc["uuid"] = str(uuid.uuid4())
db.docs.replace_one({"_id": doc["_id"]}, doc)
# Process summaries collection
summaries = list(db.summaries.find({"taskId": runner.config.task_id}))
for summary in summaries:
if "uuid" not in summary:
summary["uuid"] = str(uuid.uuid4())
if "docUuid" not in summary and docs:
# Link to first doc for simplicity
summary["docUuid"] = docs[0]["uuid"]
db.summaries.replace_one({"_id": summary["_id"]}, summary)
# Global reference to the test runner
runner = None
@ -56,7 +35,6 @@ def main():
runner = TestRunner(
steps=steps,
config_file=base_dir / "config.yaml",
config_overrides={"post_load_callback": add_uuids},
)
# Run test sequence