sync: 同步 v1.0.4 开源代码并脱敏

internal/client/upstream/client_timeout_test.go

@@ -0,0 +1,49 @@
+package upstream
+
+import (
+	"testing"
+	"time"
+)
+
+// TestResolveUpstreamTimeoutScalesWithCardCount 覆盖批量操作预算：
+// worker 不能再把批量请求压回 2 分钟，否则通道仍在按卡串行办理时会被提前判失败。
+func TestResolveUpstreamTimeoutScalesWithCardCount(t *testing.T) {
+	cases := []struct {
+		name       string
+		path       string
+		itemCount  int
+		perCard    time.Duration
+		override   time.Duration
+		maxTimeout time.Duration
+		apnTimeout time.Duration
+		want       time.Duration
+	}{
+		{"单卡", "/sim/change/sim-status", 1, 10 * time.Second, 0, defaultUpstreamProxyMaxTimeout, 240 * time.Second, 10 * time.Second},
+		{"零卡按单卡处理", "/sim/change/sim-status", 0, 10 * time.Second, 0, defaultUpstreamProxyMaxTimeout, 240 * time.Second, 10 * time.Second},
+		{"50卡按单卡预算放大", "/sim/change/sim-status", 50, 10 * time.Second, 0, defaultUpstreamProxyMaxTimeout, 240 * time.Second, 500 * time.Second},
+		{"100卡恰好低于上限", "/sim/change/sim-status", 100, 10 * time.Second, 0, defaultUpstreamProxyMaxTimeout, 240 * time.Second, 1000 * time.Second},
+		{"超大批量按上限截断", "/sim/query/usage", 1000, 10 * time.Second, 0, defaultUpstreamProxyMaxTimeout, 240 * time.Second, defaultUpstreamProxyMaxTimeout},
+		{"单卡预算缺失用默认值", "/sim/query/usage", 2, 0, 0, defaultUpstreamProxyMaxTimeout, 240 * time.Second, 20 * time.Second},
+		{"显式覆盖仍按卡数放大", "/sim/query/usage", 3, 10 * time.Second, 2 * time.Second, defaultUpstreamProxyMaxTimeout, 240 * time.Second, 6 * time.Second},
+		{"数据开停使用独立预算", "/sim/change/sim-apn", 50, 10 * time.Second, 0, defaultUpstreamProxyMaxTimeout, 240 * time.Second, 240 * time.Second},
+		{"数据开停预算缺失回退", "/sim/change/sim-apn", 1, 10 * time.Second, 0, defaultUpstreamProxyMaxTimeout, 0, 240 * time.Second},
+	}
+	for _, c := range cases {
+		t.Run(c.name, func(t *testing.T) {
+			got := resolveUpstreamTimeout(c.path, c.itemCount, c.perCard, c.override, c.maxTimeout, c.apnTimeout)
+			if got != c.want {
+				t.Fatalf("resolveUpstreamTimeout=%s，期望%s", got, c.want)
+			}
+		})
+	}
+}
+
+// TestUpstreamMaxTimeoutOutlastsProxyBatchTimeout 固化跨服务预算约束：
+// worker 上限必须大于 proxy 的 gateway.batchTimeout（默认 20 分钟），
+// 使超时由更靠近通道的一层先判定，避免 worker 提前放弃而通道仍在办理。
+func TestUpstreamMaxTimeoutOutlastsProxyBatchTimeout(t *testing.T) {
+	const proxyBatchTimeout = 20 * time.Minute
+	if defaultUpstreamProxyMaxTimeout <= proxyBatchTimeout {
+		t.Fatalf("worker 上限%s 必须大于 proxy 批量上限%s", defaultUpstreamProxyMaxTimeout, proxyBatchTimeout)
+	}
+}