wings/api/handlers_server_test.go

package api

import (
	"bytes"
	"encoding/json"
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/gin-gonic/gin"
	"github.com/pterodactyl/wings/control"
	"github.com/stretchr/testify/assert"
)

func TestHandleGetServers(t *testing.T) {
	router := gin.New()
	req, _ := http.NewRequest("GET", "/servers", nil)

	router.GET("/servers", handleGetServers)

	t.Run("returns an empty json array when no servers are configured", func(t *testing.T) {
		rec := httptest.NewRecorder()
		router.ServeHTTP(rec, req)
		body, err := ioutil.ReadAll(rec.Body)
		assert.Equal(t, http.StatusOK, rec.Code)
		assert.Nil(t, err)
		assert.Equal(t, "[]\n", string(body))
	})

	t.Run("returns an array of servers", func(t *testing.T) {

	})
}

var testServer = control.ServerStruct{
	ID: "id1",
}

func TestHandlePostServer(t *testing.T) {
	// We need to decide how deep we want to go with testing here
	// Should we just verify it works in general or test validation and
	// minimal required config options as well?
	// This will also vary depending on the Environment the server runs in
}

func TestHandleGetServer(t *testing.T) {
	router := gin.New()
	router.GET("/servers/:server", handleGetServer)

	control.CreateServer(&testServer)

	t.Run("returns not found when the server doesn't exist", func(t *testing.T) {
		req, _ := http.NewRequest("GET", "/servers/id0", nil)
		rec := httptest.NewRecorder()
		router.ServeHTTP(rec, req)
		assert.Equal(t, http.StatusNotFound, rec.Code)
	})

	t.Run("returns the server as json", func(t *testing.T) {
		req, _ := http.NewRequest("GET", "/servers/id1", nil)
		rec := httptest.NewRecorder()
		router.ServeHTTP(rec, req)
		assert.Equal(t, http.StatusOK, rec.Code)
	})
}

func TestHandlePatchServer(t *testing.T) {

}

func TestHandleDeleteServer(t *testing.T) {
	router := gin.New()
	router.DELETE("/servers/:server", handleDeleteServer)

	control.CreateServer(&testServer)

	t.Run("returns not found when the server doesn't exist", func(t *testing.T) {
		req, _ := http.NewRequest("DELETE", "/servers/id0", nil)
		rec := httptest.NewRecorder()
		router.ServeHTTP(rec, req)
		assert.Equal(t, http.StatusNotFound, rec.Code)
	})

	t.Run("deletes the server", func(t *testing.T) {
		req, _ := http.NewRequest("DELETE", "/servers/id1", nil)
		rec := httptest.NewRecorder()
		router.ServeHTTP(rec, req)
		assert.Equal(t, http.StatusOK, rec.Code)
	})
}

func bodyToJSON(body *bytes.Buffer, v interface{}) error {
	reqBody, err := ioutil.ReadAll(body)
	if err != nil {
		return err
	}
	err = json.Unmarshal(reqBody, v)
	if err != nil {
		return err
	}
	return nil
}