Something went wrong. Try again.
This repository has no description
Something went wrong. Try again.
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859"""bar-pr and bar-reeds: the summarise step, fed fixture JSON on stdin."""
import jsonimport subprocessimport timeimport unittestfrom pathlib import Path
BIN = Path(__file__).resolve().parents[1] / "bin"
def run(tool, args, payload): return subprocess.run([str(BIN / tool), "summarise", *args], input=json.dumps(payload), capture_output=True, text=True).stdout.strip()
def pr(author, decision="REVIEW_REQUIRED", requested=(), draft=False): return {"author": {"login": author}, "reviewDecision": decision, "isDraft": draft, "reviewRequests": [{"login": r} for r in requested]}
class BarPr(unittest.TestCase): def test_counts_open_approved_and_requested_from_me(self): prs = [pr("me", "APPROVED"), pr("other", requested=("me",)), pr("other"), pr("me", draft=True)] out = run("bar-pr", ["ai-gateway", "me"], prs) self.assertIn("ai-gateway 3 open", out) self.assertIn("1 approved", out) self.assertIn("1 to review", out)
def test_zero_to_review_is_not_highlighted(self): out = run("bar-pr", ["ai-gateway", "me"], [pr("me")]) self.assertIn("0 to review", out) self.assertNotIn("#[fg=#f6c177]", out)
def test_my_own_pr_requesting_me_does_not_count(self): out = run("bar-pr", ["ai-gateway", "me"], [pr("me", requested=("me",))]) self.assertIn("0 to review", out)
def test_empty_repo_renders_nothing(self): self.assertEqual(run("bar-pr", ["ai-gateway", "me"], []), "")
NOW_MS = int(time.time() * 1000)
class BarReeds(unittest.TestCase): def test_counts_recent_needs_user_entries_only(self): state = {"entries": [{"kind": "needs-user", "ts": NOW_MS}, {"kind": "status", "ts": NOW_MS}, {"kind": "needs-user", "ts": NOW_MS - 3600_000}, {"kind": "needs-user", "ts": NOW_MS - 30 * 86400_000}]} self.assertIn("reeds 2 need you", run("bar-reeds", [], state))
def test_nothing_pending_renders_nothing(self): self.assertEqual(run("bar-reeds", [], {"entries": [{"kind": "done", "ts": NOW_MS}]}), "")
if __name__ == "__main__": unittest.main()