astra/owpengram-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: sync privacy read-model rules

Browse source
This commit is contained in:
iamxvbaba 2026-07-24 11:56:59 +08:00
parent 9f467f4be7
commit 5d5883a3d1
20 changed files with 1011 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

67
internal/rpc/payments_star_gifts_rpc_test.go
View file

@ -13,6 +13,7 @@ import (
"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"
@ -983,6 +984,72 @@ func TestStarGiftSaga(t *testing.T) {
}
}
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) {