chore: 并入 iotlcm-worker-svr 历史到 worker-svr/（原提交 96f0f4e）

2026-11-02 13:28:03 +08:00
parent 6de2e503cc 96f0f4eacd
commit 2923c78755
422 changed files with 80993 additions and 0 deletions
@@ -0,0 +1,153 @@
+package rabbitmq
+
+import (
+	"context"
+	"errors"
+	"testing"
+	"time"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gcfg"
+
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/consts"
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model"
+)
+
+func stubBalanceRechargePublish(t *testing.T) {
+	t.Helper()
+	originalPublish, originalDelayed := balanceRechargePublish, balanceRechargePublishDelayed
+	t.Cleanup(func() {
+		balanceRechargePublish, balanceRechargePublishDelayed = originalPublish, originalDelayed
+	})
+	balanceRechargePublish = func(context.Context, string, any) error { return nil }
+	balanceRechargePublishDelayed = func(context.Context, string, string, time.Duration, any) error { return nil }
+}
+
+func setBalanceRechargeTestConfig(t *testing.T, content string) {
+	t.Helper()
+	oldAdapter := g.Cfg().GetAdapter()
+	adapter, err := gcfg.NewAdapterContent(content)
+	if err != nil {
+		t.Fatal(err)
+	}
+	g.Cfg().SetAdapter(adapter)
+	t.Cleanup(func() { g.Cfg().SetAdapter(oldAdapter) })
+}
+
+func TestHandleBalanceRechargeTimeoutRepublishesDelayedBelowMaxRetry(t *testing.T) {
+	stubBalanceRechargePublish(t)
+	consumers := New(&trafficUsageHandlerStub{}, &workOrderHandlerStub{}, &balanceRechargeHandlerStub{err: errors.New("Alipay credentials are not configured")})
+	consumers.balanceRechargeRetryDelay = time.Minute
+	consumers.balanceRechargeMaxRetry = 3
+
+	delayedCalls := 0
+	balanceRechargePublishDelayed = func(_ context.Context, delayQueue, targetQueue string, delay time.Duration, data any) error {
+		delayedCalls++
+		if delayQueue != consts.QueueBalanceRechargeDelay || targetQueue != consts.QueueBalanceRechargeTimeout || delay != time.Minute {
+			t.Fatalf("unexpected delayed publish: %s -> %s, delay:%s", delayQueue, targetQueue, delay)
+		}
+		message, ok := data.(model.BalanceRechargeTimeoutMessage)
+		if !ok || message.OutTradeNo != "ALI-1" || message.RetryCount != 3 {
+			t.Fatalf("unexpected retry payload: %+v", data)
+		}
+		return nil
+	}
+	deadLetterCalls := 0
+	balanceRechargePublish = func(_ context.Context, queue string, _ any) error {
+		deadLetterCalls++
+		return nil
+	}
+
+	// 自增后达到上限仍未超过，应继续重投延迟队列。
+	if err := consumers.handleBalanceRechargeTimeout(context.Background(), model.BalanceRechargeTimeoutMessage{OutTradeNo: "ALI-1", RetryCount: 2}); err != nil {
+		t.Fatalf("retry below cap returned error: %v", err)
+	}
+	if delayedCalls != 1 || deadLetterCalls != 0 {
+		t.Fatalf("delayedCalls=%d deadLetterCalls=%d, want 1 and 0", delayedCalls, deadLetterCalls)
+	}
+}
+
+func TestHandleBalanceRechargeTimeoutDeadLettersAboveMaxRetry(t *testing.T) {
+	stubBalanceRechargePublish(t)
+	consumers := New(&trafficUsageHandlerStub{}, &workOrderHandlerStub{}, &balanceRechargeHandlerStub{err: errors.New("Alipay credentials are not configured")})
+	consumers.balanceRechargeRetryDelay = time.Minute
+	consumers.balanceRechargeMaxRetry = 3
+
+	delayedCalls := 0
+	balanceRechargePublishDelayed = func(context.Context, string, string, time.Duration, any) error {
+		delayedCalls++
+		return nil
+	}
+	deadLetterCalls := 0
+	balanceRechargePublish = func(_ context.Context, queue string, data any) error {
+		deadLetterCalls++
+		if queue != consts.QueueBalanceRechargeDeadLetter {
+			t.Fatalf("dead letter queue = %s, want %s", queue, consts.QueueBalanceRechargeDeadLetter)
+		}
+		message, ok := data.(model.BalanceRechargeTimeoutMessage)
+		if !ok || message.OutTradeNo != "ALI-1" || message.RetryCount != 4 {
+			t.Fatalf("unexpected dead-letter payload: %+v", data)
+		}
+		return nil
+	}
+
+	if err := consumers.handleBalanceRechargeTimeout(context.Background(), model.BalanceRechargeTimeoutMessage{OutTradeNo: "ALI-1", RetryCount: 3}); err != nil {
+		t.Fatalf("dead-letter path returned error: %v", err)
+	}
+	if delayedCalls != 0 || deadLetterCalls != 1 {
+		t.Fatalf("delayedCalls=%d deadLetterCalls=%d, want 0 and 1", delayedCalls, deadLetterCalls)
+	}
+}
+
+func TestHandleBalanceRechargeTimeoutReturnsPublishErrors(t *testing.T) {
+	stubBalanceRechargePublish(t)
+	consumers := New(&trafficUsageHandlerStub{}, &workOrderHandlerStub{}, &balanceRechargeHandlerStub{err: errors.New("Alipay credentials are not configured")})
+	consumers.balanceRechargeRetryDelay = time.Minute
+	consumers.balanceRechargeMaxRetry = 3
+
+	expected := errors.New("broker unavailable")
+	balanceRechargePublish = func(context.Context, string, any) error { return expected }
+	balanceRechargePublishDelayed = func(context.Context, string, string, time.Duration, any) error { return expected }
+
+	// 重投失败必须向上返回错误，让消费框架 Nack 重新入队，不能静默丢失消息。
+	if err := consumers.handleBalanceRechargeTimeout(context.Background(), model.BalanceRechargeTimeoutMessage{OutTradeNo: "ALI-1", RetryCount: 0}); !errors.Is(err, expected) {
+		t.Fatalf("delayed publish failure = %v, want %v", err, expected)
+	}
+	if err := consumers.handleBalanceRechargeTimeout(context.Background(), model.BalanceRechargeTimeoutMessage{OutTradeNo: "ALI-1", RetryCount: 3}); !errors.Is(err, expected) {
+		t.Fatalf("dead-letter publish failure = %v, want %v", err, expected)
+	}
+}
+
+func TestGetBalanceRechargeMaxRetry(t *testing.T) {
+	t.Run("defaults when unconfigured", func(t *testing.T) {
+		setBalanceRechargeTestConfig(t, "rabbitmq:\n  balanceRecharge:\n    retryDelay: \"1m\"\n")
+		maxRetry, err := getBalanceRechargeMaxRetry(context.Background())
+		if err != nil || maxRetry != defaultBalanceRechargeMaxRetry {
+			t.Fatalf("maxRetry = %d (err=%v), want %d", maxRetry, err, defaultBalanceRechargeMaxRetry)
+		}
+	})
+
+	t.Run("reads configured value", func(t *testing.T) {
+		setBalanceRechargeTestConfig(t, "rabbitmq:\n  balanceRecharge:\n    maxRetry: 7\n")
+		maxRetry, err := getBalanceRechargeMaxRetry(context.Background())
+		if err != nil || maxRetry != 7 {
+			t.Fatalf("maxRetry = %d (err=%v), want 7", maxRetry, err)
+		}
+	})
+
+	for _, test := range []struct {
+		name  string
+		value string
+	}{
+		{name: "zero", value: "0"},
+		{name: "negative", value: "-1"},
+		{name: "non-numeric", value: "\"abc\""},
+	} {
+		t.Run("rejects "+test.name, func(t *testing.T) {
+			setBalanceRechargeTestConfig(t, "rabbitmq:\n  balanceRecharge:\n    maxRetry: "+test.value+"\n")
+			if _, err := getBalanceRechargeMaxRetry(context.Background()); err == nil {
+				t.Fatalf("maxRetry=%s was accepted, want error", test.value)
+			}
+		})
+	}
+}