mirror of
https://github.com/GMWalletApp/epusdt.git
synced 2026-07-07 18:26:16 +00:00
line-6000
121 lines
3.5 KiB
Go
121 lines
3.5 KiB
Go
package service
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/GMWalletApp/epusdt/internal/testutil"
|
|
"github.com/GMWalletApp/epusdt/model/dao"
|
|
"github.com/GMWalletApp/epusdt/model/data"
|
|
"github.com/GMWalletApp/epusdt/model/mdb"
|
|
"github.com/GMWalletApp/epusdt/notify"
|
|
)
|
|
|
|
func TestHandleOkPayNotifySendsPaymentNotificationOnSuccess(t *testing.T) {
|
|
cleanup := testutil.SetupTestDatabases(t)
|
|
defer cleanup()
|
|
|
|
const (
|
|
channelType = "test-okpay-pay-success"
|
|
shopID = "okpay-shop-notify"
|
|
shopToken = "okpay-shop-token"
|
|
tradeID = "okpay-notify-success-001"
|
|
orderID = "merchant-okpay-notify-success-001"
|
|
providerOrderID = "okpay-provider-success-001"
|
|
rawFormData = "raw-okpay-form"
|
|
)
|
|
|
|
got := make(chan string, 1)
|
|
notify.RegisterSender(channelType, func(config, text string) error {
|
|
got <- text
|
|
return nil
|
|
})
|
|
|
|
if err := dao.Mdb.Create(&mdb.NotificationChannel{
|
|
Type: channelType,
|
|
Name: "test okpay pay success",
|
|
Config: "{}",
|
|
Events: `{"pay_success":true}`,
|
|
Enabled: true,
|
|
}).Error; err != nil {
|
|
t.Fatalf("seed notification channel: %v", err)
|
|
}
|
|
if err := data.SetSetting(mdb.SettingGroupOkPay, mdb.SettingKeyOkPayShopID, shopID, mdb.SettingTypeString); err != nil {
|
|
t.Fatalf("seed okpay shop id: %v", err)
|
|
}
|
|
if err := data.SetSetting(mdb.SettingGroupOkPay, mdb.SettingKeyOkPayShopToken, shopToken, mdb.SettingTypeString); err != nil {
|
|
t.Fatalf("seed okpay shop token: %v", err)
|
|
}
|
|
|
|
order := &mdb.Orders{
|
|
TradeId: tradeID,
|
|
OrderId: orderID,
|
|
Amount: 1,
|
|
Currency: "CNY",
|
|
ActualAmount: 0.15,
|
|
ReceiveAddress: "OKPAY",
|
|
Token: "USDT",
|
|
Network: mdb.NetworkTron,
|
|
Status: mdb.StatusWaitPay,
|
|
PaymentType: mdb.PaymentTypeEpay,
|
|
PayProvider: mdb.PaymentProviderOkPay,
|
|
}
|
|
if err := dao.Mdb.Create(order).Error; err != nil {
|
|
t.Fatalf("seed order: %v", err)
|
|
}
|
|
if err := dao.Mdb.Create(&mdb.ProviderOrder{
|
|
TradeId: tradeID,
|
|
Provider: mdb.PaymentProviderOkPay,
|
|
ProviderOrderID: providerOrderID,
|
|
Amount: 0.15,
|
|
Coin: "USDT",
|
|
Status: mdb.ProviderOrderStatusPending,
|
|
}).Error; err != nil {
|
|
t.Fatalf("seed provider order: %v", err)
|
|
}
|
|
|
|
form := okPayNotifyTestForm(shopID, shopToken, providerOrderID, tradeID, "0.15000000", "USDT")
|
|
if err := HandleOkPayNotify(form, rawFormData); err != nil {
|
|
t.Fatalf("handle okpay notify: %v", err)
|
|
}
|
|
|
|
select {
|
|
case text := <-got:
|
|
if !strings.Contains(text, tradeID) {
|
|
t.Fatalf("notification text = %q, want trade id %s", text, tradeID)
|
|
}
|
|
if !strings.Contains(text, orderID) {
|
|
t.Fatalf("notification text = %q, want order id %s", text, orderID)
|
|
}
|
|
case <-time.After(2 * time.Second):
|
|
t.Fatal("timed out waiting for payment notification")
|
|
}
|
|
|
|
if err := HandleOkPayNotify(form, rawFormData); err != nil {
|
|
t.Fatalf("handle duplicate okpay notify: %v", err)
|
|
}
|
|
select {
|
|
case text := <-got:
|
|
t.Fatalf("duplicate notification sent: %q", text)
|
|
case <-time.After(200 * time.Millisecond):
|
|
}
|
|
}
|
|
|
|
func okPayNotifyTestForm(shopID, shopToken, providerOrderID, tradeID, amount, coin string) map[string]string {
|
|
form := map[string]string{
|
|
"code": "200",
|
|
"id": shopID,
|
|
"status": "success",
|
|
"data[order_id]": providerOrderID,
|
|
"data[unique_id]": tradeID,
|
|
"data[pay_user_id]": "123456789",
|
|
"data[amount]": amount,
|
|
"data[coin]": coin,
|
|
"data[status]": "1",
|
|
"data[type]": "deposit",
|
|
}
|
|
form["sign"] = okPayNotifySign(form, shopToken)
|
|
return form
|
|
}
|