feat(box): add invite code handler, Settings RPA, and agent onboarding pipeline

- Support invite code discovery in main chat and redemption in settings menu - Add Settings RPA primitives with Radix UI mouse dispatch and retry polling for async DOM - Attribute 'has_redeemed' binary from the Additional tokens ticker / entrypoint visibility - Unblock agent @646 by repairing warp-def/dev tunnels and redeeming REDCJ7 via dev (+1B tokens) - Implement 'box onboard' pipeline to provision infra, authenticate, and auto-redeem queued codes - Add 'box onboard feed-matrix' ranking all agents by work done over time, job count, and role - Register 'InputType.SALVAGE' in loop modulation (CRITICAL priority, 600s timeout, 3 nudges to opm) - Ingest recurring balance audits into canonical HEARTBEAT.md and TOOLS.md
2026-10-06 19:33:17 +00:00
parent 4998ffddb6
commit b824f6d405
30 changed files with 4516 additions and 0 deletions
@@ -0,0 +1,88 @@
+#!/usr/bin/env python3
+"""test_onboard_pipeline.py — Unit tests for the onboarding and invite salvage pipeline."""
+
+import json
+import sys
+import unittest
+from pathlib import Path
+from unittest.mock import MagicMock, patch
+
+REPO_ROOT = Path(__file__).resolve().parent.parent
+sys.path.insert(0, str(REPO_ROOT / "bin"))
+
+from onboard_pipeline import (
+    OnboardState,
+    finish_onboarding_redemption,
+    select_urgent_beneficiary,
+    start_onboarding,
+    submit_onboarding_otp,
+)
+
+
+class TestOnboardPipeline(unittest.TestCase):
+    def test_select_urgent_beneficiary_explicit(self):
+        b_node, code, reason = select_urgent_beneficiary("dev", "TEST01")
+        self.assertEqual(b_node, "dev")
+        self.assertEqual(code, "TEST01")
+        self.assertIn("Explicit", reason)
+
+    @patch("invite.fleet_invite_status")
+    @patch("invite.fleet_usage")
+    def test_select_urgent_beneficiary_blocked_first(self, mock_usage, mock_invite):
+        mock_usage.return_value = {
+            "646": {"weekly_used_pct": 100, "additional_used_pct": 100, "additional_left": "0 tokens left"},
+            "pip": {"weekly_used_pct": 100, "additional_used_pct": 10, "additional_left": "890M tokens left"},
+            "muse": {"weekly_used_pct": 50, "additional_used_pct": 0, "additional_left": "950M tokens left"},
+        }
+        mock_invite.return_value = {
+            "646": {"code": "REDCJ7"},
+            "pip": {"code": "F4BGHN"},
+            "muse": {"code": "81MDIR"},
+        }
+
+        b_node, code, reason = select_urgent_beneficiary()
+        self.assertEqual(b_node, "646")
+        self.assertEqual(code, "REDCJ7")
+        self.assertIn("BLOCKED", reason)
+
+    @patch("onboard_pipeline.provision_node_infra")
+    @patch("onboard_pipeline.CredClient")
+    @patch("onboard_pipeline.select_urgent_beneficiary")
+    def test_start_onboarding_awaiting_otp(self, mock_select, mock_cred_cls, mock_infra):
+        mock_select.return_value = ("646", "REDCJ7", "Blocked test")
+        mock_infra.return_value = {"ok": True}
+
+        mock_client = MagicMock()
+        mock_client.initiate.return_value = {
+            "status": "awaiting_otp",
+            "message": "OTP sent",
+        }
+        mock_cred_cls.return_value = mock_client
+
+        res = start_onboarding("testnode", "client@test.com")
+        self.assertTrue(res["ok"])
+        self.assertEqual(res["status"], "awaiting_otp")
+        self.assertEqual(res["beneficiary_node"], "646")
+        self.assertEqual(res["invite_code_queued"], "REDCJ7")
+
+    @patch("invite.redeem_invite")
+    @patch("subprocess.run")
+    def test_finish_onboarding_redemption_success(self, mock_subproc, mock_redeem):
+        mock_redeem.return_value = {"ok": True, "redemption_status": "redeemed"}
+        state = OnboardState(
+            node="testnode",
+            email="client@test.com",
+            beneficiary_node="646",
+            invite_code="REDCJ7",
+            stage="auth_active",
+        )
+
+        res = finish_onboarding_redemption(state)
+        self.assertTrue(res["ok"])
+        self.assertEqual(res["status"], "completed")
+        self.assertEqual(res["beneficiary_node"], "646")
+        mock_redeem.assert_called_once_with("testnode", "REDCJ7", timeout=20.0)
+
+
+if __name__ == "__main__":
+    unittest.main()