Byte-snake-engine/cli/commands/play_test.go

package commands

import (
	"fmt"
	"testing"

	"github.com/BattlesnakeOfficial/rules"
	"github.com/BattlesnakeOfficial/rules/test"
)

func TestGetIndividualBoardStateForSnake(t *testing.T) {
	s1 := rules.Snake{ID: "one", Body: []rules.Point{{X: 3, Y: 3}}}
	s2 := rules.Snake{ID: "two", Body: []rules.Point{{X: 4, Y: 3}}}
	state := &rules.BoardState{
		Height: 11,
		Width:  11,
		Snakes: []rules.Snake{s1, s2},
	}
	s1State := SnakeState{
		ID:    "one",
		Name:  "ONE",
		URL:   "http://example1.com",
		Head:  "safe",
		Tail:  "curled",
		Color: "#123456",
	}
	s2State := SnakeState{
		ID:    "two",
		Name:  "TWO",
		URL:   "http://example2.com",
		Head:  "silly",
		Tail:  "bolt",
		Color: "#654321",
	}
	snakeStates := map[string]SnakeState{
		s1State.ID: s1State,
		s2State.ID: s2State,
	}
	initialiseGameConfig() // initialise default config
	snakeRequest := getIndividualBoardStateForSnake(state, s1State, snakeStates, getRuleset(0, snakeStates))
	requestBody := serialiseSnakeRequest(snakeRequest)

	test.RequireJSONMatchesFixture(t, "testdata/snake_request_body.json", string(requestBody))
}

func TestSettingsRequestSerialization(t *testing.T) {
	s1 := rules.Snake{ID: "one", Body: []rules.Point{{X: 3, Y: 3}}}
	s2 := rules.Snake{ID: "two", Body: []rules.Point{{X: 4, Y: 3}}}
	state := &rules.BoardState{
		Height: 11,
		Width:  11,
		Snakes: []rules.Snake{s1, s2},
	}
	s1State := SnakeState{
		ID:    "one",
		Name:  "ONE",
		URL:   "http://example1.com",
		Head:  "safe",
		Tail:  "curled",
		Color: "#123456",
	}
	s2State := SnakeState{
		ID:    "two",
		Name:  "TWO",
		URL:   "http://example2.com",
		Head:  "silly",
		Tail:  "bolt",
		Color: "#654321",
	}
	snakeStates := map[string]SnakeState{s1State.ID: s1State, s2State.ID: s2State}

	rsb := rules.NewRulesetBuilder().
		WithParams(map[string]string{
			// standard
			rules.ParamFoodSpawnChance:     "11",
			rules.ParamMinimumFood:         "7",
			rules.ParamHazardDamagePerTurn: "19",
			rules.ParamHazardMap:           "hz_spiral",
			rules.ParamHazardMapAuthor:     "altersaddle",
			// squad
			rules.ParamAllowBodyCollisions: "true",
			rules.ParamSharedElimination:   "false",
			rules.ParamSharedHealth:        "true",
			rules.ParamSharedLength:        "false",
			// royale
			rules.ParamShrinkEveryNTurns: "17",
		})

	for _, gt := range []string{
		rules.GameTypeStandard, rules.GameTypeRoyale, rules.GameTypeSolo,
		rules.GameTypeWrapped, rules.GameTypeSquad, rules.GameTypeConstrictor,
	} {
		t.Run(gt, func(t *testing.T) {
			// apply game type
			ruleset := rsb.WithParams(map[string]string{rules.ParamGameType: gt}).Ruleset()

			snakeRequest := getIndividualBoardStateForSnake(state, s1State, snakeStates, ruleset)
			requestBody := serialiseSnakeRequest(snakeRequest)
			t.Log(string(requestBody))

			test.RequireJSONMatchesFixture(t, fmt.Sprintf("testdata/snake_request_body_%s.json", gt), string(requestBody))
		})
	}
}