86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
"""Test step definitions."""
|
|
|
|
from prometheus_test import TestStep
|
|
from functools import partial
|
|
from .stages import (
|
|
validate_api_keys,
|
|
validate_github,
|
|
fetch_summarizer_todo,
|
|
generate_summary,
|
|
submit_summary,
|
|
audit_summary,
|
|
)
|
|
|
|
steps = [
|
|
TestStep(
|
|
name="validate_api_keys",
|
|
description="Validate Anthropic API key",
|
|
prepare=validate_api_keys.prepare,
|
|
execute=validate_api_keys.execute,
|
|
worker="worker1",
|
|
),
|
|
TestStep(
|
|
name="validate_github",
|
|
description="Validate GitHub credentials",
|
|
prepare=validate_github.prepare,
|
|
execute=validate_github.execute,
|
|
worker="worker1",
|
|
),
|
|
TestStep(
|
|
name="fetch_todo_worker1",
|
|
description="Fetch summarizer todo for worker1",
|
|
prepare=fetch_summarizer_todo.prepare,
|
|
execute=fetch_summarizer_todo.execute,
|
|
worker="worker1",
|
|
),
|
|
TestStep(
|
|
name="fetch_todo_worker2",
|
|
description="Fetch summarizer todo for worker2",
|
|
prepare=fetch_summarizer_todo.prepare,
|
|
execute=fetch_summarizer_todo.execute,
|
|
worker="worker2",
|
|
),
|
|
TestStep(
|
|
name="generate_summary_worker1",
|
|
description="Generate summary for worker1's todo",
|
|
prepare=generate_summary.prepare,
|
|
execute=generate_summary.execute,
|
|
worker="worker1",
|
|
),
|
|
TestStep(
|
|
name="generate_summary_worker2",
|
|
description="Generate summary for worker2's todo",
|
|
prepare=generate_summary.prepare,
|
|
execute=generate_summary.execute,
|
|
worker="worker2",
|
|
),
|
|
TestStep(
|
|
name="submit_summary_worker1",
|
|
description="Submit summary for worker1",
|
|
prepare=submit_summary.prepare,
|
|
execute=submit_summary.execute,
|
|
worker="worker1",
|
|
),
|
|
TestStep(
|
|
name="submit_summary_worker2",
|
|
description="Submit summary for worker2",
|
|
prepare=submit_summary.prepare,
|
|
execute=submit_summary.execute,
|
|
worker="worker2",
|
|
),
|
|
TestStep(
|
|
name="audit_worker1",
|
|
description="Worker1 audits Worker2's submission",
|
|
prepare=partial(audit_summary.prepare, target_name="worker2"),
|
|
execute=audit_summary.execute,
|
|
worker="worker1",
|
|
),
|
|
TestStep(
|
|
name="audit_worker2",
|
|
description="Worker2 audits Worker1's submission",
|
|
prepare=partial(audit_summary.prepare, target_name="worker1"),
|
|
execute=audit_summary.execute,
|
|
worker="worker2",
|
|
),
|
|
]
|