feat(feedback): move feedback poster from WIP into pilot/
This commit is contained in:
@@ -0,0 +1,97 @@
|
||||
"""Tests for pilot/feedback_post.py — report delivery to Gitea.
|
||||
|
||||
Mock `ai_review.gitea_get` + `gitea_post` so we exercise the find-or-create
|
||||
+ comment-post flow without hitting the real API.
|
||||
"""
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
import tempfile
|
||||
import unittest
|
||||
from unittest.mock import patch
|
||||
|
||||
HERE = os.path.dirname(os.path.abspath(__file__))
|
||||
sys.path.insert(0, os.path.join(HERE, "..", "..", "pilot"))
|
||||
|
||||
import feedback # noqa: E402
|
||||
import feedback_analyze # noqa: E402
|
||||
import feedback_post # noqa: E402
|
||||
|
||||
|
||||
def _make_fake(method_routes: dict):
|
||||
"""`method_routes` maps HTTP path substring → (status, body, method).
|
||||
|
||||
For our purposes both gitea_get and gitea_post share the same fake —
|
||||
gitea_get is GET, gitea_post is POST, and the post helper also has a
|
||||
body param. The fake returns whatever the route's body says.
|
||||
"""
|
||||
def fake_get(api, repo, path, token, accept="application/json"):
|
||||
for needle in sorted(method_routes.keys(), key=len, reverse=True):
|
||||
status, body, _m = method_routes[needle]
|
||||
if needle in path:
|
||||
return status, json.dumps(body).encode()
|
||||
return 404, b'{"message":"not found"}'
|
||||
|
||||
def fake_post(api, repo, path, token, body):
|
||||
for needle in sorted(method_routes.keys(), key=len, reverse=True):
|
||||
status, resp_body, _m = method_routes[needle]
|
||||
if needle in path:
|
||||
return status, json.dumps(resp_body).encode()
|
||||
return 404, b'{"message":"not found"}'
|
||||
|
||||
return fake_get, fake_post
|
||||
|
||||
|
||||
class TestDeliver(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.tmp = tempfile.TemporaryDirectory()
|
||||
self.db = f"{self.tmp.name}/f.db"
|
||||
conn = feedback.init(self.db)
|
||||
rid = feedback.record_review(conn, repo="o/r", pr=1, head_sha="x")
|
||||
feedback.record_inline_finding(
|
||||
conn, review_id=rid, repo="o/r", pr=1,
|
||||
path="a.ts", line=1, severity="HIGH",
|
||||
problem="x", comment_id=99,
|
||||
)
|
||||
conn.close()
|
||||
|
||||
def tearDown(self):
|
||||
self.tmp.cleanup()
|
||||
|
||||
def test_creates_issue_then_posts_comment(self):
|
||||
routes = {
|
||||
"issues?state=open": (200, [], "GET"), # no existing issue
|
||||
"issues": (201, {"id": 42, "number": 7, "title": "..."}, "POST"),
|
||||
"issues/7/comments": (201, {"id": 777}, "POST"),
|
||||
}
|
||||
fake_get, fake_post = _make_fake(routes)
|
||||
with patch("ai_review.gitea_get", side_effect=fake_get), \
|
||||
patch("ai_review.gitea_post", side_effect=fake_post):
|
||||
stats = feedback_post.deliver(
|
||||
api="http://x", token="t", db_path=self.db,
|
||||
repo="gitea_admin/pragent", title="pragent feedback roll-up",
|
||||
)
|
||||
self.assertEqual(stats["issue_id"], 7)
|
||||
self.assertEqual(stats["comment_id"], 777)
|
||||
|
||||
def test_reuses_existing_issue(self):
|
||||
routes = {
|
||||
"issues?state=open": (200, [
|
||||
{"id": 99, "number": 9, "title": "pragent feedback roll-up"},
|
||||
{"id": 100, "number": 10, "title": "something else"},
|
||||
], "GET"),
|
||||
"issues/9/comments": (201, {"id": 888}, "POST"),
|
||||
}
|
||||
fake_get, fake_post = _make_fake(routes)
|
||||
with patch("ai_review.gitea_get", side_effect=fake_get), \
|
||||
patch("ai_review.gitea_post", side_effect=fake_post):
|
||||
stats = feedback_post.deliver(
|
||||
api="http://x", token="t", db_path=self.db,
|
||||
repo="gitea_admin/pragent", title="pragent feedback roll-up",
|
||||
)
|
||||
self.assertEqual(stats["issue_id"], 9)
|
||||
self.assertEqual(stats["comment_id"], 888)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user