104 lines
2.5 KiB
Go
104 lines
2.5 KiB
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
|
|
}
|