update mongodb config
This commit is contained in:
@ -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
|
||||
|
Reference in New Issue
Block a user