feat. Review foot note, docker fix, pass message to reviewer , update tests
This commit is contained in:
@@ -3,6 +3,7 @@ from __future__ import annotations
|
||||
from sqlalchemy.exc import IntegrityError
|
||||
|
||||
from gitea_codex_bot.db import get_session_factory
|
||||
from gitea_codex_bot.models import ReviewJob
|
||||
from gitea_codex_bot.services.jobs import cooldown_remaining_seconds, enqueue_job, persist_webhook_event
|
||||
from gitea_codex_bot.types import ParsedCommand
|
||||
|
||||
@@ -19,7 +20,16 @@ def test_enqueue_and_cooldown() -> None:
|
||||
session_factory = get_session_factory()
|
||||
with session_factory() as session:
|
||||
cmd = ParsedCommand(name="review", raw="@codex review")
|
||||
enqueue_job(session, repo="acme/repo", pr_number=42, head_sha="abc", trigger_comment_id=100, requested_by="user", command=cmd)
|
||||
enqueue_job(
|
||||
session,
|
||||
repo="acme/repo",
|
||||
pr_number=42,
|
||||
head_sha="abc",
|
||||
trigger_comment_id=100,
|
||||
trigger_comment_body="@codex review",
|
||||
requested_by="user",
|
||||
command=cmd,
|
||||
)
|
||||
remaining = cooldown_remaining_seconds(session, "acme/repo", 42, 60)
|
||||
assert remaining >= 0
|
||||
|
||||
@@ -28,11 +38,48 @@ def test_trigger_comment_unique() -> None:
|
||||
session_factory = get_session_factory()
|
||||
with session_factory() as session:
|
||||
cmd = ParsedCommand(name="review", raw="@codex review")
|
||||
enqueue_job(session, repo="acme/repo", pr_number=7, head_sha="x", trigger_comment_id=321, requested_by="user", command=cmd)
|
||||
enqueue_job(
|
||||
session,
|
||||
repo="acme/repo",
|
||||
pr_number=7,
|
||||
head_sha="x",
|
||||
trigger_comment_id=321,
|
||||
trigger_comment_body="@codex review",
|
||||
requested_by="user",
|
||||
command=cmd,
|
||||
)
|
||||
try:
|
||||
enqueue_job(session, repo="acme/repo", pr_number=7, head_sha="x", trigger_comment_id=321, requested_by="user", command=cmd)
|
||||
enqueue_job(
|
||||
session,
|
||||
repo="acme/repo",
|
||||
pr_number=7,
|
||||
head_sha="x",
|
||||
trigger_comment_id=321,
|
||||
trigger_comment_body="@codex review",
|
||||
requested_by="user",
|
||||
command=cmd,
|
||||
)
|
||||
duplicate_raised = False
|
||||
except IntegrityError:
|
||||
duplicate_raised = True
|
||||
session.rollback()
|
||||
assert duplicate_raised is True
|
||||
|
||||
|
||||
def test_enqueue_persists_full_trigger_comment_body() -> None:
|
||||
session_factory = get_session_factory()
|
||||
with session_factory() as session:
|
||||
cmd = ParsedCommand(name="review", raw="@codex review security\nplease focus auth")
|
||||
job = enqueue_job(
|
||||
session,
|
||||
repo="acme/repo",
|
||||
pr_number=55,
|
||||
head_sha="abc123",
|
||||
trigger_comment_id=9191,
|
||||
trigger_comment_body=cmd.raw,
|
||||
requested_by="alice",
|
||||
command=cmd,
|
||||
)
|
||||
stored = session.get(ReviewJob, job.id)
|
||||
assert stored is not None
|
||||
assert stored.trigger_comment_body == "@codex review security\nplease focus auth"
|
||||
|
||||
Reference in New Issue
Block a user