119 lines
3.3 KiB
Go
119 lines
3.3 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/payment"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetPublicOrderByResumeTokenReturnsMatchingOrder(t *testing.T) {
|
|
ctx := context.Background()
|
|
client := newPaymentConfigServiceTestClient(t)
|
|
user, err := client.User.Create().
|
|
SetEmail("resume@example.com").
|
|
SetPasswordHash("hash").
|
|
SetUsername("resume-user").
|
|
Save(ctx)
|
|
require.NoError(t, err)
|
|
|
|
instanceID := "12"
|
|
providerKey := payment.TypeEasyPay
|
|
order, err := client.PaymentOrder.Create().
|
|
SetUserID(user.ID).
|
|
SetUserEmail(user.Email).
|
|
SetUserName(user.Username).
|
|
SetAmount(88).
|
|
SetPayAmount(88).
|
|
SetFeeRate(0).
|
|
SetRechargeCode("RESUME-ORDER").
|
|
SetOutTradeNo("sub2_resume_lookup").
|
|
SetPaymentType(payment.TypeAlipay).
|
|
SetPaymentTradeNo("trade-1").
|
|
SetOrderType(payment.OrderTypeBalance).
|
|
SetStatus(OrderStatusPending).
|
|
SetExpiresAt(time.Now().Add(time.Hour)).
|
|
SetClientIP("127.0.0.1").
|
|
SetSrcHost("api.example.com").
|
|
SetProviderInstanceID(instanceID).
|
|
SetProviderKey(providerKey).
|
|
Save(ctx)
|
|
require.NoError(t, err)
|
|
|
|
resumeSvc := NewPaymentResumeService([]byte("0123456789abcdef0123456789abcdef"))
|
|
token, err := resumeSvc.CreateToken(ResumeTokenClaims{
|
|
OrderID: order.ID,
|
|
UserID: user.ID,
|
|
ProviderInstanceID: instanceID,
|
|
ProviderKey: providerKey,
|
|
PaymentType: payment.TypeAlipay,
|
|
CanonicalReturnURL: "https://app.example.com/payment/result",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
svc := &PaymentService{
|
|
entClient: client,
|
|
resumeService: resumeSvc,
|
|
}
|
|
|
|
got, err := svc.GetPublicOrderByResumeToken(ctx, token)
|
|
require.NoError(t, err)
|
|
require.Equal(t, order.ID, got.ID)
|
|
}
|
|
|
|
func TestGetPublicOrderByResumeTokenRejectsSnapshotMismatch(t *testing.T) {
|
|
ctx := context.Background()
|
|
client := newPaymentConfigServiceTestClient(t)
|
|
user, err := client.User.Create().
|
|
SetEmail("resume-mismatch@example.com").
|
|
SetPasswordHash("hash").
|
|
SetUsername("resume-mismatch-user").
|
|
Save(ctx)
|
|
require.NoError(t, err)
|
|
|
|
order, err := client.PaymentOrder.Create().
|
|
SetUserID(user.ID).
|
|
SetUserEmail(user.Email).
|
|
SetUserName(user.Username).
|
|
SetAmount(88).
|
|
SetPayAmount(88).
|
|
SetFeeRate(0).
|
|
SetRechargeCode("RESUME-MISMATCH").
|
|
SetOutTradeNo("sub2_resume_lookup_mismatch").
|
|
SetPaymentType(payment.TypeAlipay).
|
|
SetPaymentTradeNo("trade-2").
|
|
SetOrderType(payment.OrderTypeBalance).
|
|
SetStatus(OrderStatusPending).
|
|
SetExpiresAt(time.Now().Add(time.Hour)).
|
|
SetClientIP("127.0.0.1").
|
|
SetSrcHost("api.example.com").
|
|
SetProviderInstanceID("12").
|
|
SetProviderKey(payment.TypeEasyPay).
|
|
Save(ctx)
|
|
require.NoError(t, err)
|
|
|
|
resumeSvc := NewPaymentResumeService([]byte("0123456789abcdef0123456789abcdef"))
|
|
token, err := resumeSvc.CreateToken(ResumeTokenClaims{
|
|
OrderID: order.ID,
|
|
UserID: user.ID,
|
|
ProviderInstanceID: "99",
|
|
ProviderKey: payment.TypeEasyPay,
|
|
PaymentType: payment.TypeAlipay,
|
|
CanonicalReturnURL: "https://app.example.com/payment/result",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
svc := &PaymentService{
|
|
entClient: client,
|
|
resumeService: resumeSvc,
|
|
}
|
|
|
|
_, err = svc.GetPublicOrderByResumeToken(ctx, token)
|
|
require.Error(t, err)
|
|
require.Contains(t, err.Error(), "resume token")
|
|
}
|