From 383e834cefe08c4a9eb11cfa328f518feb6a5ac7 Mon Sep 17 00:00:00 2001 From: Marc Di Luzio Date: Fri, 26 Jun 2020 20:14:00 +0100 Subject: [PATCH] Add RoverStash test --- pkg/game/world_test.go | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/pkg/game/world_test.go b/pkg/game/world_test.go index ca4d859..4e69e77 100644 --- a/pkg/game/world_test.go +++ b/pkg/game/world_test.go @@ -141,3 +141,28 @@ func TestWorld_RadarFromRover(t *testing.T) { assert.Equal(t, objects.LargeRock, radar[(i*9)+7]) } } + +func TestWorld_RoverStash(t *testing.T) { + world := NewWorld(2, 2) + a, err := world.SpawnRover() + assert.NoError(t, err) + + pos := vector.Vector{ + X: 0.0, + Y: 0.0, + } + + err = world.WarpRover(a, pos) + assert.NoError(t, err, "Failed to set position for rover") + + err = world.Atlas.SetTile(pos, objects.SmallRock) + assert.NoError(t, err, "Failed to set tile to rock") + + o, err := world.RoverStash(a) + assert.NoError(t, err, "Failed to stash") + assert.Equal(t, objects.SmallRock, o, "Failed to get correct object") + + tile, err := world.Atlas.GetTile(pos) + assert.NoError(t, err, "Failed to get tile") + assert.Equal(t, objects.Empty, tile, "Stash failed to remove object from atlas") +}