rove/cmd/rove-server/api_test.go

package main

import (
	"fmt"
	"os"
	"testing"

	"github.com/google/uuid"
	"github.com/mdiluz/rove/pkg/game"
	"github.com/mdiluz/rove/pkg/rove"
	"github.com/mdiluz/rove/pkg/server"
	"github.com/stretchr/testify/assert"
)

// To be set by the main function
var serv rove.Server

func TestMain(m *testing.M) {
	s := server.NewServer()
	if err := s.Initialise(true); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

	serv = rove.Server(s.Addr())

	go s.Run()

	fmt.Printf("Test server hosted on %s", serv)
	code := m.Run()

	if err := s.StopAndClose(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

	os.Exit(code)
}

func TestServer_Status(t *testing.T) {
	status, err := serv.Status()
	assert.NoError(t, err)
	assert.True(t, status.Ready)
	assert.NotZero(t, len(status.Version))
}

func TestServer_Register(t *testing.T) {
	d1 := rove.RegisterData{
		Name: uuid.New().String(),
	}
	r1, err := serv.Register(d1)
	assert.NoError(t, err)
	assert.True(t, r1.Success)
	assert.NotZero(t, len(r1.Id))

	d2 := rove.RegisterData{
		Name: uuid.New().String(),
	}
	r2, err := serv.Register(d2)
	assert.NoError(t, err)
	assert.True(t, r2.Success)
	assert.NotZero(t, len(r2.Id))

	r3, err := serv.Register(d1)
	assert.NoError(t, err)
	assert.False(t, r3.Success)
}

func TestServer_Spawn(t *testing.T) {
	d1 := rove.RegisterData{
		Name: uuid.New().String(),
	}
	r1, err := serv.Register(d1)
	assert.NoError(t, err)
	assert.True(t, r1.Success)
	assert.NotZero(t, len(r1.Id))

	s := rove.SpawnData{}
	r2, err := serv.Spawn(r1.Id, s)
	assert.NoError(t, err)
	assert.True(t, r2.Success)
}

func TestServer_Command(t *testing.T) {
	d1 := rove.RegisterData{
		Name: uuid.New().String(),
	}
	r1, err := serv.Register(d1)
	assert.NoError(t, err)
	assert.True(t, r1.Success)
	assert.NotZero(t, len(r1.Id))

	s := rove.SpawnData{}
	r2, err := serv.Spawn(r1.Id, s)
	assert.NoError(t, err)
	assert.True(t, r2.Success)

	c := rove.CommandData{
		Commands: []game.Command{
			{
				Command:  game.CommandMove,
				Bearing:  "N",
				Duration: 1,
			},
		},
	}
	r3, err := serv.Command(r1.Id, c)
	assert.NoError(t, err)
	assert.True(t, r3.Success)
}

func TestServer_Radar(t *testing.T) {
	d1 := rove.RegisterData{
		Name: uuid.New().String(),
	}
	r1, err := serv.Register(d1)
	assert.NoError(t, err)
	assert.True(t, r1.Success)
	assert.NotZero(t, len(r1.Id))

	s := rove.SpawnData{}
	r2, err := serv.Spawn(r1.Id, s)
	assert.NoError(t, err)
	assert.True(t, r2.Success)

	r3, err := serv.Radar(r1.Id)
	assert.NoError(t, err)
	assert.True(t, r3.Success)
}

func TestServer_Rover(t *testing.T) {
	d1 := rove.RegisterData{
		Name: uuid.New().String(),
	}
	r1, err := serv.Register(d1)
	assert.NoError(t, err)
	assert.True(t, r1.Success)
	assert.NotZero(t, len(r1.Id))

	s := rove.SpawnData{}
	r2, err := serv.Spawn(r1.Id, s)
	assert.NoError(t, err)
	assert.True(t, r2.Success)

	r3, err := serv.Rover(r1.Id)
	assert.NoError(t, err)
	assert.True(t, r3.Success)
}