owpengram-server/internal/store/postgres/star_gift_lifecycle_test.go

package postgres

import (
	"context"
	"testing"

	"telesrv/internal/domain"
)

func TestProjectPrivateStarGiftPurchaseScopesViewerCapabilities(t *testing.T) {
	media := &domain.MessageMedia{
		Kind: domain.MessageMediaKindService,
		ServiceAction: &domain.MessageServiceAction{
			Kind: domain.MessageServiceActionStarGift,
			StarGift: &domain.MessageStarGiftAction{
				PeerUserID: 200, To: domain.Peer{Type: domain.PeerTypeUser, ID: 200},
				CanUpgrade: true, PrepaidUpgradeHash: "prepaid-upgrade-hash-0123456789",
			},
		},
	}
	req := &domain.SendPrivateTextRequest{SenderUserID: 100, RecipientUserID: 200, Media: media}
	projection, err := projectPrivateStarGiftPurchase(context.Background(), nil, req)
	if err != nil {
		t.Fatalf("project purchase: %v", err)
	}
	shared := privateStarGiftAction(projection.Shared)
	sender := privateStarGiftAction(projection.Sender)
	recipient := privateStarGiftAction(projection.Recipient)
	if shared == nil || shared.CanUpgrade || shared.PrepaidUpgradeHash != "" {
		t.Fatalf("shared projection retained viewer capability: %+v", shared)
	}
	if sender == nil || sender.CanUpgrade || sender.PrepaidUpgradeHash == "" {
		t.Fatalf("sender projection = %+v, want hash without can_upgrade", sender)
	}
	if recipient == nil || !recipient.CanUpgrade || recipient.PrepaidUpgradeHash != "" {
		t.Fatalf("recipient projection = %+v, want can_upgrade without hash", recipient)
	}
	if original := privateStarGiftAction(media); original == nil || !original.CanUpgrade || original.PrepaidUpgradeHash == "" {
		t.Fatalf("source projection was mutated: %+v", original)
	}

	selfReq := &domain.SendPrivateTextRequest{SenderUserID: 200, RecipientUserID: 200, Media: media}
	selfProjection, err := projectPrivateStarGiftPurchase(context.Background(), nil, selfReq)
	if err != nil {
		t.Fatalf("project self purchase: %v", err)
	}
	self := privateStarGiftAction(selfProjection.Sender)
	if self == nil || !self.CanUpgrade || self.PrepaidUpgradeHash != "" {
		t.Fatalf("self-owner projection = %+v, want can_upgrade without hash", self)
	}

	bad := *req
	bad.RecipientUserID = 300
	if _, err := projectPrivateStarGiftPurchase(context.Background(), nil, &bad); err == nil {
		t.Fatal("mismatched gift owner and recipient was accepted")
	}
}

func TestTransferUniqueActionSavedIDNamespace(t *testing.T) {
	saved := domain.SavedStarGift{SavedID: 42, CanCraftAt: 1_780_000_123}
	unique := domain.UniqueStarGift{ID: 7}
	user := domain.Peer{Type: domain.PeerTypeUser, ID: 100}
	channel := domain.Peer{Type: domain.PeerTypeChannel, ID: 200}

	if action := transferUniqueAction(unique, 1, user, saved); action.SavedID != 0 || action.CanCraftAt != saved.CanCraftAt {
		t.Fatalf("user transfer action leaked channel saved_id: %+v", action)
	}
	if action := transferUniqueAction(unique, 1, channel, saved); action.SavedID != saved.SavedID || action.CanCraftAt != 0 {
		t.Fatalf("channel transfer action lost channel saved_id: %+v", action)
	}
}

