72 lines
1.8 KiB
Go
72 lines
1.8 KiB
Go
package apis
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/assert"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestInitAuthEndpoints_authSignin(t *testing.T) {
|
|
_, err := InitUser("admin", "password")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
r := gin.Default()
|
|
SetupRouter(r, nil)
|
|
|
|
reqBody := loginRequest{Username: "admin", Password: "password"}
|
|
strReqBody, _ := json.Marshal(reqBody)
|
|
w := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("POST", "/gui/auth/signin", strings.NewReader(string(strReqBody)))
|
|
r.ServeHTTP(w, req)
|
|
|
|
assert.Equal(t, 200, w.Code)
|
|
assert.JSONEq(t, `{"valid": true}`, w.Body.String())
|
|
setCookie := w.Header().Get("Set-Cookie")
|
|
assert.True(t, strings.Contains(setCookie, "CLORTHO_AUTH="))
|
|
}
|
|
|
|
func TestInitAuthEndpoints_authSignout(t *testing.T) {
|
|
_, err := InitUser("admin", "admin")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
r := gin.Default()
|
|
SetupRouter(r, nil)
|
|
|
|
reqBody := loginRequest{Username: "admin", Password: "admin"}
|
|
strReqBody, _ := json.Marshal(reqBody)
|
|
w := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("POST", "/gui/auth/signout", strings.NewReader(string(strReqBody)))
|
|
r.ServeHTTP(w, req)
|
|
|
|
assert.Equal(t, 200, w.Code)
|
|
assert.JSONEq(t, `{}`, w.Body.String())
|
|
setCookie := w.Header().Get("Set-Cookie")
|
|
assert.True(t, strings.Contains(setCookie, "CLORTHO_AUTH="))
|
|
}
|
|
|
|
func TestInitAuthEndpoints_getMe(t *testing.T) {
|
|
user, err := InitUser("admin", "password")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
r := gin.Default()
|
|
SetupRouter(r, MockAuthMiddleware(*user))
|
|
|
|
w := httptest.NewRecorder()
|
|
req, _ := http.NewRequest("GET", "/gui/auth/me", nil)
|
|
r.ServeHTTP(w, req)
|
|
|
|
strUser, _ := json.Marshal(user)
|
|
assert.Equal(t, 200, w.Code)
|
|
assert.JSONEq(t, fmt.Sprintf(`{"loggedIn": true, "user": %s}`, strUser), w.Body.String())
|
|
}
|