96 lines
3.0 KiB
Go
96 lines
3.0 KiB
Go
package app
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"hyapp/pkg/walletmq"
|
|
"hyapp/services/wallet-service/internal/config"
|
|
mysqlstorage "hyapp/services/wallet-service/internal/storage/mysql"
|
|
)
|
|
|
|
func TestWalletOutboxSingleRecordLeaseCoversPublishAndStateWrite(t *testing.T) {
|
|
if got := walletOutboxSingleRecordLease(3 * time.Second); got != 9*time.Second {
|
|
t.Fatalf("single-record lease = %s, want 9s", got)
|
|
}
|
|
if got := walletOutboxSingleRecordLease(0); got != 9*time.Second {
|
|
t.Fatalf("default single-record lease = %s, want 9s", got)
|
|
}
|
|
}
|
|
|
|
func TestWalletOutboxRocketMessageUsesConfiguredTagMode(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
mode config.WalletEventTagMode
|
|
wantTag string
|
|
}{
|
|
{
|
|
name: "missing config remains legacy",
|
|
mode: "",
|
|
wantTag: walletmq.TagWalletOutboxEvent,
|
|
},
|
|
{
|
|
name: "phase a publishes legacy",
|
|
mode: config.WalletEventTagModeLegacy,
|
|
wantTag: walletmq.TagWalletOutboxEvent,
|
|
},
|
|
{
|
|
name: "phase b publishes event type",
|
|
mode: config.WalletEventTagModeEventType,
|
|
wantTag: walletmq.EventTypeWalletGiftDebited,
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
message, err := walletOutboxRocketMessage("hyapp_wallet_outbox", test.mode, validWalletOutboxRecord())
|
|
if err != nil {
|
|
t.Fatalf("walletOutboxRocketMessage failed: %v", err)
|
|
}
|
|
if message.Tag != test.wantTag {
|
|
t.Fatalf("message tag = %q, want %q", message.Tag, test.wantTag)
|
|
}
|
|
decoded, err := walletmq.DecodeWalletOutboxMessage(message.Body)
|
|
if err != nil {
|
|
t.Fatalf("DecodeWalletOutboxMessage failed: %v", err)
|
|
}
|
|
if decoded.EventType != walletmq.EventTypeWalletGiftDebited {
|
|
t.Fatalf("body event_type = %q", decoded.EventType)
|
|
}
|
|
if message.Topic != "hyapp_wallet_outbox" || len(message.Keys) != 3 {
|
|
t.Fatalf("message routing metadata mismatch: %+v", message)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestWalletOutboxRocketMessageRejectsUnsupportedTagMode(t *testing.T) {
|
|
_, err := walletOutboxRocketMessage("hyapp_wallet_outbox", config.WalletEventTagMode("all"), validWalletOutboxRecord())
|
|
if err == nil {
|
|
t.Fatal("walletOutboxRocketMessage unexpectedly accepted an unsupported tag mode")
|
|
}
|
|
}
|
|
|
|
func TestWalletOutboxRocketMessageRejectsUnsafeEventTypeTag(t *testing.T) {
|
|
record := validWalletOutboxRecord()
|
|
record.EventType = "WalletGiftDebited || *"
|
|
_, err := walletOutboxRocketMessage("hyapp_wallet_outbox", config.WalletEventTagModeLegacy, record)
|
|
if err == nil {
|
|
t.Fatal("walletOutboxRocketMessage unexpectedly accepted an unsafe event_type")
|
|
}
|
|
}
|
|
|
|
func validWalletOutboxRecord() mysqlstorage.WalletOutboxRecord {
|
|
return mysqlstorage.WalletOutboxRecord{
|
|
AppCode: "lalu",
|
|
EventID: "event-1",
|
|
EventType: " " + walletmq.EventTypeWalletGiftDebited + " ",
|
|
TransactionID: "transaction-1",
|
|
CommandID: "command-1",
|
|
UserID: 1001,
|
|
AssetType: "COIN",
|
|
AvailableDelta: -100,
|
|
PayloadJSON: `{"gift_id":"gift-1"}`,
|
|
CreatedAtMS: 1_700_000_000_000,
|
|
}
|
|
}
|