mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-14 12:02:27 +08:00
shaw
b491c3271b
#6669 renamed the constant from redeemMaxErrorsPerHour to redeemMaxFailedAttempts (fixed 10-minute window), but #6670's tests still referenced the old name, breaking compilation on main.
60 lines
2.2 KiB
Go
60 lines
2.2 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAdminFulfillmentBypassesLimitAndKeepsRedeemAffiliate(t *testing.T) {
|
|
ctx := context.Background()
|
|
client := newPaymentConfigServiceTestClient(t)
|
|
userID := int64(42)
|
|
inviterID := int64(9001)
|
|
code := &RedeemCode{
|
|
ID: 102, Code: "ADMIN-SUCCESS", Type: RedeemTypeBalance, Value: 20, Status: StatusUnused,
|
|
}
|
|
redeemRepo := &paymentFulfillmentRedeemRepo{
|
|
paymentOrderLifecycleRedeemRepo: paymentOrderLifecycleRedeemRepo{
|
|
codesByCode: map[string]*RedeemCode{code.Code: code},
|
|
},
|
|
}
|
|
userRepo := &mockUserRepo{getByIDUser: &User{ID: userID}}
|
|
userRepo.updateBalanceFn = func(context.Context, int64, float64) error { return nil }
|
|
cache := &paymentFulfillmentRedeemCacheStub{count: redeemMaxFailedAttempts}
|
|
affiliateRepo := &paymentFulfillmentAffiliateRepoStub{
|
|
inviteeSummary: &AffiliateSummary{
|
|
UserID: userID, AffCode: "INVITEE", InviterID: &inviterID, CreatedAt: time.Now().Add(-time.Hour),
|
|
},
|
|
inviterSummary: &AffiliateSummary{
|
|
UserID: inviterID, AffCode: "INVITER", CreatedAt: time.Now().Add(-2 * time.Hour),
|
|
},
|
|
}
|
|
settingSvc := NewSettingService(&paymentFulfillmentSettingRepoStub{values: map[string]string{
|
|
SettingKeyAffiliateEnabled: "true",
|
|
SettingKeyAffiliateRebateRate: "10",
|
|
SettingKeyAffiliateRebateFreezeHours: "0",
|
|
}}, nil)
|
|
affiliateSvc := NewAffiliateService(affiliateRepo, settingSvc, nil, nil)
|
|
svc := NewRedeemService(redeemRepo, userRepo, nil, cache, nil, client, nil, affiliateSvc)
|
|
|
|
result, err := svc.RedeemForAdminFulfillment(ctx, userID, code.Code)
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, result)
|
|
require.Zero(t, cache.getCalls)
|
|
require.Zero(t, cache.incrementCalls)
|
|
require.Equal(t, 1, cache.acquireCalls)
|
|
require.Equal(t, 1, cache.releaseCalls)
|
|
require.Len(t, redeemRepo.useCalls, 1)
|
|
require.Len(t, affiliateRepo.accrueCalls, 1)
|
|
require.Equal(t, inviterID, affiliateRepo.accrueCalls[0].inviterID)
|
|
require.Equal(t, userID, affiliateRepo.accrueCalls[0].inviteeUserID)
|
|
require.InDelta(t, 2, affiliateRepo.accrueCalls[0].amount, 1e-8)
|
|
require.Nil(t, affiliateRepo.accrueCalls[0].sourceOrderID)
|
|
}
|