package d2player import ( "github.com/stretchr/testify/assert" "testing" ) type TestItem struct { width int height int inventorySlotX int inventorySlotY int } func (t *TestItem) InventoryGridSize() (int, int) { return t.width, t.height } func (t *TestItem) ItemCode() string { return "" } func (t *TestItem) InventoryGridSlot() (int, int) { return t.inventorySlotX, t.inventorySlotY } func (t *TestItem) SetInventoryGridSlot(x int, y int) { t.inventorySlotX, t.inventorySlotY = x, y } func NewTestItem(width int, height int) *TestItem { return &TestItem{width: width, height: height} } func TestItemGrid_Add_Basic(t *testing.T) { grid := NewItemGrid(2, 2, 0, 0) tl := NewTestItem(1, 1) tr := NewTestItem(1, 1) bl := NewTestItem(1, 1) br := NewTestItem(1, 1) added, err := grid.Add(tl, tr, bl, br) assert.Equal(t, 4, added) assert.NoError(t, err) assert.Equal(t, tl, grid.GetSlot(0, 0)) assert.Equal(t, tr, grid.GetSlot(1, 0)) assert.Equal(t, bl, grid.GetSlot(0, 1)) assert.Equal(t, br, grid.GetSlot(1, 1)) } func TestItemGrid_Add_OverflowBasic(t *testing.T) { grid := NewItemGrid(2, 2, 0, 0) tl := NewTestItem(1, 1) tr := NewTestItem(1, 1) bl := NewTestItem(1, 1) br := NewTestItem(1, 1) o := NewTestItem(1, 1) added, err := grid.Add(tl, tr, bl, br, o) assert.Equal(t, 4, added) assert.Error(t, err) assert.Equal(t, tl, grid.GetSlot(0, 0)) assert.Equal(t, tr, grid.GetSlot(1, 0)) assert.Equal(t, bl, grid.GetSlot(0, 1)) assert.Equal(t, br, grid.GetSlot(1, 1)) } func TestItemGrid_Add_LargeItem(t *testing.T) { grid := NewItemGrid(3, 3, 0, 0) tl := NewTestItem(1, 1) o := NewTestItem(2, 2) added, err := grid.Add(tl, o) assert.Equal(t, 2, added) assert.NoError(t, err) assert.Equal(t, tl, grid.GetSlot(0, 0)) assert.Equal(t, o, grid.GetSlot(1, 0)) assert.Equal(t, o, grid.GetSlot(2, 0)) assert.Nil(t, grid.GetSlot(0, 1)) assert.Equal(t, o, grid.GetSlot(1, 1)) assert.Equal(t, o, grid.GetSlot(2, 1)) assert.Nil(t, grid.GetSlot(0, 2)) assert.Nil(t, grid.GetSlot(1, 2)) assert.Nil(t, grid.GetSlot(2, 2)) } func TestItemGrid_Add_OverflowLargeItem(t *testing.T) { grid := NewItemGrid(2, 2, 0, 0) tl := NewTestItem(1, 1) o := NewTestItem(2, 2) added, err := grid.Add(tl, o) assert.Equal(t, 1, added) assert.Error(t, err) assert.Equal(t, tl, grid.GetSlot(0, 0)) assert.Nil(t, grid.GetSlot(1, 0)) assert.Nil(t, grid.GetSlot(0, 1)) assert.Nil(t, grid.GetSlot(1, 1)) }