Add test for /spawn

Required small refactor
This commit is contained in:
Marc Di Luzio 2020-06-02 17:59:33 +01:00
parent 68d117e0d8
commit b152861222
6 changed files with 43 additions and 14 deletions

1
go.mod
View file

@ -5,4 +5,5 @@ go 1.14
require ( require (
github.com/google/uuid v1.1.1 github.com/google/uuid v1.1.1
github.com/gorilla/mux v1.7.4 github.com/gorilla/mux v1.7.4
github.com/stretchr/testify v1.6.0
) )

10
go.sum
View file

@ -1,4 +1,14 @@
github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/google/uuid v1.1.1 h1:Gkbcsh/GbpXz7lPftLA3P6TYMwjCLYm83jiFQZF/3gY= github.com/google/uuid v1.1.1 h1:Gkbcsh/GbpXz7lPftLA3P6TYMwjCLYm83jiFQZF/3gY=
github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/gorilla/mux v1.7.4 h1:VuZ8uybHlWmqV03+zRzdwKL4tUnIp1MAQtp1mIFE1bc= github.com/gorilla/mux v1.7.4 h1:VuZ8uybHlWmqV03+zRzdwKL4tUnIp1MAQtp1mIFE1bc=
github.com/gorilla/mux v1.7.4/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB71So= github.com/gorilla/mux v1.7.4/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB71So=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/testify v1.6.0 h1:jlIyCplCJFULU/01vCkhKuTyc3OorI3bJFuw6obfgho=
github.com/stretchr/testify v1.6.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

View file

@ -43,7 +43,7 @@ func NewAccountant(dataPath string) *Accountant {
} }
// RegisterAccount adds an account to the set of internal accounts // RegisterAccount adds an account to the set of internal accounts
func (a *Accountant) RegisterAccount(acc Account) (Account, error) { func (a *Accountant) RegisterAccount(name string) (acc Account, err error) {
// Set the account ID to a new UUID // Set the account ID to a new UUID
acc.Id = uuid.New() acc.Id = uuid.New()
@ -60,7 +60,7 @@ func (a *Accountant) RegisterAccount(acc Account) (Account, error) {
// Simply add the account to the map // Simply add the account to the map
a.Accounts[acc.Id] = acc a.Accounts[acc.Id] = acc
return acc, nil return
} }
// path returns the full path to the data file // path returns the full path to the data file

View file

@ -22,8 +22,7 @@ func TestAccountant_RegisterAccount(t *testing.T) {
// Start by making two accounts // Start by making two accounts
namea := "one" namea := "one"
a := Account{Name: namea} acca, err := accountant.RegisterAccount(namea)
acca, err := accountant.RegisterAccount(a)
if err != nil { if err != nil {
t.Error(err) t.Error(err)
} else if acca.Name != namea { } else if acca.Name != namea {
@ -31,8 +30,7 @@ func TestAccountant_RegisterAccount(t *testing.T) {
} }
nameb := "two" nameb := "two"
b := Account{Name: nameb} accb, err := accountant.RegisterAccount(nameb)
accb, err := accountant.RegisterAccount(b)
if err != nil { if err != nil {
t.Error(err) t.Error(err)
} else if accb.Name != nameb { } else if accb.Name != nameb {
@ -45,7 +43,7 @@ func TestAccountant_RegisterAccount(t *testing.T) {
} }
// Verify another request gets rejected // Verify another request gets rejected
_, err = accountant.RegisterAccount(a) _, err = accountant.RegisterAccount(namea)
if err == nil { if err == nil {
t.Error("Duplicate account name did not produce error") t.Error("Duplicate account name did not produce error")
} }
@ -58,8 +56,7 @@ func TestAccountant_LoadSave(t *testing.T) {
} }
name := "one" name := "one"
a := Account{Name: name} a, err := accountant.RegisterAccount(name)
a, err := accountant.RegisterAccount(a)
if err != nil { if err != nil {
t.Error(err) t.Error(err)
} }
@ -101,8 +98,7 @@ func TestAccountant_AssignPrimary(t *testing.T) {
} }
name := "one" name := "one"
a := Account{Name: name} a, err := accountant.RegisterAccount(name)
a, err := accountant.RegisterAccount(a)
if err != nil { if err != nil {
t.Error(err) t.Error(err)
} }

View file

@ -6,7 +6,6 @@ import (
"net/http" "net/http"
"github.com/google/uuid" "github.com/google/uuid"
"github.com/mdiluz/rove/pkg/accounts"
"github.com/mdiluz/rove/pkg/game" "github.com/mdiluz/rove/pkg/game"
"github.com/mdiluz/rove/pkg/version" "github.com/mdiluz/rove/pkg/version"
) )
@ -125,8 +124,7 @@ func (s *Server) HandleRegister(w http.ResponseWriter, r *http.Request) {
fmt.Printf("\tdata: %+v\n", data) fmt.Printf("\tdata: %+v\n", data)
// Register the account with the server // Register the account with the server
acc := accounts.Account{Name: data.Name} acc, err := s.accountant.RegisterAccount(data.Name)
acc, err := s.accountant.RegisterAccount(acc)
// If we didn't fail, respond with the account ID string // If we didn't fail, respond with the account ID string
if err == nil { if err == nil {

View file

@ -6,6 +6,8 @@ import (
"net/http" "net/http"
"net/http/httptest" "net/http/httptest"
"testing" "testing"
"github.com/stretchr/testify/assert"
) )
func TestHandleStatus(t *testing.T) { func TestHandleStatus(t *testing.T) {
@ -47,3 +49,25 @@ func TestHandleRegister(t *testing.T) {
t.Errorf("got false for /register") t.Errorf("got false for /register")
} }
} }
func TestHandleSpawn(t *testing.T) {
s := NewServer()
a, err := s.accountant.RegisterAccount("test")
assert.NoError(t, err, "Error registering account")
data := SpawnData{BasicAccountData: BasicAccountData{Id: a.Id.String()}}
b, err := json.Marshal(data)
assert.NoError(t, err, "Error marshalling data")
request, _ := http.NewRequest(http.MethodPost, "/spawn", bytes.NewReader(b))
response := httptest.NewRecorder()
s.HandleSpawn(response, request)
var status SpawnResponse
json.NewDecoder(response.Body).Decode(&status)
if status.Success != true {
t.Errorf("got false for /spawn")
}
}