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

package rental

import (
	"context"
	"errors"
	"testing"

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

func TestOwnerChargeCommitsPointsAndGold(t *testing.T) {
	ctx := context.Background()
	repositories := dnfrepomemory.NewMemoryGroup()
	seedRentalOwner(t, ctx, repositories)
	owner, err := NewOwner(repositories)
	if err != nil {
		t.Fatalf("NewOwner error = %v", err)
	}

	result, err := owner.Charge(ctx, ChargeCommand{
		AccountID:    "dnf:1",
		CharacterID:  "77",
		Count:        2,
		Limit:        20,
		GoldPerPoint: 100,
	})
	if err != nil {
		t.Fatalf("Charge error = %v", err)
	}
	if result.Points != 7 || result.Gold != 800 {
		t.Fatalf("Charge result = %+v", result)
	}
	account, _, _ := repositories.Account.Load(ctx, "dnf:1")
	character, _, _ := repositories.Character.Load(ctx, "77")
	if account.Metadata[PointMetadataKey] != "7" || character.Stats["gold"] != 800 {
		t.Fatalf("persisted account=%+v character=%+v", account, character)
	}
}

func TestOwnerRentCommitsWalletAndProjectedItemsAtomically(t *testing.T) {
	ctx := context.Background()
	repositories := dnfrepomemory.NewMemoryGroup()
	seedRentalOwner(t, ctx, repositories)
	owner, err := NewOwner(repositories)
	if err != nil {
		t.Fatalf("NewOwner error = %v", err)
	}

	result, err := owner.Rent(ctx, RentCommand{
		AccountID:   "dnf:1",
		CharacterID: "77",
		PointCost:   3,
		Project: func(assets *Assets) (Changes, error) {
			assets.Inventory.Slots["0:9"] = dnfrepo.ItemStack{ItemID: 416000000, Count: 1}
			assets.Equipment.Entries["weapon"] = dnfrepo.EquipmentEntry{ItemID: 416000001, SlotIndex: 0}
			return Changes{Inventory: true, Equipment: true}, nil
		},
	})
	if err != nil {
		t.Fatalf("Rent error = %v", err)
	}
	if result.Points != 2 || result.Gold != 1000 {
		t.Fatalf("Rent result = %+v", result)
	}
	account, _, _ := repositories.Account.Load(ctx, "dnf:1")
	inventory, _, _ := repositories.Inventory.Load(ctx, "77")
	equipment, _, _ := repositories.Equipment.Load(ctx, "77")
	if account.Metadata[PointMetadataKey] != "2" ||
		inventory.Slots["0:9"].ItemID != 416000000 ||
		equipment.Entries["weapon"].ItemID != 416000001 {
		t.Fatalf("persisted account=%+v inventory=%+v equipment=%+v", account, inventory, equipment)
	}
}

func TestOwnerRentRollsBackProjectionAndRejectsCrossAccount(t *testing.T) {
	ctx := context.Background()
	repositories := dnfrepomemory.NewMemoryGroup()
	seedRentalOwner(t, ctx, repositories)
	owner, err := NewOwner(repositories)
	if err != nil {
		t.Fatalf("NewOwner error = %v", err)
	}
	wantErr := errors.New("reject rental projection")
	errResult, err := owner.Rent(ctx, RentCommand{
		AccountID:   "dnf:1",
		CharacterID: "77",
		PointCost:   3,
		Project: func(assets *Assets) (Changes, error) {
			assets.Inventory.Slots["0:9"] = dnfrepo.ItemStack{ItemID: 416000000, Count: 1}
			return Changes{Inventory: true}, wantErr
		},
	})
	if !errors.Is(err, wantErr) || errResult != (WalletResult{}) {
		t.Fatalf("Rent result=%+v error=%v", errResult, err)
	}
	account, _, _ := repositories.Account.Load(ctx, "dnf:1")
	inventory, _, _ := repositories.Inventory.Load(ctx, "77")
	if account.Metadata[PointMetadataKey] != "5" || len(inventory.Slots) != 0 {
		t.Fatalf("rollback account=%+v inventory=%+v", account, inventory)
	}

	if err := repositories.Account.Save(ctx, dnfrepo.AccountRecord{
		AccountID: "dnf:2",
		Metadata:  map[string]string{PointMetadataKey: "5"},
	}); err != nil {
		t.Fatal(err)
	}
	called := false
	_, err = owner.Rent(ctx, RentCommand{
		AccountID:   "dnf:2",
		CharacterID: "77",
		PointCost:   1,
		Project: func(*Assets) (Changes, error) {
			called = true
			return Changes{}, nil
		},
	})
	if !errors.Is(err, ErrOwnerMismatch) || called {
		t.Fatalf("cross-account Rent error=%v called=%t", err, called)
	}
}

func TestOwnerCleanupPersistsOnlyProjectedContainers(t *testing.T) {
	ctx := context.Background()
	repositories := dnfrepomemory.NewMemoryGroup()
	seedRentalOwner(t, ctx, repositories)
	if err := repositories.Inventory.Save(ctx, dnfrepo.InventoryRecord{
		CharacterID: "77",
		Slots:       map[string]dnfrepo.ItemStack{"0:9": {ItemID: 416000000, Count: 1}},
	}); err != nil {
		t.Fatal(err)
	}
	owner, err := NewOwner(repositories)
	if err != nil {
		t.Fatalf("NewOwner error = %v", err)
	}
	err = owner.Cleanup(ctx, CleanupCommand{
		AccountID:   "dnf:1",
		CharacterID: "77",
		Project: func(assets *Assets) (Changes, error) {
			delete(assets.Inventory.Slots, "0:9")
			return Changes{Inventory: true}, nil
		},
	})
	if err != nil {
		t.Fatalf("Cleanup error = %v", err)
	}
	inventory, _, _ := repositories.Inventory.Load(ctx, "77")
	account, _, _ := repositories.Account.Load(ctx, "dnf:1")
	if len(inventory.Slots) != 0 || account.Metadata[PointMetadataKey] != "5" {
		t.Fatalf("cleanup inventory=%+v account=%+v", inventory, account)
	}
}

func TestOwnerUsesScopedRentalAssetTransactionWhenAvailable(t *testing.T) {
	ctx := context.Background()
	repositories := dnfrepomemory.NewMemoryGroup()
	seedRentalOwner(t, ctx, repositories)
	spy := &rentalAssetTransactionSpy{delegate: repositories.RentalAssets}
	repositories.RentalAssets = spy
	owner, err := NewOwner(repositories)
	if err != nil {
		t.Fatalf("NewOwner error = %v", err)
	}

	if _, err := owner.Charge(ctx, ChargeCommand{
		AccountID:    "dnf:1",
		CharacterID:  "77",
		Count:        1,
		Limit:        20,
		GoldPerPoint: 100,
	}); err != nil {
		t.Fatalf("Charge error = %v", err)
	}
	if spy.scopedCalls != 1 || spy.legacyCalls != 0 {
		t.Fatalf("rental transaction routes scoped=%d legacy=%d, want scoped only", spy.scopedCalls, spy.legacyCalls)
	}
}

type rentalAssetTransactionSpy struct {
	delegate    dnfrepo.RentalAssetUnitOfWork
	scopedCalls int
	legacyCalls int
}

func (s *rentalAssetTransactionSpy) WithinRentalAssets(
	ctx context.Context,
	accountID string,
	characterID string,
	apply func(dnfrepo.AccountRepository, dnfrepo.CharacterRepository, dnfrepo.InventoryRepository, dnfrepo.EquipmentRepository) error,
) error {
	s.legacyCalls++
	return s.delegate.WithinRentalAssets(ctx, accountID, characterID, apply)
}

func (s *rentalAssetTransactionSpy) WithinRentalAssetsScoped(
	ctx context.Context,
	accountID string,
	characterID string,
	apply func(dnfrepo.AccountRepository, dnfrepo.CharacterRepository, dnfrepo.InventoryRepository, dnfrepo.EquipmentRepository) error,
) error {
	s.scopedCalls++
	return s.delegate.WithinRentalAssets(ctx, accountID, characterID, apply)
}

func seedRentalOwner(t *testing.T, ctx context.Context, repositories dnfrepo.Group) {
	t.Helper()
	if err := repositories.Account.Save(ctx, dnfrepo.AccountRecord{
		AccountID: "dnf:1",
		Metadata:  map[string]string{PointMetadataKey: "5"},
	}); err != nil {
		t.Fatal(err)
	}
	if err := repositories.Character.Save(ctx, dnfrepo.CharacterRecord{
		CharacterID: "77",
		AccountID:   "dnf:1",
		Stats:       map[string]int64{"gold": 1000},
	}); err != nil {
		t.Fatal(err)
	}
	if err := repositories.Inventory.Save(ctx, dnfrepo.InventoryRecord{
		CharacterID: "77",
		Slots:       map[string]dnfrepo.ItemStack{},
	}); err != nil {
		t.Fatal(err)
	}
	if err := repositories.Equipment.Save(ctx, dnfrepo.EquipmentRecord{
		CharacterID: "77",
		Entries:     map[string]dnfrepo.EquipmentEntry{},
	}); err != nil {
		t.Fatal(err)
	}
}