package rpc import ( "context" "testing" "github.com/iamxvbaba/td/bin" "github.com/iamxvbaba/td/clock" "github.com/iamxvbaba/td/tg" "github.com/iamxvbaba/td/tgerr" "go.uber.org/zap/zaptest" "github.com/iamxvbaba/td/tlprofile" appchannels "telesrv/internal/app/channels" appmessages "telesrv/internal/app/messages" appprivacy "telesrv/internal/app/privacy" appstargifts "telesrv/internal/app/stargifts" appstars "telesrv/internal/app/stars" appusers "telesrv/internal/app/users" "telesrv/internal/domain" "telesrv/internal/store/memory" ) func starGiftTestRouter(t *testing.T) (*Router, domain.User, domain.User, domain.StarGift) { return starGiftTestRouterWithPremium(t, false) } func starGiftTestRouterWithPremium(t *testing.T, requirePremium bool) (*Router, domain.User, domain.User, domain.StarGift) { t.Helper() ctx := context.Background() users := memory.NewUserStore() dialogs := memory.NewDialogStore() msgStore := memory.NewMessageStore(dialogs) channelStore := memory.NewChannelStore() sender, err := users.Create(ctx, domain.User{AccessHash: 7101, Phone: "15550007101", FirstName: "Sender"}) if err != nil { t.Fatalf("create sender: %v", err) } recipient, err := users.Create(ctx, domain.User{AccessHash: 7102, Phone: "15550007102", FirstName: "Recipient"}) if err != nil { t.Fatalf("create recipient: %v", err) } gift := domain.StarGift{ ID: 8001, RevisionID: 9001, Stars: 50, ConvertStars: 50, Title: "Cake", RequirePremium: requirePremium, Sticker: domain.Document{ID: 700, AccessHash: 7, DCID: 2, MimeType: "application/x-tgsticker", Attributes: []domain.DocumentAttribute{{Kind: domain.DocAttrSticker}}}, } giftStore := memory.NewStarGiftStore() giftStore.SeedCatalog([]domain.StarGift{gift}) gifts := appstargifts.NewService(giftStore, nil, 2) r := New(Config{DC: 2, IP: "127.0.0.1", Port: 2398}, Deps{ Users: appusers.NewService(users), Messages: appmessages.NewService(msgStore, dialogs), Channels: appchannels.NewService(channelStore), Stars: appstars.NewService(memory.NewStarsStore(), appstars.WithStartingGrant(1000)), Gifts: gifts, }, zaptest.NewLogger(t), clock.System) return r, sender, recipient, gift } func TestStarGiftPurchaseRequiresActivePremium(t *testing.T) { r, sender, recipient, gift := starGiftTestRouterWithPremium(t, true) ctx := WithUserID(context.Background(), sender.ID) inv := &tg.InputInvoiceStarGift{Peer: &tg.InputPeerUser{UserID: recipient.ID, AccessHash: recipient.AccessHash}, GiftID: gift.ID} if _, err := r.onPaymentsGetPaymentForm(ctx, &tg.PaymentsGetPaymentFormRequest{Invoice: inv}); !tgerr.Is(err, "PREMIUM_ACCOUNT_REQUIRED") { t.Fatalf("non-premium gift form err = %v, want PREMIUM_ACCOUNT_REQUIRED", err) } premium, ok := r.deps.Users.(UserPremiumService) if !ok { t.Fatalf("users service %T does not implement premium grants", r.deps.Users) } if _, err := premium.GrantPremium(context.Background(), sender.ID, 1); err != nil { t.Fatalf("grant premium: %v", err) } formRes, err := r.onPaymentsGetPaymentForm(ctx, &tg.PaymentsGetPaymentFormRequest{Invoice: inv}) if err != nil { t.Fatalf("premium gift form: %v", err) } form, ok := formRes.(*tg.PaymentsPaymentFormStarGift) if !ok { t.Fatalf("premium gift form = %T", formRes) } if _, err := r.onPaymentsSendStarsForm(ctx, &tg.PaymentsSendStarsFormRequest{FormID: form.FormID, Invoice: inv}); err != nil { t.Fatalf("premium gift purchase: %v", err) } } type uniqueGiftRPCService struct { GiftsService unique domain.UniqueStarGift } func (s *uniqueGiftRPCService) UniqueBySlug(_ context.Context, slug string) (domain.UniqueStarGift, bool, error) { return s.unique, slug == s.unique.Slug, nil } type craftStarGiftRPCService struct { GiftsService uniques map[string]domain.UniqueStarGift saved map[int64]domain.SavedStarGift result domain.StarGiftCraftResult craftReq domain.StarGiftCraftRequest craftCall int } func (s *craftStarGiftRPCService) UniqueBySlug(_ context.Context, slug string) (domain.UniqueStarGift, bool, error) { unique, ok := s.uniques[slug] return unique, ok, nil } func (s *craftStarGiftRPCService) GetSaved(_ context.Context, ref domain.SavedStarGiftRef) (domain.SavedStarGift, bool, error) { for _, saved := range s.saved { if saved.Owner != ref.Owner { continue } if ref.Slug != "" { unique, ok := s.uniques[ref.Slug] if ok && unique.ID == saved.UniqueGiftID { return saved, true, nil } continue } if saved.MsgID == ref.MsgID || saved.UpgradeMsgID == ref.MsgID { return saved, true, nil } } return domain.SavedStarGift{}, false, nil } func (s *craftStarGiftRPCService) Craft(_ context.Context, req domain.StarGiftCraftRequest) (domain.StarGiftCraftResult, error) { s.craftCall++ s.craftReq = req return s.result, nil } func TestCraftStarGiftAcceptsOfficialSlugAndCanonicalizesAliases(t *testing.T) { owner := domain.Peer{Type: domain.PeerTypeUser, ID: 7102} service := &craftStarGiftRPCService{ uniques: map[string]domain.UniqueStarGift{ "official-8001-2": {ID: 902, Slug: "official-8001-2", Owner: owner, SourceSavedGiftID: 52}, }, saved: map[int64]domain.SavedStarGift{ 50: {ID: 50, Owner: owner, MsgID: 115, UniqueGiftID: 901, UpgradeMsgID: 116}, 52: {ID: 52, Owner: owner, MsgID: 111, UniqueGiftID: 902, UpgradeMsgID: 112}, }, result: domain.StarGiftCraftResult{Chance: 500, SourceEdits: []domain.EditedMessageForUser{{ UserID: owner.ID, Message: domain.Message{ID: 116, OwnerUserID: owner.ID, Peer: owner, From: owner, Date: 100}, Event: domain.UpdateEvent{UserID: owner.ID, Type: domain.UpdateEventEditMessage, Pts: 41, PtsCount: 1, Date: 100, Message: domain.Message{ID: 116, OwnerUserID: owner.ID, Peer: owner, From: owner, Date: 100}}, }}}, } r := New(Config{DC: 2}, Deps{Gifts: service}, zaptest.NewLogger(t), clock.System) ctx := WithUserID(context.Background(), owner.ID) updates, err := r.onPaymentsCraftStarGift(ctx, &tg.PaymentsCraftStarGiftRequest{Stargift: []tg.InputSavedStarGiftClass{ &tg.InputSavedStarGiftUser{MsgID: 115}, &tg.InputSavedStarGiftSlug{Slug: "OFFICIAL-8001-2"}, }}) if err != nil || updates == nil { t.Fatalf("craft mixed official refs: updates=%T err=%v", updates, err) } if service.craftCall != 1 || service.craftReq.CommandKey != "rpc:50,52" || len(service.craftReq.Refs) != 2 || service.craftReq.Refs[1].Slug != "official-8001-2" { t.Fatalf("craft request = %+v calls=%d", service.craftReq, service.craftCall) } full, ok := updates.(*tg.Updates) if !ok || len(full.Updates) != 2 { t.Fatalf("craft failure updates = %T %#v", updates, updates) } if edit, ok := full.Updates[0].(*tg.UpdateEditMessage); !ok || edit.Pts != 41 || edit.PtsCount != 1 { t.Fatalf("craft failure source update = %T %#v", full.Updates[0], full.Updates[0]) } if _, ok := full.Updates[1].(*tg.UpdateStarGiftCraftFail); !ok { t.Fatalf("craft terminal update = %T %#v", full.Updates[1], full.Updates[1]) } service.craftCall = 0 _, err = r.onPaymentsCraftStarGift(ctx, &tg.PaymentsCraftStarGiftRequest{Stargift: []tg.InputSavedStarGiftClass{ &tg.InputSavedStarGiftUser{MsgID: 111}, &tg.InputSavedStarGiftSlug{Slug: "official-8001-2"}, }}) if !tgerr.Is(err, "STARGIFT_INVALID") || service.craftCall != 0 { t.Fatalf("duplicate aliases err=%v craft calls=%d", err, service.craftCall) } updates, err = r.onPaymentsCraftStarGift(ctx, &tg.PaymentsCraftStarGiftRequest{Stargift: []tg.InputSavedStarGiftClass{ &tg.InputSavedStarGiftUser{MsgID: 116}, }}) if err != nil || updates == nil || service.craftCall != 1 || service.craftReq.CommandKey != "rpc:50" || len(service.craftReq.Refs) != 1 || service.craftReq.Refs[0].MsgID != 116 { t.Fatalf("upgrade message alias craft: updates=%T req=%+v err=%v calls=%d", updates, service.craftReq, err, service.craftCall) } } type upgradeReplayRPCService struct { GiftsService saved domain.SavedStarGift receipt domain.StarGiftUpgradeReceipt result domain.StarGiftUpgradeResult upgradeCalls int previewCalls int lastRequest domain.StarGiftUpgradeRequest } func (s *upgradeReplayRPCService) GetSaved(_ context.Context, _ domain.SavedStarGiftRef) (domain.SavedStarGift, bool, error) { return s.saved, true, nil } func (s *upgradeReplayRPCService) UpgradeReceipt(_ context.Context, userID int64, _ string) (domain.StarGiftUpgradeReceipt, bool, error) { if userID != s.receipt.UserID { return domain.StarGiftUpgradeReceipt{}, false, nil } return s.receipt, true, nil } func (s *upgradeReplayRPCService) CollectiblePreview(context.Context, int64) (domain.StarGiftUpgradePreview, bool, error) { s.previewCalls++ return domain.StarGiftUpgradePreview{}, false, nil } func (s *upgradeReplayRPCService) Upgrade(_ context.Context, req domain.StarGiftUpgradeRequest) (domain.StarGiftUpgradeResult, error) { s.upgradeCalls++ s.lastRequest = req return s.result, nil } func collectibleRPCAttribute(kind domain.StarGiftCollectibleAttributeKind, id int64, name string) domain.StarGiftCollectibleAttribute { attribute := domain.StarGiftCollectibleAttribute{ Kind: kind, Name: name, RarityKind: domain.StarGiftRarityPermille, RarityPermille: 1000, } if kind == domain.StarGiftCollectibleBackdrop { attribute.BackdropID = int(id) attribute.CenterColor = 0x112233 attribute.EdgeColor = 0x223344 attribute.PatternColor = 0x334455 attribute.TextColor = 0xffffff return attribute } attribute.Document = &domain.Document{ ID: id, AccessHash: id + 1, FileReference: []byte("collectible-rpc"), Date: 1700000000, MimeType: "application/x-tgsticker", Size: 3, DCID: 2, Attributes: []domain.DocumentAttribute{{Kind: domain.DocAttrSticker}, {Kind: domain.DocAttrFilename, FileName: "gift.tgs"}}, } if kind == domain.StarGiftCollectiblePattern { attribute.Document.Attributes[0] = domain.DocumentAttribute{Kind: domain.DocAttrCustomEmoji, TextColor: true} attribute.Document.Thumbs = []domain.PhotoSize{{Kind: domain.PhotoSizeKindPath, Type: "j", Bytes: []byte{1}}} } attribute.Animation = &domain.StarGiftAnimation{ SourceName: "gift.tgs", SourceFormat: domain.StarGiftAnimationTGS, JSON: []byte(`{"v":"5.7"}`), TGS: []byte("tgs"), SHA256: make([]byte, 32), Width: 512, Height: 512, } attribute.Blob = &domain.FileBlob{LocationKey: "doc:test", Backend: domain.MediaBackendLocalFS, ObjectKey: "test", Size: 3} return attribute } func TestSavedStarGiftProjectionCombinesHistoricalCatalogWithCurrentCollectibleAvailability(t *testing.T) { historical := domain.StarGift{ ID: 8001, RevisionID: 9001, Stars: 50, ConvertStars: 25, Title: "Historical Cake", Sticker: domain.Document{ID: 700, AccessHash: 7, DCID: 2, MimeType: "application/x-tgsticker"}, } saved := domain.SavedStarGift{GiftID: historical.ID, RevisionID: historical.RevisionID, MsgID: 44, Date: 100, ConvertStars: historical.ConvertStars} availability := map[int64]domain.StarGiftCollectibleAvailability{ historical.ID: {UpgradeStars: 75, SupplyTotal: 500, Issued: 12}, } projected := tgSavedStarGifts([]domain.SavedStarGift{saved}, map[int64]domain.StarGift{historical.RevisionID: historical}, availability) if len(projected) != 1 || !projected[0].CanUpgrade { t.Fatalf("saved gift = %#v, want current pool to make historical gift upgradable", projected) } gift, ok := projected[0].Gift.(*tg.StarGift) if !ok { t.Fatalf("saved gift inner = %T, want *tg.StarGift", projected[0].Gift) } if gift.Title != historical.Title || gift.Stars != historical.Stars || gift.ConvertStars != historical.ConvertStars { t.Fatalf("historical snapshot changed: %#v", gift) } if upgradeStars, ok := gift.GetUpgradeStars(); !ok || upgradeStars != 75 { t.Fatalf("upgrade_stars = %d ok=%v, want current price 75", upgradeStars, ok) } for _, profile := range []tlprofile.Profile{tlprofile.Profile227, tlprofile.Profile228} { wire := &tg.PaymentsSavedStarGifts{Count: 1, Gifts: projected, Chats: []tg.ChatClass{}, Users: []tg.UserClass{}} encoded := &bin.Buffer{} if err := tlprofile.EncodeObject(profile, wire, encoded); err != nil { t.Fatalf("encode Layer %d saved gift: %v", profile, err) } decodedObject, err := tlprofile.DecodeObject(profile, &bin.Buffer{Buf: encoded.Buf}, tlprofile.Limits{}) if err != nil { t.Fatalf("decode Layer %d saved gift: %v", profile, err) } decoded, ok := decodedObject.(*tg.PaymentsSavedStarGifts) if !ok { t.Fatalf("decode Layer %d saved gift type = %T", profile, decodedObject) } inner, ok := decoded.Gifts[0].Gift.(*tg.StarGift) if !ok || !decoded.Gifts[0].CanUpgrade || inner.UpgradeStars != 75 { t.Fatalf("Layer %d projection lost upgrade flags: %#v", profile, decoded.Gifts[0]) } } availability[historical.ID] = domain.StarGiftCollectibleAvailability{UpgradeStars: 75, SupplyTotal: 500, Issued: 500} soldOut := tgSavedStarGifts([]domain.SavedStarGift{saved}, map[int64]domain.StarGift{historical.RevisionID: historical}, availability)[0] if soldOut.CanUpgrade { t.Fatal("sold-out collectible pool must not advertise upgrade") } if gift, ok := soldOut.Gift.(*tg.StarGift); !ok { t.Fatalf("sold-out inner = %T, want *tg.StarGift", soldOut.Gift) } else if _, ok := gift.GetUpgradeStars(); ok { t.Fatal("sold-out catalog projection must not expose upgrade_stars") } saved.PrepaidUpgradeStars = 75 soldOutPrepaid := tgSavedStarGifts([]domain.SavedStarGift{saved}, map[int64]domain.StarGift{historical.RevisionID: historical}, availability)[0] if soldOutPrepaid.CanUpgrade { t.Fatal("sold-out prepaid gift must not advertise an upgrade the aggregate will reject") } if _, ok := soldOutPrepaid.GetUpgradeStars(); ok { t.Fatal("sold-out prepaid gift must not expose stale prepaid upgrade_stars") } } func TestSavedStarGiftProjectionPreservesCollectibleLifecycle(t *testing.T) { const ( giftID = int64(8001) revision = int64(9001) readyAt = 1_780_000_123 exportAt = 1_780_000_200 transferAt = 1_780_000_300 resellAt = 1_780_000_400 ) unique := domain.UniqueStarGift{ID: 9901, GiftID: giftID, Title: "Craftable", Slug: "craftable-1", Num: 1, Owner: domain.Peer{Type: domain.PeerTypeUser, ID: 7102}, CraftChancePermille: 250} saved := domain.SavedStarGift{ Owner: domain.Peer{Type: domain.PeerTypeUser, ID: 7102}, GiftID: giftID, RevisionID: revision, MsgID: 44, Date: 100, UniqueGiftID: unique.ID, Unique: &unique, CanExportAt: exportAt, TransferStars: 25, CanTransferAt: transferAt, CanResellAt: resellAt, DropOriginalDetailsStars: 30, CanCraftAt: readyAt, } projected := tgSavedStarGifts([]domain.SavedStarGift{saved}, nil, nil) if len(projected) != 1 { t.Fatalf("saved lifecycle projection count = %d", len(projected)) } assertLifecycle := func(t *testing.T, item tg.SavedStarGift) { t.Helper() if value, ok := item.GetCanExportAt(); !ok || value != exportAt { t.Fatalf("can_export_at = %d set=%v", value, ok) } if value, ok := item.GetTransferStars(); !ok || value != 25 { t.Fatalf("transfer_stars = %d set=%v", value, ok) } if value, ok := item.GetCanTransferAt(); !ok || value != transferAt { t.Fatalf("can_transfer_at = %d set=%v", value, ok) } if value, ok := item.GetCanResellAt(); !ok || value != resellAt { t.Fatalf("can_resell_at = %d set=%v", value, ok) } if value, ok := item.GetDropOriginalDetailsStars(); !ok || value != 30 { t.Fatalf("drop_original_details_stars = %d set=%v", value, ok) } if value, ok := item.GetCanCraftAt(); !ok || value != readyAt { t.Fatalf("can_craft_at = %d set=%v", value, ok) } } assertLifecycle(t, projected[0]) zero := tgSavedStarGifts([]domain.SavedStarGift{{ Owner: domain.Peer{Type: domain.PeerTypeUser, ID: 7102}, GiftID: giftID, RevisionID: revision, MsgID: 45, Date: 101, UniqueGiftID: unique.ID, Unique: &unique, }}, nil, nil)[0] if _, ok := zero.GetCanExportAt(); ok { t.Fatal("zero can_export_at must be absent") } if _, ok := zero.GetTransferStars(); ok { t.Fatal("zero transfer_stars must be absent") } if _, ok := zero.GetCanTransferAt(); ok { t.Fatal("zero can_transfer_at must be absent") } if _, ok := zero.GetCanResellAt(); ok { t.Fatal("zero can_resell_at must be absent") } if _, ok := zero.GetDropOriginalDetailsStars(); ok { t.Fatal("zero drop_original_details_stars must be absent") } if _, ok := zero.GetCanCraftAt(); ok { t.Fatal("zero can_craft_at must be absent") } channelSaved := saved channelSaved.Owner = domain.Peer{Type: domain.PeerTypeChannel, ID: 8102} channelSaved.MsgID = 0 channelSaved.SavedID = 51 channelProjected := tgSavedStarGifts([]domain.SavedStarGift{channelSaved}, nil, nil)[0] if _, ok := channelProjected.GetCanCraftAt(); ok { t.Fatal("channel can_craft_at must be absent until channel Craft is executable") } if value, ok := channelProjected.GetSavedID(); !ok || value != channelSaved.SavedID { t.Fatalf("channel saved_id = %d set=%v", value, ok) } for _, profile := range []tlprofile.Profile{tlprofile.Profile225, tlprofile.Profile226, tlprofile.Profile227, tlprofile.Profile228} { wire := &tg.PaymentsSavedStarGifts{Count: 1, Gifts: projected, Chats: []tg.ChatClass{}, Users: []tg.UserClass{}} encoded := &bin.Buffer{} if err := tlprofile.EncodeObject(profile, wire, encoded); err != nil { t.Fatalf("encode Layer %d saved lifecycle: %v", profile, err) } decodedObject, err := tlprofile.DecodeObject(profile, &bin.Buffer{Buf: encoded.Buf}, tlprofile.Limits{}) if err != nil { t.Fatalf("decode Layer %d saved lifecycle: %v", profile, err) } decoded, ok := decodedObject.(*tg.PaymentsSavedStarGifts) if !ok || len(decoded.Gifts) != 1 { t.Fatalf("decode Layer %d saved lifecycle type = %T", profile, decodedObject) } assertLifecycle(t, decoded.Gifts[0]) channelWire := &tg.PaymentsSavedStarGifts{Count: 1, Gifts: []tg.SavedStarGift{channelProjected}, Chats: []tg.ChatClass{}, Users: []tg.UserClass{}} channelEncoded := &bin.Buffer{} if err := tlprofile.EncodeObject(profile, channelWire, channelEncoded); err != nil { t.Fatalf("encode Layer %d channel saved lifecycle: %v", profile, err) } channelDecodedObject, err := tlprofile.DecodeObject(profile, &bin.Buffer{Buf: channelEncoded.Buf}, tlprofile.Limits{}) if err != nil { t.Fatalf("decode Layer %d channel saved lifecycle: %v", profile, err) } channelDecoded, ok := channelDecodedObject.(*tg.PaymentsSavedStarGifts) if !ok || len(channelDecoded.Gifts) != 1 { t.Fatalf("decode Layer %d channel saved lifecycle type = %T", profile, channelDecodedObject) } if _, ok := channelDecoded.Gifts[0].GetCanCraftAt(); ok { t.Fatalf("Layer %d channel saved gift exposed can_craft_at", profile) } } } func TestChannelUniqueActionSuppressesCraftReadinessAcrossProfiles(t *testing.T) { const readyAt = 1_780_000_123 unique := domain.UniqueStarGift{ ID: 9902, GiftID: 8002, Title: "Channel Craftable", Slug: "channel-craftable-1", Num: 1, Owner: domain.Peer{Type: domain.PeerTypeChannel, ID: 8102}, CraftChancePermille: 250, } action := tgMessageActionStarGiftUnique(&domain.MessageStarGiftUniqueAction{ Gift: unique, Peer: unique.Owner, SavedID: 52, Saved: true, CanCraftAt: readyAt, }).(*tg.MessageActionStarGiftUnique) if _, ok := action.GetCanCraftAt(); ok { t.Fatal("channel unique action must not expose can_craft_at") } projectedGift, ok := action.Gift.(*tg.StarGiftUnique) if !ok || projectedGift.CraftChancePermille != unique.CraftChancePermille { t.Fatalf("channel unique gift lost intrinsic Craft chance: %#v", action.Gift) } for _, profile := range []tlprofile.Profile{tlprofile.Profile225, tlprofile.Profile226, tlprofile.Profile227, tlprofile.Profile228} { wire := &bin.Buffer{} if err := tlprofile.EncodeObject(profile, action, wire); err != nil { t.Fatalf("encode Layer %d channel unique action: %v", profile, err) } decodedObject, err := tlprofile.DecodeObject(profile, &bin.Buffer{Buf: wire.Buf}, tlprofile.Limits{}) if err != nil { t.Fatalf("decode Layer %d channel unique action: %v", profile, err) } decoded, ok := decodedObject.(*tg.MessageActionStarGiftUnique) if !ok { t.Fatalf("decode Layer %d channel unique action type = %T", profile, decodedObject) } if _, ok := decoded.GetCanCraftAt(); ok { t.Fatalf("Layer %d channel unique action exposed can_craft_at", profile) } gift, ok := decoded.Gift.(*tg.StarGiftUnique) if !ok || gift.CraftChancePermille != unique.CraftChancePermille { t.Fatalf("Layer %d channel unique gift = %#v", profile, decoded.Gift) } } } func TestStarGiftLifecycleCraftUnavailableError(t *testing.T) { if err := starGiftLifecycleErr(domain.ErrStarGiftCraftUnavailable); !tgerr.Is(err, "STARGIFT_CRAFT_UNAVAILABLE") { t.Fatalf("craft unavailable mapping = %v", err) } } func TestMessageStarGiftProjectionSeparatesPaidPriceFromPrepaidAmount(t *testing.T) { ordinary, ok := tgMessageActionStarGift(&domain.MessageStarGiftAction{ GiftID: 8001, Stars: 50, ConvertStars: 25, CanUpgrade: true, UpgradePriceStars: 75, }).(*tg.MessageActionStarGift) if !ok { t.Fatalf("ordinary action = %T", ordinary) } ordinaryGift, ok := ordinary.Gift.(*tg.StarGift) if !ok { t.Fatalf("ordinary inner gift = %T", ordinary.Gift) } if price, set := ordinaryGift.GetUpgradeStars(); !set || price != 75 { t.Fatalf("ordinary inner upgrade_stars = %d set=%v, want paid price 75", price, set) } if amount, set := ordinary.GetUpgradeStars(); set || amount != 0 || ordinary.PrepaidUpgrade { t.Fatalf("ordinary outer upgrade_stars = %d set=%v prepaid=%v, want absent", amount, set, ordinary.PrepaidUpgrade) } prepaid, ok := tgMessageActionStarGift(&domain.MessageStarGiftAction{ GiftID: 8001, Stars: 50, ConvertStars: 25, CanUpgrade: true, PrepaidUpgrade: true, UpgradePriceStars: 75, UpgradeStars: 75, }).(*tg.MessageActionStarGift) if !ok { t.Fatalf("prepaid action = %T", prepaid) } if amount, set := prepaid.GetUpgradeStars(); !set || amount != 75 || !prepaid.PrepaidUpgrade { t.Fatalf("prepaid outer upgrade_stars = %d set=%v prepaid=%v, want 75", amount, set, prepaid.PrepaidUpgrade) } upgraded, ok := tgMessageActionStarGift(&domain.MessageStarGiftAction{ GiftID: 8001, Stars: 50, ConvertStars: 25, UpgradeMsgID: 88, }).(*tg.MessageActionStarGift) if !ok { t.Fatalf("upgraded action = %T", upgraded) } if msgID, set := upgraded.GetUpgradeMsgID(); !set || msgID != 88 { t.Fatalf("upgrade_msg_id = %d set=%v, want 88", msgID, set) } for _, profile := range []tlprofile.Profile{tlprofile.Profile227, tlprofile.Profile228} { wire := &bin.Buffer{} if err := tlprofile.EncodeObject(profile, ordinary, wire); err != nil { t.Fatalf("encode Layer %d ordinary action: %v", profile, err) } decodedObject, err := tlprofile.DecodeObject(profile, &bin.Buffer{Buf: wire.Buf}, tlprofile.Limits{}) if err != nil { t.Fatalf("decode Layer %d ordinary action: %v", profile, err) } decoded, ok := decodedObject.(*tg.MessageActionStarGift) if !ok { t.Fatalf("decode Layer %d action = %T", profile, decodedObject) } inner, ok := decoded.Gift.(*tg.StarGift) if !ok || inner.UpgradeStars != 75 || decoded.UpgradeStars != 0 || decoded.PrepaidUpgrade { t.Fatalf("Layer %d ordinary action lost paid/prepaid split: %#v", profile, decoded) } upgradedWire := &bin.Buffer{} if err := tlprofile.EncodeObject(profile, upgraded, upgradedWire); err != nil { t.Fatalf("encode Layer %d upgraded action: %v", profile, err) } decodedUpgradedObject, err := tlprofile.DecodeObject(profile, &bin.Buffer{Buf: upgradedWire.Buf}, tlprofile.Limits{}) if err != nil { t.Fatalf("decode Layer %d upgraded action: %v", profile, err) } decodedUpgraded, ok := decodedUpgradedObject.(*tg.MessageActionStarGift) if !ok || !decodedUpgraded.Upgraded || decodedUpgraded.UpgradeMsgID != 88 || decodedUpgraded.CanUpgrade { t.Fatalf("Layer %d upgraded action lost transition flags: %#v", profile, decodedUpgradedObject) } } } func TestStarGiftUpgradeRPCReplaysCommittedReceiptAfterTerminalTransition(t *testing.T) { r, sender, owner, gift := starGiftTestRouter(t) ownerPeer := domain.Peer{Type: domain.PeerTypeUser, ID: owner.ID} saved := domain.SavedStarGift{ ID: 47, Owner: ownerPeer, FromUserID: sender.ID, GiftID: gift.ID, RevisionID: gift.RevisionID, MsgID: 105, UniqueGiftID: 9200000000000004, } result := domain.StarGiftUpgradeResult{ Saved: saved, Unique: domain.UniqueStarGift{ID: saved.UniqueGiftID, GiftID: gift.ID, Owner: ownerPeer}, Balance: domain.StarsBalance{UserID: owner.ID, Balance: 1000}, Duplicate: true, Send: domain.SendPrivateTextResult{ RecipientMessage: domain.Message{ID: 107, OwnerUserID: owner.ID, Peer: domain.Peer{Type: domain.PeerTypeUser, ID: sender.ID}, From: domain.Peer{Type: domain.PeerTypeUser, ID: sender.ID}, Date: 1700000001}, RecipientEvent: domain.UpdateEvent{UserID: owner.ID, Pts: 42, PtsCount: 1, Date: 1700000001}, }, } service := &upgradeReplayRPCService{GiftsService: r.deps.Gifts, saved: saved, result: result, receipt: domain.StarGiftUpgradeReceipt{UserID: owner.ID, SourceSavedGiftID: saved.ID, UniqueGiftID: saved.UniqueGiftID, RequirePrepaid: true, KeepOriginalDetails: true, BalanceAfter: 1000}} r.deps.Gifts = service ctx := WithUserID(context.Background(), owner.ID) if _, err := r.onPaymentsUpgradeStarGift(ctx, &tg.PaymentsUpgradeStarGiftRequest{ KeepOriginalDetails: true, Stargift: &tg.InputSavedStarGiftUser{MsgID: saved.MsgID}, }); err != nil { t.Fatalf("replay prepaid upgrade after terminal transition: %v", err) } if service.upgradeCalls != 1 || service.previewCalls != 0 || !service.lastRequest.RequirePrepaid || service.lastRequest.ChargeStars != 0 { t.Fatalf("prepaid replay calls=%d preview=%d req=%+v", service.upgradeCalls, service.previewCalls, service.lastRequest) } const paidFormID int64 = -7611777087885039132 service.receipt = domain.StarGiftUpgradeReceipt{UserID: owner.ID, SourceSavedGiftID: saved.ID, FormID: paidFormID, UniqueGiftID: saved.UniqueGiftID, ChargeStars: 25, KeepOriginalDetails: true, BalanceAfter: 975} service.upgradeCalls, service.previewCalls = 0, 0 if _, err := r.sendStarGiftUpgradeForm(ctx, owner.ID, paidFormID, &tg.InputInvoiceStarGiftUpgrade{ KeepOriginalDetails: true, Stargift: &tg.InputSavedStarGiftUser{MsgID: saved.MsgID}, }); err != nil { t.Fatalf("replay paid upgrade after terminal transition: %v", err) } if service.upgradeCalls != 1 || service.previewCalls != 0 || service.lastRequest.ChargeStars != 25 || service.lastRequest.FormID != paidFormID { t.Fatalf("paid replay calls=%d preview=%d req=%+v", service.upgradeCalls, service.previewCalls, service.lastRequest) } } func TestStarGiftCollectiblePreviewUpgradeFormUniqueAndServiceProjection(t *testing.T) { r, sender, owner, gift := starGiftTestRouter(t) ctx := context.Background() ownerCtx := WithUserID(ctx, owner.ID) giftService, ok := r.deps.Gifts.(*appstargifts.Service) if !ok { t.Fatalf("gift service = %T", r.deps.Gifts) } model := collectibleRPCAttribute(domain.StarGiftCollectibleModel, 8101, "Aurora") modelTwo := collectibleRPCAttribute(domain.StarGiftCollectibleModel, 8104, "Aurora Two") crafted := collectibleRPCAttribute(domain.StarGiftCollectibleModel, 8103, "Crafted Aurora") crafted.Crafted = true crafted.RarityKind = domain.StarGiftRarityLegendary crafted.RarityPermille = 0 pattern := collectibleRPCAttribute(domain.StarGiftCollectiblePattern, 8102, "Orbit") patternTwo := collectibleRPCAttribute(domain.StarGiftCollectiblePattern, 8105, "Orbit Two") backdrop := collectibleRPCAttribute(domain.StarGiftCollectibleBackdrop, 1, "Midnight") backdropTwo := collectibleRPCAttribute(domain.StarGiftCollectibleBackdrop, 2, "Daylight") if _, err := giftService.PublishCollectibleRevision(ctx, domain.StarGiftCollectibleWrite{ GiftID: gift.ID, UpgradeStars: 75, SupplyTotal: 500, SlugPrefix: "cake", Models: []domain.StarGiftCollectibleAttribute{model, crafted, modelTwo}, Patterns: []domain.StarGiftCollectibleAttribute{pattern, patternTwo}, Backdrops: []domain.StarGiftCollectibleAttribute{backdrop, backdropTwo}, Actor: "test", CommandID: "collectible-rpc", }); err != nil { t.Fatalf("publish collectible pool: %v", err) } if _, err := r.deps.Gifts.RecordSavedGift(ctx, domain.SavedStarGift{ Owner: domain.Peer{Type: domain.PeerTypeUser, ID: owner.ID}, FromUserID: sender.ID, GiftID: gift.ID, RevisionID: gift.RevisionID, MsgID: 444, Date: 1700000000, ConvertStars: gift.ConvertStars, }); err != nil { t.Fatalf("record upgrade target: %v", err) } preview, err := r.onPaymentsGetStarGiftUpgradePreview(ownerCtx, gift.ID) if err != nil || len(preview.SampleAttributes) != 6 { t.Fatalf("upgrade preview = %#v err %v", preview, err) } attributes, err := r.onPaymentsGetStarGiftUpgradeAttributes(ownerCtx, gift.ID) if err != nil || len(attributes.Attributes) != 7 { t.Fatalf("upgrade attributes = %#v err %v", attributes, err) } craftedTG, ok := attributes.Attributes[1].(*tg.StarGiftAttributeModel) if !ok || !craftedTG.Crafted { t.Fatalf("crafted attribute = %T %#v", attributes.Attributes[1], attributes.Attributes[1]) } if _, ok := craftedTG.Rarity.(*tg.StarGiftAttributeRarityLegendary); !ok { t.Fatalf("crafted rarity = %T", craftedTG.Rarity) } invoice := &tg.InputInvoiceStarGiftUpgrade{Stargift: &tg.InputSavedStarGiftUser{MsgID: 444}} formClass, err := r.onPaymentsGetPaymentForm(ownerCtx, &tg.PaymentsGetPaymentFormRequest{Invoice: invoice}) if err != nil { t.Fatalf("get upgrade payment form: %v", err) } form, ok := formClass.(*tg.PaymentsPaymentFormStarGift) if !ok || form.FormID == 0 || form.Invoice.Currency != "XTR" || len(form.Invoice.Prices) != 1 || form.Invoice.Prices[0].Amount != 75 { t.Fatalf("upgrade payment form = %T %#v", formClass, formClass) } unique := domain.UniqueStarGift{ ID: 9200000000000001, GiftID: gift.ID, Title: gift.Title, Slug: "cake-1", Num: 1, Owner: domain.Peer{Type: domain.PeerTypeUser, ID: owner.ID}, Model: model, Pattern: pattern, Backdrop: backdrop, AvailabilityIssued: 1, AvailabilityTotal: 500, KeepOriginalDetails: true, OriginalFromUserID: sender.ID, OriginalOwner: domain.Peer{Type: domain.PeerTypeUser, ID: owner.ID}, OriginalDate: 1700000000, OriginalMessage: "hello", } r.deps.Gifts = &uniqueGiftRPCService{GiftsService: r.deps.Gifts, unique: unique} uniqueResponse, err := r.onPaymentsGetUniqueStarGift(WithUserID(ctx, sender.ID), unique.Slug) if err != nil { t.Fatalf("get unique gift = %#v err %v", uniqueResponse, err) } uniqueGift, ok := uniqueResponse.Gift.(*tg.StarGiftUnique) if !ok || uniqueGift.Slug != unique.Slug || len(uniqueGift.Attributes) != 4 || len(uniqueResponse.Users) != 2 { t.Fatalf("get unique gift = %#v", uniqueResponse) } message := domain.Message{Media: &domain.MessageMedia{Kind: domain.MessageMediaKindService, ServiceAction: &domain.MessageServiceAction{ Kind: domain.MessageServiceActionStarGiftUnique, StarGiftUnique: &domain.MessageStarGiftUniqueAction{ Gift: unique, FromUserID: sender.ID, Peer: unique.Owner, SavedID: 444, Upgrade: true, Saved: true, }, }}} action, ok := tgMessageServiceAction(message).(*tg.MessageActionStarGiftUnique) if !ok { t.Fatalf("unique service action type = %T", tgMessageServiceAction(message)) } projectedGift, giftOK := action.Gift.(*tg.StarGiftUnique) if !giftOK || !action.Upgrade || !action.Saved || projectedGift.ID != unique.ID || projectedGift.Slug != unique.Slug { t.Fatalf("unique service action = %#v", tgMessageServiceAction(message)) } if peer, ok := action.GetPeer(); !ok { t.Fatal("unique service action missing owner peer") } else if user, ok := peer.(*tg.PeerUser); !ok || user.UserID != owner.ID { t.Fatalf("unique service action peer = %#v", peer) } if savedID, ok := action.GetSavedID(); !ok || savedID != 444 { t.Fatalf("unique service action saved_id = %d set=%v, want 444", savedID, ok) } for _, profile := range []tlprofile.Profile{tlprofile.Profile227, tlprofile.Profile228} { responseWire := &bin.Buffer{} if err := tlprofile.EncodeObject(profile, uniqueResponse, responseWire); err != nil { t.Fatalf("encode Layer %d unique response: %v", profile, err) } decodedResponseObject, err := tlprofile.DecodeObject(profile, &bin.Buffer{Buf: responseWire.Buf}, tlprofile.Limits{}) if err != nil { t.Fatalf("decode Layer %d unique response: %v", profile, err) } decodedResponse, ok := decodedResponseObject.(*tg.PaymentsUniqueStarGift) if !ok { t.Fatalf("decode Layer %d unique response type = %T", profile, decodedResponseObject) } decodedGift, ok := decodedResponse.Gift.(*tg.StarGiftUnique) if !ok || decodedGift.Slug != unique.Slug || len(decodedGift.Attributes) != 4 { t.Fatalf("Layer %d unique response lost fields: %#v", profile, decodedResponse.Gift) } actionWire := &bin.Buffer{} if err := tlprofile.EncodeObject(profile, action, actionWire); err != nil { t.Fatalf("encode Layer %d unique action: %v", profile, err) } decodedActionObject, err := tlprofile.DecodeObject(profile, &bin.Buffer{Buf: actionWire.Buf}, tlprofile.Limits{}) if err != nil { t.Fatalf("decode Layer %d unique action: %v", profile, err) } decodedAction, ok := decodedActionObject.(*tg.MessageActionStarGiftUnique) if !ok { t.Fatalf("decode Layer %d unique action type = %T", profile, decodedActionObject) } if decodedActionGift, ok := decodedAction.Gift.(*tg.StarGiftUnique); !ok || !decodedAction.Upgrade || decodedAction.SavedID != 444 || decodedActionGift.Slug != unique.Slug { t.Fatalf("Layer %d unique action lost fields: %#v", profile, decodedAction) } } } func TestStarGiftCollectionsCRUDFilterOrderAndPin(t *testing.T) { r, _, owner, gift := starGiftTestRouter(t) ctx := WithUserID(context.Background(), owner.ID) ownerPeer := domain.Peer{Type: domain.PeerTypeUser, ID: owner.ID} for _, msgID := range []int{101, 102} { if _, err := r.deps.Gifts.RecordSavedGift(context.Background(), domain.SavedStarGift{ Owner: ownerPeer, GiftID: gift.ID, RevisionID: gift.RevisionID, MsgID: msgID, Date: 1700000000 + msgID, ConvertStars: gift.ConvertStars, }); err != nil { t.Fatalf("record saved gift %d: %v", msgID, err) } } ref101 := tg.InputSavedStarGiftClass(&tg.InputSavedStarGiftUser{MsgID: 101}) ref102 := tg.InputSavedStarGiftClass(&tg.InputSavedStarGiftUser{MsgID: 102}) first, err := r.onPaymentsCreateStarGiftCollection(ctx, &tg.PaymentsCreateStarGiftCollectionRequest{ Peer: &tg.InputPeerSelf{}, Title: " Favorites ", Stargift: []tg.InputSavedStarGiftClass{ref101}, }) if err != nil { t.Fatalf("create first collection: %v", err) } second, err := r.onPaymentsCreateStarGiftCollection(ctx, &tg.PaymentsCreateStarGiftCollectionRequest{ Peer: &tg.InputPeerSelf{}, Title: "Archive", Stargift: []tg.InputSavedStarGiftClass{ref102}, }) if err != nil { t.Fatalf("create second collection: %v", err) } if first.Title != "Favorites" || first.GiftsCount != 1 || second.GiftsCount != 1 { t.Fatalf("created collections = %#v / %#v", first, second) } listedClass, err := r.onPaymentsGetStarGiftCollections(ctx, &tg.PaymentsGetStarGiftCollectionsRequest{Peer: &tg.InputPeerSelf{}}) if err != nil { t.Fatalf("list collections: %v", err) } listed, ok := listedClass.(*tg.PaymentsStarGiftCollections) if !ok || len(listed.Collections) != 2 { t.Fatalf("list collections = %T %#v, want two", listedClass, listedClass) } domainCollections, err := r.deps.Gifts.ListCollections(context.Background(), ownerPeer) if err != nil { t.Fatalf("list domain collections: %v", err) } if notModified, err := r.onPaymentsGetStarGiftCollections(ctx, &tg.PaymentsGetStarGiftCollectionsRequest{ Peer: &tg.InputPeerSelf{}, Hash: domain.StarGiftCollectionsHash(domainCollections), }); err != nil { t.Fatalf("hash list collections: %v", err) } else if _, ok := notModified.(*tg.PaymentsStarGiftCollectionsNotModified); !ok { t.Fatalf("hash response = %T, want not modified", notModified) } update := &tg.PaymentsUpdateStarGiftCollectionRequest{Peer: &tg.InputPeerSelf{}, CollectionID: first.CollectionID} update.SetTitle("Best") update.SetAddStargift([]tg.InputSavedStarGiftClass{ref102}) update.SetOrder([]tg.InputSavedStarGiftClass{ref102, ref101}) updated, err := r.onPaymentsUpdateStarGiftCollection(ctx, update) if err != nil { t.Fatalf("update collection: %v", err) } if updated.Title != "Best" || updated.GiftsCount != 2 { t.Fatalf("updated collection = %#v", updated) } filteredReq := &tg.PaymentsGetSavedStarGiftsRequest{Peer: &tg.InputPeerSelf{}, Limit: 10} filteredReq.SetCollectionID(first.CollectionID) filtered, err := r.onPaymentsGetSavedStarGifts(ctx, filteredReq) if err != nil { t.Fatalf("filter saved gifts by collection: %v", err) } if filtered.Count != 2 || len(filtered.Gifts) != 2 { t.Fatalf("filtered gifts count=%d len=%d, want 2/2", filtered.Count, len(filtered.Gifts)) } for _, saved := range filtered.Gifts { ids, ok := saved.GetCollectionID() if !ok || len(ids) == 0 || ids[0] != first.CollectionID { t.Fatalf("saved gift collection projection = %#v", saved) } } if ok, err := r.onPaymentsToggleStarGiftsPinnedToTop(ctx, &tg.PaymentsToggleStarGiftsPinnedToTopRequest{ Peer: &tg.InputPeerSelf{}, Stargift: []tg.InputSavedStarGiftClass{ref101, ref102}, }); err != nil || !ok { t.Fatalf("pin gifts = %v err %v", ok, err) } pinned, err := r.onPaymentsGetSavedStarGifts(ctx, &tg.PaymentsGetSavedStarGiftsRequest{Peer: &tg.InputPeerSelf{}, Limit: 10}) if err != nil { t.Fatalf("list pinned gifts: %v", err) } if len(pinned.Gifts) != 2 || !pinned.Gifts[0].PinnedToTop || !pinned.Gifts[1].PinnedToTop { t.Fatalf("pinned projection = %#v", pinned.Gifts) } if ok, err := r.onPaymentsReorderStarGiftCollections(ctx, &tg.PaymentsReorderStarGiftCollectionsRequest{ Peer: &tg.InputPeerSelf{}, Order: []int{second.CollectionID, first.CollectionID}, }); err != nil || !ok { t.Fatalf("reorder collections = %v err %v", ok, err) } reorderedClass, err := r.onPaymentsGetStarGiftCollections(ctx, &tg.PaymentsGetStarGiftCollectionsRequest{Peer: &tg.InputPeerSelf{}}) if err != nil { t.Fatalf("list reordered collections: %v", err) } reordered := reorderedClass.(*tg.PaymentsStarGiftCollections) if reordered.Collections[0].CollectionID != second.CollectionID { t.Fatalf("reordered collections = %#v", reordered.Collections) } if ok, err := r.onPaymentsDeleteStarGiftCollection(ctx, &tg.PaymentsDeleteStarGiftCollectionRequest{ Peer: &tg.InputPeerSelf{}, CollectionID: first.CollectionID, }); err != nil || !ok { t.Fatalf("delete collection = %v err %v", ok, err) } afterDelete, err := r.onPaymentsGetSavedStarGifts(ctx, &tg.PaymentsGetSavedStarGiftsRequest{Peer: &tg.InputPeerSelf{}, Limit: 10}) if err != nil { t.Fatalf("list gifts after collection delete: %v", err) } for _, saved := range afterDelete.Gifts { if ids, ok := saved.GetCollectionID(); ok { for _, id := range ids { if id == first.CollectionID { t.Fatalf("deleted collection %d leaked in saved gift %#v", id, saved) } } } } } // 完整 star gift saga:catalog → getPaymentForm(paymentFormStarGift) → sendStarsForm(扣费+服务消息 // +paymentResult) → 收礼人 getSavedStarGifts → save/convert。 func TestStarGiftSaga(t *testing.T) { r, sender, recipient, gift := starGiftTestRouter(t) ctx := context.Background() senderCtx := WithUserID(ctx, sender.ID) recipientCtx := WithUserID(ctx, recipient.ID) // 1. 目录。 catRes, err := r.onPaymentsGetStarGifts(senderCtx, 0) if err != nil { t.Fatalf("getStarGifts: %v", err) } cat, ok := catRes.(*tg.PaymentsStarGifts) if !ok || len(cat.Gifts) != 1 { t.Fatalf("catalog = %T %+v, want 1 gift", catRes, catRes) } if g, ok := cat.Gifts[0].(*tg.StarGift); !ok || g.ID != gift.ID || g.Stars != 50 { t.Fatalf("catalog gift = %#v, want id %d stars 50", cat.Gifts[0], gift.ID) } // 即便 hash 命中也始终回完整目录(DrKLO force-stop 后保留 hash 但丢礼物缓存, // 返 NotModified 会让送礼选择器永远空)。 if again, err := r.onPaymentsGetStarGifts(senderCtx, cat.Hash); err != nil { t.Fatalf("getStarGifts hash: %v", err) } else if full, ok := again.(*tg.PaymentsStarGifts); !ok || len(full.Gifts) != 1 { t.Fatalf("hash match = %T, want full catalog (不返 NotModified)", again) } inv := &tg.InputInvoiceStarGift{ Peer: &tg.InputPeerUser{UserID: recipient.ID, AccessHash: recipient.AccessHash}, GiftID: gift.ID, } // 2. getPaymentForm → paymentFormStarGift(XTR + 非空 prices)。 formRes, err := r.onPaymentsGetPaymentForm(senderCtx, &tg.PaymentsGetPaymentFormRequest{Invoice: inv}) if err != nil { t.Fatalf("getPaymentForm: %v", err) } form, ok := formRes.(*tg.PaymentsPaymentFormStarGift) if !ok { t.Fatalf("form = %T, want *tg.PaymentsPaymentFormStarGift (TDesktop 单分支 match)", formRes) } if form.Invoice.Currency != "XTR" || len(form.Invoice.Prices) != 1 || form.Invoice.Prices[0].Amount != 50 { t.Fatalf("form invoice = %+v, want XTR + 1 price 50", form.Invoice) } // 3. sendStarsForm → paymentResult(扣费 + 服务消息 + updateStarsBalance)。 payRes, err := r.onPaymentsSendStarsForm(senderCtx, &tg.PaymentsSendStarsFormRequest{FormID: form.FormID, Invoice: inv}) if err != nil { t.Fatalf("sendStarsForm: %v", err) } pay, ok := payRes.(*tg.PaymentsPaymentResult) if !ok { t.Fatalf("pay result = %T, want *tg.PaymentsPaymentResult (DrKLO 强转)", payRes) } updates, ok := pay.Updates.(*tg.Updates) if !ok { t.Fatalf("pay updates = %T, want *tg.Updates", pay.Updates) } hasBalance, hasGiftMsg := false, false for _, up := range updates.Updates { switch u := up.(type) { case *tg.UpdateStarsBalance: hasBalance = true if amt, ok := u.Balance.(*tg.StarsAmount); !ok || amt.Amount != 950 { t.Fatalf("updateStarsBalance = %#v, want 950", u.Balance) } case *tg.UpdateNewMessage: if svc, ok := u.Message.(*tg.MessageService); ok { if _, ok := svc.Action.(*tg.MessageActionStarGift); ok { hasGiftMsg = true } } } } if !hasBalance || !hasGiftMsg { t.Fatalf("pay updates: balance=%v giftMsg=%v, want both (崩溃约束:必返合法 Updates)", hasBalance, hasGiftMsg) } // 送礼人余额扣 50。 if bal, _ := r.deps.Stars.GetBalance(ctx, sender.ID); bal.Balance != 950 { t.Fatalf("sender balance = %d, want 950", bal.Balance) } // 4. 收礼人 getSavedStarGifts。 savedRes, err := r.onPaymentsGetSavedStarGifts(recipientCtx, &tg.PaymentsGetSavedStarGiftsRequest{Peer: &tg.InputPeerSelf{}}) if err != nil { t.Fatalf("getSavedStarGifts: %v", err) } if savedRes.Count != 1 || len(savedRes.Gifts) != 1 { t.Fatalf("saved gifts = count %d len %d, want 1/1", savedRes.Count, len(savedRes.Gifts)) } saved := savedRes.Gifts[0] msgID, ok := saved.GetMsgID() if !ok || msgID <= 0 { t.Fatalf("saved gift msg_id = %d ok %v, want positive", msgID, ok) } if g, ok := saved.Gift.(*tg.StarGift); !ok || g.ID != gift.ID { t.Fatalf("saved gift inner = %#v, want gift %d", saved.Gift, gift.ID) } if from, ok := saved.GetFromID(); !ok { t.Fatalf("saved gift from = %v, want sender peer", from) } // 4b. 收礼人 userFull 必须带 stargifts_count(否则客户端资料页 Gifts 区段不出现)。 fullRes, err := r.onUsersGetFullUser(senderCtx, &tg.InputUser{UserID: recipient.ID, AccessHash: recipient.AccessHash}) if err != nil { t.Fatalf("getFullUser: %v", err) } if cnt, ok := fullRes.FullUser.GetStargiftsCount(); !ok || cnt != 1 { t.Fatalf("recipient userFull stargifts_count = %d ok %v, want 1 (资料页 Gifts 门控)", cnt, ok) } // 5. saveStarGift(隐藏)。 if ok, err := r.onPaymentsSaveStarGift(recipientCtx, &tg.PaymentsSaveStarGiftRequest{ Unsave: true, Stargift: &tg.InputSavedStarGiftUser{MsgID: msgID}, }); err != nil || !ok { t.Fatalf("saveStarGift hide = %v err %v", ok, err) } // 6. convertStarGift(转回 Stars,收礼人 +50)。 recipBefore, _ := r.deps.Stars.GetBalance(ctx, recipient.ID) if ok, err := r.onPaymentsConvertStarGift(recipientCtx, &tg.InputSavedStarGiftUser{MsgID: msgID}); err != nil || !ok { t.Fatalf("convertStarGift = %v err %v", ok, err) } recipAfter, _ := r.deps.Stars.GetBalance(ctx, recipient.ID) if recipAfter.Balance != recipBefore.Balance+50 { t.Fatalf("recipient balance %d -> %d, want +50", recipBefore.Balance, recipAfter.Balance) } // 转换后从列表消失。 afterRes, _ := r.onPaymentsGetSavedStarGifts(recipientCtx, &tg.PaymentsGetSavedStarGiftsRequest{Peer: &tg.InputPeerSelf{}}) if afterRes.Count != 0 { t.Fatalf("saved gifts after convert = %d, want 0", afterRes.Count) } // 重复转换被拒。 if _, err := r.onPaymentsConvertStarGift(recipientCtx, &tg.InputSavedStarGiftUser{MsgID: msgID}); err == nil { t.Fatalf("double convert should error") } } func TestStarGiftAutoSavePrivacyCreatesPendingGiftUntilRecipientApproves(t *testing.T) { r, sender, recipient, gift := starGiftTestRouter(t) ctx := context.Background() privacy := appprivacy.NewService(memory.NewPrivacyStore(), memory.NewContactStore()) if _, err := privacy.SetRules(ctx, recipient.ID, domain.PrivacyKeyStarGiftsAutoSave, []domain.PrivacyRule{ {Kind: domain.PrivacyRuleDisallowAll}, }); err != nil { t.Fatalf("set gift auto-save privacy: %v", err) } r.deps.Privacy = privacy senderCtx := WithUserID(ctx, sender.ID) recipientCtx := WithUserID(ctx, recipient.ID) invoice := &tg.InputInvoiceStarGift{ Peer: &tg.InputPeerUser{UserID: recipient.ID, AccessHash: recipient.AccessHash}, GiftID: gift.ID, } formClass, err := r.onPaymentsGetPaymentForm(senderCtx, &tg.PaymentsGetPaymentFormRequest{Invoice: invoice}) if err != nil { t.Fatalf("get gift form: %v", err) } form := formClass.(*tg.PaymentsPaymentFormStarGift) if _, err := r.onPaymentsSendStarsForm(senderCtx, &tg.PaymentsSendStarsFormRequest{ FormID: form.FormID, Invoice: invoice, }); err != nil { t.Fatalf("send gift: %v", err) } all, err := r.onPaymentsGetSavedStarGifts(recipientCtx, &tg.PaymentsGetSavedStarGiftsRequest{ Peer: &tg.InputPeerSelf{}, Limit: 10, }) if err != nil || all.Count != 1 || len(all.Gifts) != 1 { t.Fatalf("all saved gifts count=%d len=%d err=%v, want pending gift", all.Count, len(all.Gifts), err) } if !all.Gifts[0].Unsaved { t.Fatal("privacy-denied incoming gift must be pending (unsaved=true), not rejected") } msgID, ok := all.Gifts[0].GetMsgID() if !ok || msgID == 0 { t.Fatalf("pending gift msg_id=%d present=%v", msgID, ok) } visible, err := r.onPaymentsGetSavedStarGifts(recipientCtx, &tg.PaymentsGetSavedStarGiftsRequest{ Peer: &tg.InputPeerSelf{}, ExcludeUnsaved: true, Limit: 10, }) if err != nil || visible.Count != 0 || len(visible.Gifts) != 0 { t.Fatalf("exclude-unsaved count=%d len=%d err=%v, want hidden pending gift", visible.Count, len(visible.Gifts), err) } if ok, err := r.onPaymentsSaveStarGift(recipientCtx, &tg.PaymentsSaveStarGiftRequest{ Stargift: &tg.InputSavedStarGiftUser{MsgID: msgID}, }); err != nil || !ok { t.Fatalf("recipient approve gift = %v err=%v", ok, err) } visible, err = r.onPaymentsGetSavedStarGifts(recipientCtx, &tg.PaymentsGetSavedStarGiftsRequest{ Peer: &tg.InputPeerSelf{}, ExcludeUnsaved: true, Limit: 10, }) if err != nil || visible.Count != 1 || len(visible.Gifts) != 1 || visible.Gifts[0].Unsaved { t.Fatalf("approved visible gifts=%+v err=%v, want one saved gift", visible, err) } } // 频道 star gift saga:channel peer 能付款发送,但不生成频道历史消息; // saved gift 用 inputSavedStarGiftChat.saved_id 定位,Recent Actions 用 admin log 快照承载。 func TestStarGiftChannelSaga(t *testing.T) { r, sender, owner, gift := starGiftTestRouter(t) ctx := context.Background() senderCtx := WithUserID(ctx, sender.ID) ownerCtx := WithUserID(ctx, owner.ID) created, err := r.deps.Channels.CreateChannel(ctx, owner.ID, domain.CreateChannelRequest{ CreatorUserID: owner.ID, Title: "Gift Channel", Broadcast: true, MemberUserIDs: []int64{sender.ID}, Date: 1700001000, }) if err != nil { t.Fatalf("create channel: %v", err) } channel := created.Channel channelPeer := &tg.InputPeerChannel{ChannelID: channel.ID, AccessHash: channel.AccessHash} channelInput := &tg.InputChannel{ChannelID: channel.ID, AccessHash: channel.AccessHash} giftService := r.deps.Gifts.(*appstargifts.Service) if _, err := giftService.PublishCollectibleRevision(ctx, domain.StarGiftCollectibleWrite{ GiftID: gift.ID, UpgradeStars: 75, SupplyTotal: 10, SlugPrefix: "channel-cake", Models: []domain.StarGiftCollectibleAttribute{ collectibleRPCAttribute(domain.StarGiftCollectibleModel, 8201, "Aurora"), collectibleRPCAttribute(domain.StarGiftCollectibleModel, 8204, "Aurora Two"), }, Patterns: []domain.StarGiftCollectibleAttribute{ collectibleRPCAttribute(domain.StarGiftCollectiblePattern, 8202, "Orbit"), collectibleRPCAttribute(domain.StarGiftCollectiblePattern, 8205, "Orbit Two"), }, Backdrops: []domain.StarGiftCollectibleAttribute{ collectibleRPCAttribute(domain.StarGiftCollectibleBackdrop, 2, "Midnight"), collectibleRPCAttribute(domain.StarGiftCollectibleBackdrop, 3, "Daylight"), }, Actor: "test", CommandID: "channel-collectible-rpc", }); err != nil { t.Fatalf("publish channel collectible pool: %v", err) } upgradeFormRes, err := r.onPaymentsGetPaymentForm(senderCtx, &tg.PaymentsGetPaymentFormRequest{Invoice: &tg.InputInvoiceStarGift{ Peer: channelPeer, GiftID: gift.ID, IncludeUpgrade: true, }}) if err != nil { t.Fatalf("getPaymentForm(channel include_upgrade): %v", err) } upgradeForm, ok := upgradeFormRes.(*tg.PaymentsPaymentFormStarGift) if !ok || len(upgradeForm.Invoice.Prices) != 1 || upgradeForm.Invoice.Prices[0].Amount != gift.Stars+75 { t.Fatalf("channel include_upgrade form = %T %+v, want total %d", upgradeFormRes, upgradeFormRes, gift.Stars+75) } inv := &tg.InputInvoiceStarGift{ Peer: channelPeer, GiftID: gift.ID, } formRes, err := r.onPaymentsGetPaymentForm(senderCtx, &tg.PaymentsGetPaymentFormRequest{Invoice: inv}) if err != nil { t.Fatalf("getPaymentForm(channel): %v", err) } form, ok := formRes.(*tg.PaymentsPaymentFormStarGift) if !ok || form.Invoice.Currency != "XTR" || len(form.Invoice.Prices) != 1 { t.Fatalf("form(channel) = %T %+v, want star gift XTR form", formRes, formRes) } payRes, err := r.onPaymentsSendStarsForm(senderCtx, &tg.PaymentsSendStarsFormRequest{FormID: form.FormID, Invoice: inv}) if err != nil { t.Fatalf("sendStarsForm(channel): %v", err) } pay, ok := payRes.(*tg.PaymentsPaymentResult) if !ok { t.Fatalf("pay result = %T, want *tg.PaymentsPaymentResult", payRes) } updates, ok := pay.Updates.(*tg.Updates) if !ok { t.Fatalf("pay updates = %T, want *tg.Updates", pay.Updates) } var ( hasBalance bool ) for _, up := range updates.Updates { switch u := up.(type) { case *tg.UpdateStarsBalance: hasBalance = true if amt, ok := u.Balance.(*tg.StarsAmount); !ok || amt.Amount != 950 { t.Fatalf("channel updateStarsBalance = %#v, want 950", u.Balance) } case *tg.UpdateNewChannelMessage: t.Fatalf("channel gift must not be pushed as UpdateNewChannelMessage: %#v", u.Message) } } if !hasBalance { t.Fatalf("channel pay updates: balance=%v, want updateStarsBalance", hasBalance) } savedRes, err := r.onPaymentsGetSavedStarGifts(senderCtx, &tg.PaymentsGetSavedStarGiftsRequest{Peer: channelPeer}) if err != nil { t.Fatalf("getSavedStarGifts(channel): %v", err) } if savedRes.Count != 1 || len(savedRes.Gifts) != 1 { t.Fatalf("channel saved gifts = count %d len %d, want 1/1", savedRes.Count, len(savedRes.Gifts)) } if !savedRes.Gifts[0].CanUpgrade { t.Fatal("channel saved gift must advertise upgrade when a collectible pool is available") } savedID, ok := savedRes.Gifts[0].GetSavedID() if !ok || savedID <= 0 { t.Fatalf("saved gift saved_id = %d ok %v, want positive", savedID, ok) } if _, ok := savedRes.Gifts[0].GetMsgID(); ok { t.Fatalf("channel saved gift should not expose inputSavedStarGiftUser.msg_id") } nextHistoryID := channel.TopMessageID + 1 history, err := r.onChannelsGetMessages(ownerCtx, &tg.ChannelsGetMessagesRequest{ Channel: channelInput, ID: []tg.InputMessageClass{&tg.InputMessageID{ID: nextHistoryID}}, }) if err != nil { t.Fatalf("get channel message after gift payment: %v", err) } gotMessages := history.(*tg.MessagesMessages).Messages if len(gotMessages) != 1 { t.Fatalf("channel getMessages len = %d, want 1 messageEmpty", len(gotMessages)) } if _, ok := gotMessages[0].(*tg.MessageEmpty); !ok { t.Fatalf("channel gift leaked into message history as %T", gotMessages[0]) } sendFilter := tg.ChannelAdminLogEventsFilter{} sendFilter.SetSend(true) adminReq := &tg.ChannelsGetAdminLogRequest{Channel: channelInput, Limit: 10} adminReq.SetEventsFilter(sendFilter) adminLog, err := r.onChannelsGetAdminLog(ownerCtx, adminReq) if err != nil { t.Fatalf("getAdminLog(channel gift): %v", err) } foundAdminGift := false for _, event := range adminLog.Events { send, ok := event.Action.(*tg.ChannelAdminLogEventActionSendMessage) if !ok { continue } svc, ok := send.Message.(*tg.MessageService) if !ok { continue } action, ok := svc.Action.(*tg.MessageActionStarGift) if !ok { continue } if got, ok := action.GetSavedID(); !ok || got != savedID { t.Fatalf("admin log star gift saved_id = %d ok %v, want %d", got, ok, savedID) } if peer, ok := action.GetPeer(); !ok { t.Fatalf("admin log star gift peer missing") } else if ch, ok := peer.(*tg.PeerChannel); !ok || ch.ChannelID != channel.ID { t.Fatalf("admin log star gift peer = %#v, want channel %d", peer, channel.ID) } foundAdminGift = true } if !foundAdminGift { t.Fatalf("admin log did not include star gift send_message action") } oneRes, err := r.onPaymentsGetSavedStarGift(senderCtx, []tg.InputSavedStarGiftClass{ &tg.InputSavedStarGiftChat{Peer: channelPeer, SavedID: savedID}, }) if err != nil || oneRes.Count != 1 || len(oneRes.Gifts) != 1 { t.Fatalf("getSavedStarGift(channel) count=%d len=%d err=%v, want 1/1", oneRes.Count, len(oneRes.Gifts), err) } fullRes, err := r.onChannelsGetFullChannel(ownerCtx, channelInput) if err != nil { t.Fatalf("getFullChannel: %v", err) } full, ok := fullRes.FullChat.(*tg.ChannelFull) if !ok { t.Fatalf("full chat = %T, want *tg.ChannelFull", fullRes.FullChat) } if cnt, ok := full.GetStargiftsCount(); !ok || cnt != 1 { t.Fatalf("channelFull stargifts_count = %d ok %v, want 1", cnt, ok) } if !full.GetStargiftsAvailable() { t.Fatalf("channelFull stargifts_available = false, want true for broadcast channel gift entry") } if ok, err := r.onPaymentsSaveStarGift(ownerCtx, &tg.PaymentsSaveStarGiftRequest{ Unsave: true, Stargift: &tg.InputSavedStarGiftChat{Peer: channelPeer, SavedID: savedID}, }); err != nil || !ok { t.Fatalf("saveStarGift(channel hide) = %v err %v", ok, err) } hiddenRes, err := r.onPaymentsGetSavedStarGifts(senderCtx, &tg.PaymentsGetSavedStarGiftsRequest{Peer: channelPeer, ExcludeUnsaved: true}) if err != nil || hiddenRes.Count != 0 || len(hiddenRes.Gifts) != 0 { t.Fatalf("channel excludeUnsaved after hide = count %d len %d err %v, want 0/0", hiddenRes.Count, len(hiddenRes.Gifts), err) } } // 余额不足 → sendStarsForm 返回 BALANCE_TOO_LOW(不发礼、不扣费)。 func TestStarGiftInsufficientBalance(t *testing.T) { ctx := context.Background() users := memory.NewUserStore() dialogs := memory.NewDialogStore() msgStore := memory.NewMessageStore(dialogs) sender, _ := users.Create(ctx, domain.User{AccessHash: 7201, Phone: "15550007201", FirstName: "Poor"}) recipient, _ := users.Create(ctx, domain.User{AccessHash: 7202, Phone: "15550007202", FirstName: "Rich"}) gift := domain.StarGift{ID: 8002, RevisionID: 9002, Stars: 5000, ConvertStars: 5000, Title: "Expensive", Sticker: domain.Document{ID: 701, AccessHash: 7, DCID: 2, MimeType: "application/x-tgsticker", Attributes: []domain.DocumentAttribute{{Kind: domain.DocAttrSticker}}}} giftStore := memory.NewStarGiftStore() giftStore.SeedCatalog([]domain.StarGift{gift}) r := New(Config{DC: 2, IP: "127.0.0.1", Port: 2398}, Deps{ Users: appusers.NewService(users), Messages: appmessages.NewService(msgStore, dialogs), Stars: appstars.NewService(memory.NewStarsStore(), appstars.WithStartingGrant(1000)), // < 5000 Gifts: appstargifts.NewService(giftStore, nil, 2), }, zaptest.NewLogger(t), clock.System) senderCtx := WithUserID(ctx, sender.ID) inv := &tg.InputInvoiceStarGift{Peer: &tg.InputPeerUser{UserID: recipient.ID, AccessHash: recipient.AccessHash}, GiftID: gift.ID} formRes, err := r.onPaymentsGetPaymentForm(senderCtx, &tg.PaymentsGetPaymentFormRequest{Invoice: inv}) if err != nil { t.Fatalf("get expensive gift form: %v", err) } form := formRes.(*tg.PaymentsPaymentFormStarGift) if _, err := r.onPaymentsSendStarsForm(senderCtx, &tg.PaymentsSendStarsFormRequest{FormID: form.FormID, Invoice: inv}); !tgerr.Is(err, "BALANCE_TOO_LOW") { t.Fatalf("over-budget gift should error BALANCE_TOO_LOW") } // 余额未变。 if bal, _ := r.deps.Stars.GetBalance(ctx, sender.ID); bal.Balance != 1000 { t.Fatalf("sender balance = %d, want 1000 unchanged", bal.Balance) } } func TestStarGiftPurchaseFormsAreFreshAndBindPurpose(t *testing.T) { r, sender, recipient, gift := starGiftTestRouter(t) ctx := WithUserID(context.Background(), sender.ID) inv := &tg.InputInvoiceStarGift{Peer: &tg.InputPeerUser{UserID: recipient.ID, AccessHash: recipient.AccessHash}, GiftID: gift.ID} firstRes, err := r.onPaymentsGetPaymentForm(ctx, &tg.PaymentsGetPaymentFormRequest{Invoice: inv}) if err != nil { t.Fatalf("first form: %v", err) } secondRes, err := r.onPaymentsGetPaymentForm(ctx, &tg.PaymentsGetPaymentFormRequest{Invoice: inv}) if err != nil { t.Fatalf("second form: %v", err) } first := firstRes.(*tg.PaymentsPaymentFormStarGift) second := secondRes.(*tg.PaymentsPaymentFormStarGift) if first.FormID == 0 || second.FormID == 0 || first.FormID == second.FormID { t.Fatalf("fresh form ids = %d/%d, want distinct non-zero TL longs", first.FormID, second.FormID) } if _, err := r.onPaymentsSendStarsForm(ctx, &tg.PaymentsSendStarsFormRequest{FormID: first.FormID + second.FormID, Invoice: inv}); !tgerr.Is(err, "FORM_EXPIRED") { t.Fatalf("unknown form err=%v, want FORM_EXPIRED", err) } tampered := *inv tampered.HideName = true if _, err := r.onPaymentsSendStarsForm(ctx, &tg.PaymentsSendStarsFormRequest{FormID: first.FormID, Invoice: &tampered}); !tgerr.Is(err, "PURPOSE_INVALID") { t.Fatalf("tampered form err=%v, want PURPOSE_INVALID", err) } } func TestStarGiftCanPurchaseSameCatalogGiftTwice(t *testing.T) { r, sender, recipient, gift := starGiftTestRouter(t) ctx := WithUserID(context.Background(), sender.ID) inv := &tg.InputInvoiceStarGift{Peer: &tg.InputPeerUser{UserID: recipient.ID, AccessHash: recipient.AccessHash}, GiftID: gift.ID} var formIDs []int64 for i := 0; i < 2; i++ { formRes, err := r.onPaymentsGetPaymentForm(ctx, &tg.PaymentsGetPaymentFormRequest{Invoice: inv}) if err != nil { t.Fatalf("get form %d: %v", i, err) } form := formRes.(*tg.PaymentsPaymentFormStarGift) formIDs = append(formIDs, form.FormID) if _, err := r.onPaymentsSendStarsForm(ctx, &tg.PaymentsSendStarsFormRequest{FormID: form.FormID, Invoice: inv}); err != nil { t.Fatalf("purchase %d: %v", i, err) } } if formIDs[0] == formIDs[1] { t.Fatalf("repeated purchase reused form id %d", formIDs[0]) } saved, err := r.onPaymentsGetSavedStarGifts(WithUserID(context.Background(), recipient.ID), &tg.PaymentsGetSavedStarGiftsRequest{ Peer: &tg.InputPeerSelf{}, Limit: 10, }) if err != nil { t.Fatalf("get recipient gifts: %v", err) } if saved.Count != 2 || len(saved.Gifts) != 2 { t.Fatalf("recipient gifts = count %d len %d, want two independent gifts", saved.Count, len(saved.Gifts)) } } func TestStarsTopupInvoiceFallbackCreditsBalance(t *testing.T) { r, sender, _, _ := starGiftTestRouter(t) ctx := context.Background() senderCtx := WithUserID(ctx, sender.ID) opt := devStarsTopupOptions()[1] inv := &tg.InputInvoiceStars{Purpose: &tg.InputStorePaymentStarsTopup{ Stars: opt.Stars, Currency: opt.Currency, Amount: opt.Amount, }} formRes, err := r.onPaymentsGetPaymentForm(senderCtx, &tg.PaymentsGetPaymentFormRequest{Invoice: inv}) if err != nil { t.Fatalf("getPaymentForm topup: %v", err) } form, ok := formRes.(*tg.PaymentsPaymentFormStars) if !ok { t.Fatalf("form = %T, want *tg.PaymentsPaymentFormStars", formRes) } if form.FormID != starsTopupFormID(sender.ID, opt.Stars, opt.Currency, opt.Amount) { t.Fatalf("form id = %d, want deterministic topup id", form.FormID) } if form.BotID != domain.OfficialSystemUserID || len(form.Users) != 1 { t.Fatalf("form bot/users = %d/%d, want official system user", form.BotID, len(form.Users)) } if form.Invoice.Currency != "XTR" || len(form.Invoice.Prices) != 1 || form.Invoice.Prices[0].Amount != opt.Stars { t.Fatalf("form invoice = %+v, want XTR + 1 price %d", form.Invoice, opt.Stars) } if _, err := r.onPaymentsSendStarsForm(senderCtx, &tg.PaymentsSendStarsFormRequest{FormID: form.FormID + 1, Invoice: inv}); !tgerr.Is(err, "STARS_FORM_AMOUNT_MISMATCH") { t.Fatalf("sendStarsForm bad form err = %v, want STARS_FORM_AMOUNT_MISMATCH", err) } if bal, _ := r.deps.Stars.GetBalance(ctx, sender.ID); bal.Balance != 1000 { t.Fatalf("balance after bad form = %d, want 1000 unchanged", bal.Balance) } payRes, err := r.onPaymentsSendStarsForm(senderCtx, &tg.PaymentsSendStarsFormRequest{FormID: form.FormID, Invoice: inv}) if err != nil { t.Fatalf("sendStarsForm topup: %v", err) } pay, ok := payRes.(*tg.PaymentsPaymentResult) if !ok { t.Fatalf("pay result = %T, want *tg.PaymentsPaymentResult", payRes) } updates, ok := pay.Updates.(*tg.Updates) if !ok { t.Fatalf("pay updates = %T, want *tg.Updates", pay.Updates) } foundBalance := false for _, up := range updates.Updates { if balance, ok := up.(*tg.UpdateStarsBalance); ok { foundBalance = true if amt, ok := balance.Balance.(*tg.StarsAmount); !ok || amt.Amount != 3500 { t.Fatalf("updateStarsBalance = %#v, want 3500", balance.Balance) } } } if !foundBalance { t.Fatalf("payment updates missing updateStarsBalance: %#v", updates.Updates) } if bal, _ := r.deps.Stars.GetBalance(ctx, sender.ID); bal.Balance != 3500 { t.Fatalf("balance after topup = %d, want 3500", bal.Balance) } page, err := r.deps.Stars.ListTransactions(ctx, sender.ID, "", 10) if err != nil { t.Fatalf("list transactions: %v", err) } hasTopup := false for _, tx := range page.Transactions { if tx.Reason == domain.StarsReasonTopup && tx.Amount == opt.Stars { hasTopup = true } } if !hasTopup { t.Fatalf("transactions missing topup %d: %+v", opt.Stars, page.Transactions) } } func TestStarsTopupRejectsUnlistedAmount(t *testing.T) { r, sender, _, _ := starGiftTestRouter(t) ctx := WithUserID(context.Background(), sender.ID) inv := &tg.InputInvoiceStars{Purpose: &tg.InputStorePaymentStarsTopup{ Stars: 2501, Currency: "USD", Amount: 199, }} _, err := r.onPaymentsGetPaymentForm(ctx, &tg.PaymentsGetPaymentFormRequest{Invoice: inv}) if !tgerr.Is(err, "STARS_FORM_AMOUNT_MISMATCH") { t.Fatalf("getPaymentForm unlisted err = %v, want STARS_FORM_AMOUNT_MISMATCH", err) } }