func TestStarGiftCraftReadyAt(t *testing.T) {
	const date = 1_780_000_000
	if got := starGiftCraftReadyAt(date, 0); got != date {
		t.Fatalf("zero-delay craft ready_at = %d, want %d", got, date)
	}
	if got := starGiftCraftReadyAt(date, 60); got != date+60 {
		t.Fatalf("delayed craft ready_at = %d, want %d", got, date+60)
	}
	if got := starGiftCraftReadyAt(0, 0); got != 0 {
		t.Fatalf("invalid-date craft ready_at = %d, want 0", got)
	}
	if got := starGiftCraftReadyAt(1<<31-10, 60); got != 1<<31-1 {
		t.Fatalf("overflow craft ready_at = %d, want max int32", got)
	}
	if got := starGiftCraftReadyAt(1<<31+10, 0); got != 1<<31-1 {
		t.Fatalf("oversized-date craft ready_at = %d, want max int32", got)
	}
}

func TestEncodeSharedPrivateStarGiftMediaOmitsUserBoxLocalRefs(t *testing.T) {
	ordinary := &domain.MessageMedia{
		Kind: domain.MessageMediaKindService,
		ServiceAction: &domain.MessageServiceAction{
			Kind: domain.MessageServiceActionStarGift,
			StarGift: &domain.MessageStarGiftAction{
				PeerUserID: 9,
				SavedID:    10, GiftMsgID: 11, UpgradeMsgID: 12,
			},
		},
	}
	encoded, err := encodeSharedPrivateStarGiftMedia(ordinary)
	if err != nil {
		t.Fatalf("encode ordinary shared projection: %v", err)
	}
	sharedOrdinary, err := decodeMessageMedia(string(encoded))
	if err != nil {
		t.Fatalf("decode ordinary shared projection: %v", err)
	}
	ordinaryAction := sharedOrdinary.ServiceAction.StarGift
	if ordinaryAction.SavedID != 0 || ordinaryAction.GiftMsgID != 0 || ordinaryAction.UpgradeMsgID != 0 {
		t.Fatalf("ordinary shared projection retained box-local refs: %+v", ordinaryAction)
	}
	if original := ordinary.ServiceAction.StarGift; original.SavedID != 10 || original.GiftMsgID != 11 || original.UpgradeMsgID != 12 {
		t.Fatalf("ordinary source projection was mutated: %+v", original)
	}

	unique := &domain.MessageMedia{
		Kind: domain.MessageMediaKindService,
		ServiceAction: &domain.MessageServiceAction{
			Kind: domain.MessageServiceActionStarGiftUnique,
			StarGiftUnique: &domain.MessageStarGiftUniqueAction{
				Peer: domain.Peer{Type: domain.PeerTypeUser, ID: 9}, SavedID: 13,
			},
		},
	}
	encoded, err = encodeSharedPrivateStarGiftMedia(unique)
	if err != nil {
		t.Fatalf("encode unique shared projection: %v", err)
	}
	sharedUnique, err := decodeMessageMedia(string(encoded))
	if err != nil {
		t.Fatalf("decode unique shared projection: %v", err)
	}
	if action := sharedUnique.ServiceAction.StarGiftUnique; action.SavedID != 0 {
		t.Fatalf("unique shared projection retained user saved_id: %+v", action)
	}
	if unique.ServiceAction.StarGiftUnique.SavedID != 13 {
		t.Fatalf("unique source projection was mutated: %+v", unique.ServiceAction.StarGiftUnique)
	}
}

func TestEncodeSharedPrivateStarGiftMediaPreservesChannelSavedID(t *testing.T) {
	media := &domain.MessageMedia{
		Kind: domain.MessageMediaKindService,
		ServiceAction: &domain.MessageServiceAction{
			Kind: domain.MessageServiceActionStarGiftUnique,
			StarGiftUnique: &domain.MessageStarGiftUniqueAction{
				Peer: domain.Peer{Type: domain.PeerTypeChannel, ID: 9}, SavedID: 14,
			},
		},
	}
	encoded, err := encodeSharedPrivateStarGiftMedia(media)
	if err != nil {
		t.Fatalf("encode channel shared projection: %v", err)
	}
	shared, err := decodeMessageMedia(string(encoded))
	if err != nil {
		t.Fatalf("decode channel shared projection: %v", err)
	}
	if action := shared.ServiceAction.StarGiftUnique; action.SavedID != 14 {
		t.Fatalf("channel shared projection lost saved_id: %+v", action)
	}
}