iotlcm/worker-svr/internal/consumer/rabbitmq/balance_recharge_test.go

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)
			}
		})
	}
}