90CNServer/server/internal/modules/dnf/booster/owner_test.go

package booster

import (
	"context"
	"errors"
	"testing"

	dnfrepo "longheng.io/server/internal/modules/dnf/repository"
	dnfrepomemory "longheng.io/server/internal/modules/dnf/repository/memory"
)

func TestOwnerCommitsAccountAndCharacterInventoryTogether(t *testing.T) {
	ctx := context.Background()
	repositories := newOwnerTestRepositories(t, "dnf:1")
	owner, err := NewOwner(repositories)
	if err != nil {
		t.Fatal(err)
	}
	err = owner.Open(ctx, Command{
		AccountID:   "dnf:1",
		CharacterID: "77",
		Project: func(assets *Assets) (Changes, error) {
			if assets.Account.Metadata == nil {
				assets.Account.Metadata = make(map[string]string)
			}
			assets.Account.Metadata["premium_expire_97"] = "4102444800"
			delete(assets.Inventory.Slots, "0:5")
			assets.Inventory.Slots["1:2"] = dnfrepo.ItemStack{ItemID: 9101, Count: 1}
			assets.AccountInventory.Slots[dnfrepo.AccountSharedInventorySlotKey(362)] = dnfrepo.ItemStack{ItemID: 10099773, Count: 1}
			assets.EpicBook.Balances[710] = 23
			return Changes{Account: true, AccountInventory: true, Inventory: true, EpicBook: true}, nil
		},
	})
	if err != nil {
		t.Fatal(err)
	}
	inventory, found, err := repositories.Inventory.Load(ctx, "77")
	if err != nil || !found {
		t.Fatalf("load inventory found=%v err=%v", found, err)
	}
	if _, found := inventory.Slots["0:5"]; found {
		t.Fatalf("source item was not consumed: %+v", inventory.Slots)
	}
	if inventory.Slots["1:2"].ItemID != 9101 {
		t.Fatalf("avatar reward missing: %+v", inventory.Slots)
	}
	accountRecord, found, err := repositories.Account.Load(ctx, "dnf:1")
	if err != nil || !found || accountRecord.Metadata["premium_expire_97"] != "4102444800" {
		t.Fatalf("account premium=%+v found=%v err=%v", accountRecord, found, err)
	}
	account, found, err := repositories.AccountInventory.Load(ctx, "dnf:1")
	if err != nil || !found {
		t.Fatalf("load account inventory found=%v err=%v", found, err)
	}
	if account.Slots[dnfrepo.AccountSharedInventorySlotKey(362)].ItemID != 10099773 {
		t.Fatalf("shared reward missing: %+v", account.Slots)
	}
	book, found, err := repositories.EpicBook.Load(ctx, "dnf:1")
	if err != nil || !found || book.Balances[710] != 23 {
		t.Fatalf("epic book=%+v found=%v err=%v", book, found, err)
	}
}

func TestOwnerRollsBackProjectorFailure(t *testing.T) {
	ctx := context.Background()
	repositories := newOwnerTestRepositories(t, "dnf:1")
	owner, err := NewOwner(repositories)
	if err != nil {
		t.Fatal(err)
	}
	wantErr := errors.New("projection failed")
	err = owner.Open(ctx, Command{
		AccountID:   "dnf:1",
		CharacterID: "77",
		Project: func(assets *Assets) (Changes, error) {
			if assets.Account.Metadata == nil {
				assets.Account.Metadata = make(map[string]string)
			}
			assets.Account.Metadata["premium_expire_97"] = "4102444800"
			delete(assets.Inventory.Slots, "0:5")
			assets.EpicBook.Balances[710] = 23
			return Changes{Inventory: true, EpicBook: true}, wantErr
		},
	})
	if !errors.Is(err, wantErr) {
		t.Fatalf("Open() error=%v want=%v", err, wantErr)
	}
	inventory, found, err := repositories.Inventory.Load(ctx, "77")
	if err != nil || !found {
		t.Fatalf("load inventory found=%v err=%v", found, err)
	}
	if inventory.Slots["0:5"].Count != 1 {
		t.Fatalf("failed projection leaked mutation: %+v", inventory.Slots)
	}
	if accountRecord, found, err := repositories.Account.Load(ctx, "dnf:1"); err != nil || found || len(accountRecord.Metadata) != 0 {
		t.Fatalf("failed projection leaked account=%+v found=%v err=%v", accountRecord, found, err)
	}
	if book, found, err := repositories.EpicBook.Load(ctx, "dnf:1"); err != nil || found || len(book.Balances) != 0 {
		t.Fatalf("failed projection leaked epic book=%+v found=%v err=%v", book, found, err)
	}
}

func TestOwnerRejectsCrossAccountCharacter(t *testing.T) {
	ctx := context.Background()
	repositories := newOwnerTestRepositories(t, "dnf:2")
	owner, err := NewOwner(repositories)
	if err != nil {
		t.Fatal(err)
	}
	err = owner.Open(ctx, Command{
		AccountID:   "dnf:1",
		CharacterID: "77",
		Project: func(*Assets) (Changes, error) {
			return Changes{}, nil
		},
	})
	if !errors.Is(err, ErrAccountMismatch) {
		t.Fatalf("Open() error=%v want=%v", err, ErrAccountMismatch)
	}
}

func newOwnerTestRepositories(t *testing.T, characterAccountID string) dnfrepo.Group {
	t.Helper()
	ctx := context.Background()
	repositories := dnfrepomemory.NewMemoryGroup()
	if err := repositories.Character.Save(ctx, dnfrepo.CharacterRecord{CharacterID: "77", AccountID: characterAccountID}); err != nil {
		t.Fatal(err)
	}
	if err := repositories.Inventory.Save(ctx, dnfrepo.InventoryRecord{
		CharacterID: "77",
		Slots: map[string]dnfrepo.ItemStack{
			"0:5": {ItemID: 490701318, Count: 1},
		},
	}); err != nil {
		t.Fatal(err)
	}
	return repositories
}