mirror of
https://github.com/Mezeporta/Erupe.git
synced 2026-03-21 23:22:34 +01:00
f17cb96b52
The config package used `package _config` with a leading underscore, which is unconventional in Go. Rename to `package config` (matching the directory name) and use `cfg` as the standard import alias across all 93 importing files.
527 lines
13 KiB
Go
527 lines
13 KiB
Go
package mhfitem
|
|
|
|
import (
|
|
"bytes"
|
|
"erupe-ce/common/byteframe"
|
|
"erupe-ce/common/token"
|
|
cfg "erupe-ce/config"
|
|
"testing"
|
|
)
|
|
|
|
func TestReadWarehouseItem(t *testing.T) {
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteUint32(12345) // WarehouseID
|
|
bf.WriteUint16(100) // ItemID
|
|
bf.WriteUint16(5) // Quantity
|
|
bf.WriteUint32(999999) // Unk0
|
|
|
|
_, _ = bf.Seek(0, 0)
|
|
item := ReadWarehouseItem(bf)
|
|
|
|
if item.WarehouseID != 12345 {
|
|
t.Errorf("WarehouseID = %d, want 12345", item.WarehouseID)
|
|
}
|
|
if item.Item.ItemID != 100 {
|
|
t.Errorf("ItemID = %d, want 100", item.Item.ItemID)
|
|
}
|
|
if item.Quantity != 5 {
|
|
t.Errorf("Quantity = %d, want 5", item.Quantity)
|
|
}
|
|
if item.Unk0 != 999999 {
|
|
t.Errorf("Unk0 = %d, want 999999", item.Unk0)
|
|
}
|
|
}
|
|
|
|
func TestReadWarehouseItem_ZeroWarehouseID(t *testing.T) {
|
|
// When WarehouseID is 0, it should be replaced with a random value
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteUint32(0) // WarehouseID = 0
|
|
bf.WriteUint16(100) // ItemID
|
|
bf.WriteUint16(5) // Quantity
|
|
bf.WriteUint32(0) // Unk0
|
|
|
|
_, _ = bf.Seek(0, 0)
|
|
item := ReadWarehouseItem(bf)
|
|
|
|
if item.WarehouseID == 0 {
|
|
t.Error("WarehouseID should be replaced with random value when input is 0")
|
|
}
|
|
}
|
|
|
|
func TestMHFItemStack_ToBytes(t *testing.T) {
|
|
item := MHFItemStack{
|
|
WarehouseID: 12345,
|
|
Item: MHFItem{ItemID: 100},
|
|
Quantity: 5,
|
|
Unk0: 999999,
|
|
}
|
|
|
|
data := item.ToBytes()
|
|
if len(data) != 12 { // 4 + 2 + 2 + 4
|
|
t.Errorf("ToBytes() length = %d, want 12", len(data))
|
|
}
|
|
|
|
// Read it back
|
|
bf := byteframe.NewByteFrameFromBytes(data)
|
|
readItem := ReadWarehouseItem(bf)
|
|
|
|
if readItem.WarehouseID != item.WarehouseID {
|
|
t.Errorf("WarehouseID = %d, want %d", readItem.WarehouseID, item.WarehouseID)
|
|
}
|
|
if readItem.Item.ItemID != item.Item.ItemID {
|
|
t.Errorf("ItemID = %d, want %d", readItem.Item.ItemID, item.Item.ItemID)
|
|
}
|
|
if readItem.Quantity != item.Quantity {
|
|
t.Errorf("Quantity = %d, want %d", readItem.Quantity, item.Quantity)
|
|
}
|
|
if readItem.Unk0 != item.Unk0 {
|
|
t.Errorf("Unk0 = %d, want %d", readItem.Unk0, item.Unk0)
|
|
}
|
|
}
|
|
|
|
func TestSerializeWarehouseItems(t *testing.T) {
|
|
items := []MHFItemStack{
|
|
{WarehouseID: 1, Item: MHFItem{ItemID: 100}, Quantity: 5, Unk0: 0},
|
|
{WarehouseID: 2, Item: MHFItem{ItemID: 200}, Quantity: 10, Unk0: 0},
|
|
}
|
|
|
|
data := SerializeWarehouseItems(items)
|
|
bf := byteframe.NewByteFrameFromBytes(data)
|
|
|
|
count := bf.ReadUint16()
|
|
if count != 2 {
|
|
t.Errorf("count = %d, want 2", count)
|
|
}
|
|
|
|
bf.ReadUint16() // Skip unused
|
|
|
|
for i := 0; i < 2; i++ {
|
|
item := ReadWarehouseItem(bf)
|
|
if item.WarehouseID != items[i].WarehouseID {
|
|
t.Errorf("item[%d] WarehouseID = %d, want %d", i, item.WarehouseID, items[i].WarehouseID)
|
|
}
|
|
if item.Item.ItemID != items[i].Item.ItemID {
|
|
t.Errorf("item[%d] ItemID = %d, want %d", i, item.Item.ItemID, items[i].Item.ItemID)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSerializeWarehouseItems_Empty(t *testing.T) {
|
|
items := []MHFItemStack{}
|
|
data := SerializeWarehouseItems(items)
|
|
bf := byteframe.NewByteFrameFromBytes(data)
|
|
|
|
count := bf.ReadUint16()
|
|
if count != 0 {
|
|
t.Errorf("count = %d, want 0", count)
|
|
}
|
|
}
|
|
|
|
func TestDiffItemStacks(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
old []MHFItemStack
|
|
update []MHFItemStack
|
|
wantLen int
|
|
checkFn func(t *testing.T, result []MHFItemStack)
|
|
}{
|
|
{
|
|
name: "update existing quantity",
|
|
old: []MHFItemStack{
|
|
{WarehouseID: 1, Item: MHFItem{ItemID: 100}, Quantity: 5},
|
|
},
|
|
update: []MHFItemStack{
|
|
{WarehouseID: 1, Item: MHFItem{ItemID: 100}, Quantity: 10},
|
|
},
|
|
wantLen: 1,
|
|
checkFn: func(t *testing.T, result []MHFItemStack) {
|
|
if result[0].Quantity != 10 {
|
|
t.Errorf("Quantity = %d, want 10", result[0].Quantity)
|
|
}
|
|
},
|
|
},
|
|
{
|
|
name: "add new item",
|
|
old: []MHFItemStack{
|
|
{WarehouseID: 1, Item: MHFItem{ItemID: 100}, Quantity: 5},
|
|
},
|
|
update: []MHFItemStack{
|
|
{WarehouseID: 1, Item: MHFItem{ItemID: 100}, Quantity: 5},
|
|
{WarehouseID: 0, Item: MHFItem{ItemID: 200}, Quantity: 3}, // WarehouseID 0 = new
|
|
},
|
|
wantLen: 2,
|
|
checkFn: func(t *testing.T, result []MHFItemStack) {
|
|
hasNewItem := false
|
|
for _, item := range result {
|
|
if item.Item.ItemID == 200 {
|
|
hasNewItem = true
|
|
if item.WarehouseID == 0 {
|
|
t.Error("New item should have generated WarehouseID")
|
|
}
|
|
}
|
|
}
|
|
if !hasNewItem {
|
|
t.Error("New item should be in result")
|
|
}
|
|
},
|
|
},
|
|
{
|
|
name: "remove item (quantity 0)",
|
|
old: []MHFItemStack{
|
|
{WarehouseID: 1, Item: MHFItem{ItemID: 100}, Quantity: 5},
|
|
{WarehouseID: 2, Item: MHFItem{ItemID: 200}, Quantity: 10},
|
|
},
|
|
update: []MHFItemStack{
|
|
{WarehouseID: 1, Item: MHFItem{ItemID: 100}, Quantity: 0}, // Removed
|
|
},
|
|
wantLen: 1,
|
|
checkFn: func(t *testing.T, result []MHFItemStack) {
|
|
for _, item := range result {
|
|
if item.WarehouseID == 1 {
|
|
t.Error("Item with quantity 0 should be removed")
|
|
}
|
|
}
|
|
},
|
|
},
|
|
{
|
|
name: "empty old, add new",
|
|
old: []MHFItemStack{},
|
|
update: []MHFItemStack{{WarehouseID: 0, Item: MHFItem{ItemID: 100}, Quantity: 5}},
|
|
wantLen: 1,
|
|
checkFn: func(t *testing.T, result []MHFItemStack) {
|
|
if len(result) != 1 || result[0].Item.ItemID != 100 {
|
|
t.Error("Should add new item to empty list")
|
|
}
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
result := DiffItemStacks(tt.old, tt.update)
|
|
if len(result) != tt.wantLen {
|
|
t.Errorf("DiffItemStacks() length = %d, want %d", len(result), tt.wantLen)
|
|
}
|
|
if tt.checkFn != nil {
|
|
tt.checkFn(t, result)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestReadWarehouseEquipment(t *testing.T) {
|
|
mode := cfg.Z1
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteUint32(12345) // WarehouseID
|
|
bf.WriteUint8(1) // ItemType
|
|
bf.WriteUint8(2) // Unk0
|
|
bf.WriteUint16(100) // ItemID
|
|
bf.WriteUint16(5) // Level
|
|
|
|
// Write 3 decorations
|
|
bf.WriteUint16(201)
|
|
bf.WriteUint16(202)
|
|
bf.WriteUint16(203)
|
|
|
|
// Write 3 sigils (G1+)
|
|
for i := 0; i < 3; i++ {
|
|
// 3 effects per sigil
|
|
for j := 0; j < 3; j++ {
|
|
bf.WriteUint16(uint16(300 + i*10 + j)) // Effect ID
|
|
}
|
|
for j := 0; j < 3; j++ {
|
|
bf.WriteUint16(uint16(1 + j)) // Effect Level
|
|
}
|
|
bf.WriteUint8(10)
|
|
bf.WriteUint8(11)
|
|
bf.WriteUint8(12)
|
|
bf.WriteUint8(13)
|
|
}
|
|
|
|
// Unk1 (Z1+)
|
|
bf.WriteUint16(9999)
|
|
|
|
_, _ = bf.Seek(0, 0)
|
|
equipment := ReadWarehouseEquipment(bf, mode)
|
|
|
|
if equipment.WarehouseID != 12345 {
|
|
t.Errorf("WarehouseID = %d, want 12345", equipment.WarehouseID)
|
|
}
|
|
if equipment.ItemType != 1 {
|
|
t.Errorf("ItemType = %d, want 1", equipment.ItemType)
|
|
}
|
|
if equipment.ItemID != 100 {
|
|
t.Errorf("ItemID = %d, want 100", equipment.ItemID)
|
|
}
|
|
if equipment.Level != 5 {
|
|
t.Errorf("Level = %d, want 5", equipment.Level)
|
|
}
|
|
if equipment.Decorations[0].ItemID != 201 {
|
|
t.Errorf("Decoration[0] = %d, want 201", equipment.Decorations[0].ItemID)
|
|
}
|
|
if equipment.Sigils[0].Effects[0].ID != 300 {
|
|
t.Errorf("Sigil[0].Effect[0].ID = %d, want 300", equipment.Sigils[0].Effects[0].ID)
|
|
}
|
|
if equipment.Unk1 != 9999 {
|
|
t.Errorf("Unk1 = %d, want 9999", equipment.Unk1)
|
|
}
|
|
}
|
|
|
|
func TestReadWarehouseEquipment_ZeroWarehouseID(t *testing.T) {
|
|
mode := cfg.Z1
|
|
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteUint32(0) // WarehouseID = 0
|
|
bf.WriteUint8(1)
|
|
bf.WriteUint8(2)
|
|
bf.WriteUint16(100)
|
|
bf.WriteUint16(5)
|
|
// Write decorations
|
|
for i := 0; i < 3; i++ {
|
|
bf.WriteUint16(0)
|
|
}
|
|
// Write sigils
|
|
for i := 0; i < 3; i++ {
|
|
for j := 0; j < 6; j++ {
|
|
bf.WriteUint16(0)
|
|
}
|
|
bf.WriteUint8(0)
|
|
bf.WriteUint8(0)
|
|
bf.WriteUint8(0)
|
|
bf.WriteUint8(0)
|
|
}
|
|
bf.WriteUint16(0)
|
|
|
|
_, _ = bf.Seek(0, 0)
|
|
equipment := ReadWarehouseEquipment(bf, mode)
|
|
|
|
if equipment.WarehouseID == 0 {
|
|
t.Error("WarehouseID should be replaced with random value when input is 0")
|
|
}
|
|
}
|
|
|
|
func TestMHFEquipment_ToBytes(t *testing.T) {
|
|
mode := cfg.Z1
|
|
|
|
equipment := MHFEquipment{
|
|
WarehouseID: 12345,
|
|
ItemType: 1,
|
|
Unk0: 2,
|
|
ItemID: 100,
|
|
Level: 5,
|
|
Decorations: []MHFItem{{ItemID: 201}, {ItemID: 202}, {ItemID: 203}},
|
|
Sigils: make([]MHFSigil, 3),
|
|
Unk1: 9999,
|
|
}
|
|
for i := 0; i < 3; i++ {
|
|
equipment.Sigils[i].Effects = make([]MHFSigilEffect, 3)
|
|
}
|
|
|
|
data := equipment.ToBytes(mode)
|
|
bf := byteframe.NewByteFrameFromBytes(data)
|
|
readEquipment := ReadWarehouseEquipment(bf, mode)
|
|
|
|
if readEquipment.WarehouseID != equipment.WarehouseID {
|
|
t.Errorf("WarehouseID = %d, want %d", readEquipment.WarehouseID, equipment.WarehouseID)
|
|
}
|
|
if readEquipment.ItemID != equipment.ItemID {
|
|
t.Errorf("ItemID = %d, want %d", readEquipment.ItemID, equipment.ItemID)
|
|
}
|
|
if readEquipment.Level != equipment.Level {
|
|
t.Errorf("Level = %d, want %d", readEquipment.Level, equipment.Level)
|
|
}
|
|
if readEquipment.Unk1 != equipment.Unk1 {
|
|
t.Errorf("Unk1 = %d, want %d", readEquipment.Unk1, equipment.Unk1)
|
|
}
|
|
}
|
|
|
|
func TestSerializeWarehouseEquipment(t *testing.T) {
|
|
mode := cfg.Z1
|
|
|
|
equipment := []MHFEquipment{
|
|
{
|
|
WarehouseID: 1,
|
|
ItemType: 1,
|
|
ItemID: 100,
|
|
Level: 5,
|
|
Decorations: []MHFItem{{ItemID: 0}, {ItemID: 0}, {ItemID: 0}},
|
|
Sigils: make([]MHFSigil, 3),
|
|
},
|
|
{
|
|
WarehouseID: 2,
|
|
ItemType: 2,
|
|
ItemID: 200,
|
|
Level: 10,
|
|
Decorations: []MHFItem{{ItemID: 0}, {ItemID: 0}, {ItemID: 0}},
|
|
Sigils: make([]MHFSigil, 3),
|
|
},
|
|
}
|
|
for i := range equipment {
|
|
for j := 0; j < 3; j++ {
|
|
equipment[i].Sigils[j].Effects = make([]MHFSigilEffect, 3)
|
|
}
|
|
}
|
|
|
|
data := SerializeWarehouseEquipment(equipment, mode)
|
|
bf := byteframe.NewByteFrameFromBytes(data)
|
|
|
|
count := bf.ReadUint16()
|
|
if count != 2 {
|
|
t.Errorf("count = %d, want 2", count)
|
|
}
|
|
}
|
|
|
|
func TestMHFEquipment_RoundTrip(t *testing.T) {
|
|
mode := cfg.Z1
|
|
|
|
original := MHFEquipment{
|
|
WarehouseID: 99999,
|
|
ItemType: 5,
|
|
Unk0: 10,
|
|
ItemID: 500,
|
|
Level: 25,
|
|
Decorations: []MHFItem{{ItemID: 1}, {ItemID: 2}, {ItemID: 3}},
|
|
Sigils: make([]MHFSigil, 3),
|
|
Unk1: 12345,
|
|
}
|
|
for i := 0; i < 3; i++ {
|
|
original.Sigils[i].Effects = []MHFSigilEffect{
|
|
{ID: uint16(100 + i), Level: 1},
|
|
{ID: uint16(200 + i), Level: 2},
|
|
{ID: uint16(300 + i), Level: 3},
|
|
}
|
|
}
|
|
|
|
// Write to bytes
|
|
data := original.ToBytes(mode)
|
|
|
|
// Read back
|
|
bf := byteframe.NewByteFrameFromBytes(data)
|
|
recovered := ReadWarehouseEquipment(bf, mode)
|
|
|
|
// Compare
|
|
if recovered.WarehouseID != original.WarehouseID {
|
|
t.Errorf("WarehouseID = %d, want %d", recovered.WarehouseID, original.WarehouseID)
|
|
}
|
|
if recovered.ItemType != original.ItemType {
|
|
t.Errorf("ItemType = %d, want %d", recovered.ItemType, original.ItemType)
|
|
}
|
|
if recovered.ItemID != original.ItemID {
|
|
t.Errorf("ItemID = %d, want %d", recovered.ItemID, original.ItemID)
|
|
}
|
|
if recovered.Level != original.Level {
|
|
t.Errorf("Level = %d, want %d", recovered.Level, original.Level)
|
|
}
|
|
for i := 0; i < 3; i++ {
|
|
if recovered.Decorations[i].ItemID != original.Decorations[i].ItemID {
|
|
t.Errorf("Decoration[%d] = %d, want %d", i, recovered.Decorations[i].ItemID, original.Decorations[i].ItemID)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkReadWarehouseItem(b *testing.B) {
|
|
bf := byteframe.NewByteFrame()
|
|
bf.WriteUint32(12345)
|
|
bf.WriteUint16(100)
|
|
bf.WriteUint16(5)
|
|
bf.WriteUint32(0)
|
|
data := bf.Data()
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
bf := byteframe.NewByteFrameFromBytes(data)
|
|
_ = ReadWarehouseItem(bf)
|
|
}
|
|
}
|
|
|
|
func BenchmarkDiffItemStacks(b *testing.B) {
|
|
old := []MHFItemStack{
|
|
{WarehouseID: 1, Item: MHFItem{ItemID: 100}, Quantity: 5},
|
|
{WarehouseID: 2, Item: MHFItem{ItemID: 200}, Quantity: 10},
|
|
{WarehouseID: 3, Item: MHFItem{ItemID: 300}, Quantity: 15},
|
|
}
|
|
update := []MHFItemStack{
|
|
{WarehouseID: 1, Item: MHFItem{ItemID: 100}, Quantity: 8},
|
|
{WarehouseID: 0, Item: MHFItem{ItemID: 400}, Quantity: 3},
|
|
}
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_ = DiffItemStacks(old, update)
|
|
}
|
|
}
|
|
|
|
func BenchmarkSerializeWarehouseItems(b *testing.B) {
|
|
items := make([]MHFItemStack, 100)
|
|
for i := range items {
|
|
items[i] = MHFItemStack{
|
|
WarehouseID: uint32(i),
|
|
Item: MHFItem{ItemID: uint16(i)},
|
|
Quantity: uint16(i % 99),
|
|
}
|
|
}
|
|
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
_ = SerializeWarehouseItems(items)
|
|
}
|
|
}
|
|
|
|
func TestMHFItemStack_ToBytes_RoundTrip(t *testing.T) {
|
|
original := MHFItemStack{
|
|
WarehouseID: 12345,
|
|
Item: MHFItem{ItemID: 999},
|
|
Quantity: 42,
|
|
Unk0: 777,
|
|
}
|
|
|
|
data := original.ToBytes()
|
|
bf := byteframe.NewByteFrameFromBytes(data)
|
|
recovered := ReadWarehouseItem(bf)
|
|
|
|
if !bytes.Equal(original.ToBytes(), recovered.ToBytes()) {
|
|
t.Error("Round-trip serialization failed")
|
|
}
|
|
}
|
|
|
|
func TestDiffItemStacks_PreserveOldWarehouseID(t *testing.T) {
|
|
// Verify that when updating existing items, the old WarehouseID is preserved
|
|
old := []MHFItemStack{
|
|
{WarehouseID: 555, Item: MHFItem{ItemID: 100}, Quantity: 5},
|
|
}
|
|
update := []MHFItemStack{
|
|
{WarehouseID: 555, Item: MHFItem{ItemID: 100}, Quantity: 10},
|
|
}
|
|
|
|
result := DiffItemStacks(old, update)
|
|
if len(result) != 1 {
|
|
t.Fatalf("Expected 1 item, got %d", len(result))
|
|
}
|
|
if result[0].WarehouseID != 555 {
|
|
t.Errorf("WarehouseID = %d, want 555", result[0].WarehouseID)
|
|
}
|
|
if result[0].Quantity != 10 {
|
|
t.Errorf("Quantity = %d, want 10", result[0].Quantity)
|
|
}
|
|
}
|
|
|
|
func TestDiffItemStacks_GeneratesNewWarehouseID(t *testing.T) {
|
|
// Verify that new items get a generated WarehouseID
|
|
old := []MHFItemStack{}
|
|
update := []MHFItemStack{
|
|
{WarehouseID: 0, Item: MHFItem{ItemID: 100}, Quantity: 5},
|
|
}
|
|
|
|
// Reset RNG for consistent test
|
|
token.RNG = token.NewSafeRand()
|
|
|
|
result := DiffItemStacks(old, update)
|
|
if len(result) != 1 {
|
|
t.Fatalf("Expected 1 item, got %d", len(result))
|
|
}
|
|
if result[0].WarehouseID == 0 {
|
|
t.Error("New item should have generated WarehouseID, got 0")
|
|
}
|
|
}
|