mirror of https://github.com/cbeuw/Cloak
POST to /admin/users/{UID} will POST json string directly, rather than being encoded in FormData
This commit is contained in:
parent
a3cd5a4029
commit
1ec11c175a
|
|
@ -92,13 +92,8 @@ func (ar *APIRouter) writeUserInfoHlr(w http.ResponseWriter, r *http.Request) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
jsonUinfo := r.FormValue("UserInfo")
|
|
||||||
if jsonUinfo == "" {
|
|
||||||
http.Error(w, "UserInfo cannot be empty", http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
var uinfo UserInfo
|
var uinfo UserInfo
|
||||||
err = json.Unmarshal([]byte(jsonUinfo), &uinfo)
|
err = json.NewDecoder(r.Body).Decode(&uinfo)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||||
return
|
return
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,47 @@
|
||||||
|
package usermanager
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/base64"
|
||||||
|
"encoding/json"
|
||||||
|
"io/ioutil"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
var mockUIDb64 = base64.StdEncoding.EncodeToString(mockUID)
|
||||||
|
|
||||||
|
func makeRouter(t *testing.T) (router *APIRouter, cleaner func()) {
|
||||||
|
var tmpDB, _ = ioutil.TempFile("", "ck_user_info")
|
||||||
|
cleaner = func() { os.Remove(tmpDB.Name()) }
|
||||||
|
mgr, err := MakeLocalManager(tmpDB.Name(), mockWorldState)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
router = APIRouterOf(mgr)
|
||||||
|
return router, cleaner
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestWriteUserInfoHlr(t *testing.T) {
|
||||||
|
router, cleaner := makeRouter(t)
|
||||||
|
defer cleaner()
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
|
||||||
|
marshalled, err := json.Marshal(mockUserInfo)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
req, err := http.NewRequest("POST", "/admin/users/"+mockUIDb64, bytes.NewBuffer(marshalled))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
router.ServeHTTP(rr, req)
|
||||||
|
|
||||||
|
if status := rr.Code; status != http.StatusCreated {
|
||||||
|
t.Errorf("handler returned wrong status code: got %v with body %v, want %v",
|
||||||
|
status, rr.Body, http.StatusCreated)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -25,28 +25,29 @@ var mockUserInfo = UserInfo{
|
||||||
ExpiryTime: 100,
|
ExpiryTime: 100,
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLocalManager_WriteUserInfo(t *testing.T) {
|
func makeManager(t *testing.T) (mgr *localManager, cleaner func()) {
|
||||||
var tmpDB, _ = ioutil.TempFile("", "ck_user_info")
|
var tmpDB, _ = ioutil.TempFile("", "ck_user_info")
|
||||||
defer os.Remove(tmpDB.Name())
|
cleaner = func() { os.Remove(tmpDB.Name()) }
|
||||||
mgr, err := MakeLocalManager(tmpDB.Name(), mockWorldState)
|
mgr, err := MakeLocalManager(tmpDB.Name(), mockWorldState)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
return mgr, cleaner
|
||||||
|
}
|
||||||
|
|
||||||
err = mgr.WriteUserInfo(mockUserInfo)
|
func TestLocalManager_WriteUserInfo(t *testing.T) {
|
||||||
|
mgr, cleaner := makeManager(t)
|
||||||
|
defer cleaner()
|
||||||
|
|
||||||
|
err := mgr.WriteUserInfo(mockUserInfo)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Error(err)
|
t.Error(err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLocalManager_GetUserInfo(t *testing.T) {
|
func TestLocalManager_GetUserInfo(t *testing.T) {
|
||||||
var tmpDB, _ = ioutil.TempFile("", "ck_user_info")
|
mgr, cleaner := makeManager(t)
|
||||||
defer os.Remove(tmpDB.Name())
|
defer cleaner()
|
||||||
mgr, err := MakeLocalManager(tmpDB.Name(), mockWorldState)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
t.Run("simple fetch", func(t *testing.T) {
|
t.Run("simple fetch", func(t *testing.T) {
|
||||||
_ = mgr.WriteUserInfo(mockUserInfo)
|
_ = mgr.WriteUserInfo(mockUserInfo)
|
||||||
|
|
@ -64,7 +65,7 @@ func TestLocalManager_GetUserInfo(t *testing.T) {
|
||||||
updatedUserInfo := mockUserInfo
|
updatedUserInfo := mockUserInfo
|
||||||
updatedUserInfo.SessionsCap = mockUserInfo.SessionsCap + 1
|
updatedUserInfo.SessionsCap = mockUserInfo.SessionsCap + 1
|
||||||
|
|
||||||
err = mgr.WriteUserInfo(updatedUserInfo)
|
err := mgr.WriteUserInfo(updatedUserInfo)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Error(err)
|
t.Error(err)
|
||||||
}
|
}
|
||||||
|
|
@ -87,15 +88,11 @@ func TestLocalManager_GetUserInfo(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLocalManager_DeleteUser(t *testing.T) {
|
func TestLocalManager_DeleteUser(t *testing.T) {
|
||||||
var tmpDB, _ = ioutil.TempFile("", "ck_user_info")
|
mgr, cleaner := makeManager(t)
|
||||||
defer os.Remove(tmpDB.Name())
|
defer cleaner()
|
||||||
mgr, err := MakeLocalManager(tmpDB.Name(), mockWorldState)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_ = mgr.WriteUserInfo(mockUserInfo)
|
_ = mgr.WriteUserInfo(mockUserInfo)
|
||||||
err = mgr.DeleteUser(mockUID)
|
err := mgr.DeleteUser(mockUID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Error(err)
|
t.Error(err)
|
||||||
}
|
}
|
||||||
|
|
@ -169,12 +166,8 @@ func TestLocalManager_AuthenticateUser(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLocalManager_AuthoriseNewSession(t *testing.T) {
|
func TestLocalManager_AuthoriseNewSession(t *testing.T) {
|
||||||
var tmpDB, _ = ioutil.TempFile("", "ck_user_info")
|
mgr, cleaner := makeManager(t)
|
||||||
defer os.Remove(tmpDB.Name())
|
defer cleaner()
|
||||||
mgr, err := MakeLocalManager(tmpDB.Name(), mockWorldState)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
t.Run("normal auth", func(t *testing.T) {
|
t.Run("normal auth", func(t *testing.T) {
|
||||||
_ = mgr.WriteUserInfo(validUserInfo)
|
_ = mgr.WriteUserInfo(validUserInfo)
|
||||||
|
|
@ -212,12 +205,9 @@ func TestLocalManager_AuthoriseNewSession(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLocalManager_UploadStatus(t *testing.T) {
|
func TestLocalManager_UploadStatus(t *testing.T) {
|
||||||
var tmpDB, _ = ioutil.TempFile("", "ck_user_info")
|
mgr, cleaner := makeManager(t)
|
||||||
defer os.Remove(tmpDB.Name())
|
defer cleaner()
|
||||||
mgr, err := MakeLocalManager(tmpDB.Name(), mockWorldState)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
t.Run("simple update", func(t *testing.T) {
|
t.Run("simple update", func(t *testing.T) {
|
||||||
_ = mgr.WriteUserInfo(validUserInfo)
|
_ = mgr.WriteUserInfo(validUserInfo)
|
||||||
|
|
||||||
|
|
@ -318,12 +308,8 @@ func TestLocalManager_UploadStatus(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLocalManager_ListAllUsers(t *testing.T) {
|
func TestLocalManager_ListAllUsers(t *testing.T) {
|
||||||
var tmpDB, _ = ioutil.TempFile("", "ck_user_info")
|
mgr, cleaner := makeManager(t)
|
||||||
defer os.Remove(tmpDB.Name())
|
defer cleaner()
|
||||||
mgr, err := MakeLocalManager(tmpDB.Name(), mockWorldState)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
var wg sync.WaitGroup
|
var wg sync.WaitGroup
|
||||||
var users []UserInfo
|
var users []UserInfo
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue