feat(api): add load-aware client fallback

2026-05-12 16:59:51 +08:00
parent c5cede2359
commit c2696e7bbe
7 changed files with 558 additions and 5 deletions
@@ -46,6 +46,24 @@ func TestFailoverTimeBudgetExceeded(t *testing.T) {
 	}
 }

+func TestLoadAvoidanceFallbackContinuesToAvoidedCandidate(t *testing.T) {
+	candidates := []store.RuntimeModelCandidate{
+		{PlatformID: "available-candidate"},
+		{PlatformID: "avoided-full-candidate", LoadAvoided: true},
+	}
+
+	if !hasLoadAvoidanceFallback(candidates, 0, 99) {
+		t.Fatal("expected non-avoided candidate to fall back to later avoided candidate")
+	}
+	if hasLoadAvoidanceFallback(candidates, 1, 99) {
+		t.Fatal("avoided candidate should not force another load-avoidance fallback")
+	}
+	decision := loadAvoidanceFallbackDecision(&clients.ClientError{Code: "bad_request", StatusCode: 400, Retryable: false})
+	if !decision.Retry || decision.Reason != "load_avoidance_fallback" || decision.Action != "next" {
+		t.Fatalf("expected active load avoidance fallback to force next candidate, got %+v", decision)
+	}
+}
+
 func TestFailoverHardStopBeatsModelOverride(t *testing.T) {
 	runnerPolicy := store.RunnerPolicy{
 		Status: "active",