cmd: vote

internal/model/game/race_test.go

@@ -0,0 +1,22 @@
+package game_test
+
+import (
+	"testing"
+
+	e "github.com/iliadenisov/galaxy/internal/error"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGiveVotes(t *testing.T) {
+	g := newGame()
+
+	assert.Equal(t, g.Race[Race_0_idx].ID, g.Race[Race_0_idx].Vote)
+	assert.Equal(t, g.Race[Race_1_idx].ID, g.Race[Race_1_idx].Vote)
+
+	assert.NoError(t, g.GiveVotes(Race_0.Name, Race_1.Name))
+	assert.Equal(t, g.Race[Race_1_idx].ID, g.Race[Race_0_idx].Vote)
+	assert.Equal(t, g.Race[Race_1_idx].ID, g.Race[Race_1_idx].Vote)
+
+	assert.ErrorContains(t, g.GiveVotes("UnknownRace", Race_1.Name), e.GenericErrorText(e.ErrInputUnknownRace))
+	assert.ErrorContains(t, g.GiveVotes(Race_0.Name, "UnknownRace"), e.GenericErrorText(e.ErrInputUnknownRace))
+}