mirror of
https://github.com/gravitl/netmaker.git
synced 2025-09-13 00:24:31 +08:00
Group Tests Complete
This commit is contained in:
parent
a34684ed82
commit
ab35ee2c99
1 changed files with 542 additions and 48 deletions
590
group_test.go
590
group_test.go
|
@ -2,6 +2,7 @@ package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"io/ioutil"
|
||||||
"net/http"
|
"net/http"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
|
@ -12,8 +13,42 @@ import (
|
||||||
|
|
||||||
var Groups []models.Group
|
var Groups []models.Group
|
||||||
|
|
||||||
|
func TestCreateGroup(t *testing.T) {
|
||||||
|
group := models.Group{}
|
||||||
|
group.NameID = "skynet"
|
||||||
|
group.AddressRange = "10.71.0.0/16"
|
||||||
|
t.Run("CreateGroup", func(t *testing.T) {
|
||||||
|
response, err := api(t, group, http.MethodPost, "http://localhost:8081/api/groups", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("InvalidToken", func(t *testing.T) {
|
||||||
|
response, err := api(t, group, http.MethodPost, "http://localhost:8081/api/groups", "badkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
||||||
|
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
||||||
|
})
|
||||||
|
t.Run("BadName", func(t *testing.T) {
|
||||||
|
//issue #42
|
||||||
|
t.Skip()
|
||||||
|
})
|
||||||
|
t.Run("BadAddress", func(t *testing.T) {
|
||||||
|
//issue #42
|
||||||
|
t.Skip()
|
||||||
|
})
|
||||||
|
t.Run("DuplicateGroup", func(t *testing.T) {
|
||||||
|
//issue #42
|
||||||
|
t.Skip()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
func TestGetGroups(t *testing.T) {
|
func TestGetGroups(t *testing.T) {
|
||||||
t.Run("GetGroupsValidToken", func(t *testing.T) {
|
t.Run("ValidToken", func(t *testing.T) {
|
||||||
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups", "secretkey")
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups", "secretkey")
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
defer response.Body.Close()
|
defer response.Body.Close()
|
||||||
|
@ -21,7 +56,7 @@ func TestGetGroups(t *testing.T) {
|
||||||
err = json.NewDecoder(response.Body).Decode(&Groups)
|
err = json.NewDecoder(response.Body).Decode(&Groups)
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
})
|
})
|
||||||
t.Run("GetGroupsInvalidToken", func(t *testing.T) {
|
t.Run("InvalidToken", func(t *testing.T) {
|
||||||
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups", "badkey")
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups", "badkey")
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
defer response.Body.Close()
|
defer response.Body.Close()
|
||||||
|
@ -34,42 +69,8 @@ func TestGetGroups(t *testing.T) {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestCreateGroup(t *testing.T) {
|
|
||||||
group := models.Group{}
|
|
||||||
group.NameID = "skynet"
|
|
||||||
group.AddressRange = "10.71.0.0/16"
|
|
||||||
t.Run("CreateGroup", func(t *testing.T) {
|
|
||||||
response, err := api(t, group, http.MethodPost, "http://localhost:8081/api/groups", "secretkey")
|
|
||||||
assert.Nil(t, err, err)
|
|
||||||
assert.Equal(t, http.StatusOK, response.StatusCode)
|
|
||||||
})
|
|
||||||
t.Run("CreateGroupInvalidToken", func(t *testing.T) {
|
|
||||||
response, err := api(t, group, http.MethodPost, "http://localhost:8081/api/groups", "badkey")
|
|
||||||
assert.Nil(t, err, err)
|
|
||||||
assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
|
|
||||||
defer response.Body.Close()
|
|
||||||
var message models.ErrorResponse
|
|
||||||
err = json.NewDecoder(response.Body).Decode(&message)
|
|
||||||
assert.Nil(t, err, err)
|
|
||||||
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
|
||||||
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
|
||||||
})
|
|
||||||
t.Run("CreateGroupBadName", func(t *testing.T) {
|
|
||||||
//issue #42
|
|
||||||
t.Skip()
|
|
||||||
})
|
|
||||||
t.Run("CreateGroupBadAddress", func(t *testing.T) {
|
|
||||||
//issue #42
|
|
||||||
t.Skip()
|
|
||||||
})
|
|
||||||
t.Run("CreateGroupDuplicateGroup", func(t *testing.T) {
|
|
||||||
//issue #42
|
|
||||||
t.Skip()
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestGetGroup(t *testing.T) {
|
func TestGetGroup(t *testing.T) {
|
||||||
t.Run("GetGroupValidToken", func(t *testing.T) {
|
t.Run("ValidToken", func(t *testing.T) {
|
||||||
var group models.Group
|
var group models.Group
|
||||||
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/skynet", "secretkey")
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
|
@ -77,9 +78,9 @@ func TestGetGroup(t *testing.T) {
|
||||||
assert.Equal(t, http.StatusOK, response.StatusCode)
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
err = json.NewDecoder(response.Body).Decode(&group)
|
err = json.NewDecoder(response.Body).Decode(&group)
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
assert.Equal(t, group.DisplayName, "skynet")
|
assert.Equal(t, "skynet", group.DisplayName)
|
||||||
})
|
})
|
||||||
t.Run("GetGroupInvalidToken", func(t *testing.T) {
|
t.Run("InvalidToken", func(t *testing.T) {
|
||||||
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/skynet", "badkey")
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/skynet", "badkey")
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
defer response.Body.Close()
|
defer response.Body.Close()
|
||||||
|
@ -90,7 +91,7 @@ func TestGetGroup(t *testing.T) {
|
||||||
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
||||||
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
||||||
})
|
})
|
||||||
t.Run("GetGroupInvalidGroup", func(t *testing.T) {
|
t.Run("InvalidGroup", func(t *testing.T) {
|
||||||
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/badgroup", "secretkey")
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/badgroup", "secretkey")
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
defer response.Body.Close()
|
defer response.Body.Close()
|
||||||
|
@ -103,7 +104,7 @@ func TestGetGroup(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestGetGroupNodeNumber(t *testing.T) {
|
func TestGetGroupNodeNumber(t *testing.T) {
|
||||||
t.Run("GetNumberNodesValidKey", func(t *testing.T) {
|
t.Run("ValidKey", func(t *testing.T) {
|
||||||
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/skynet/numnodes", "secretkey")
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/skynet/numnodes", "secretkey")
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
defer response.Body.Close()
|
defer response.Body.Close()
|
||||||
|
@ -113,7 +114,7 @@ func TestGetGroupNodeNumber(t *testing.T) {
|
||||||
//assert.Equal(t, "W1R3: This group does not exist.", message.Message)
|
//assert.Equal(t, "W1R3: This group does not exist.", message.Message)
|
||||||
assert.Equal(t, http.StatusOK, response.StatusCode)
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
})
|
})
|
||||||
t.Run("GetNumberNodesInvalidKey", func(t *testing.T) {
|
t.Run("InvalidKey", func(t *testing.T) {
|
||||||
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/skynet/numnodes", "badkey")
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/skynet/numnodes", "badkey")
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
defer response.Body.Close()
|
defer response.Body.Close()
|
||||||
|
@ -124,7 +125,7 @@ func TestGetGroupNodeNumber(t *testing.T) {
|
||||||
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
||||||
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
||||||
})
|
})
|
||||||
t.Run("GetNumNodesBadGroup", func(t *testing.T) {
|
t.Run("BadGroup", func(t *testing.T) {
|
||||||
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/badgroup/numnodes", "secretkey")
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/badgroup/numnodes", "secretkey")
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
defer response.Body.Close()
|
defer response.Body.Close()
|
||||||
|
@ -137,7 +138,7 @@ func TestGetGroupNodeNumber(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDeleteGroup(t *testing.T) {
|
func TestDeleteGroup(t *testing.T) {
|
||||||
t.Run("DeleteGroupInvalidKey", func(t *testing.T) {
|
t.Run("InvalidKey", func(t *testing.T) {
|
||||||
response, err := api(t, "", http.MethodDelete, "http://localhost:8081/api/groups/skynet", "badkey")
|
response, err := api(t, "", http.MethodDelete, "http://localhost:8081/api/groups/skynet", "badkey")
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
defer response.Body.Close()
|
defer response.Body.Close()
|
||||||
|
@ -148,7 +149,7 @@ func TestDeleteGroup(t *testing.T) {
|
||||||
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
||||||
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
||||||
})
|
})
|
||||||
t.Run("DeleteGroupValidKey", func(t *testing.T) {
|
t.Run("ValidKey", func(t *testing.T) {
|
||||||
response, err := api(t, "", http.MethodDelete, "http://localhost:8081/api/groups/skynet", "secretkey")
|
response, err := api(t, "", http.MethodDelete, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
defer response.Body.Close()
|
defer response.Body.Close()
|
||||||
|
@ -159,7 +160,7 @@ func TestDeleteGroup(t *testing.T) {
|
||||||
assert.Equal(t, int64(1), message.DeletedCount)
|
assert.Equal(t, int64(1), message.DeletedCount)
|
||||||
|
|
||||||
})
|
})
|
||||||
t.Run("DeleteGroupBadGroup", func(t *testing.T) {
|
t.Run("BadGroup", func(t *testing.T) {
|
||||||
response, err := api(t, "", http.MethodDelete, "http://localhost:8081/api/groups/badgroup", "secretkey")
|
response, err := api(t, "", http.MethodDelete, "http://localhost:8081/api/groups/badgroup", "secretkey")
|
||||||
assert.Nil(t, err, err)
|
assert.Nil(t, err, err)
|
||||||
defer response.Body.Close()
|
defer response.Body.Close()
|
||||||
|
@ -169,8 +170,501 @@ func TestDeleteGroup(t *testing.T) {
|
||||||
assert.Equal(t, "W1R3: This group does not exist.", message.Message)
|
assert.Equal(t, "W1R3: This group does not exist.", message.Message)
|
||||||
assert.Equal(t, http.StatusNotFound, response.StatusCode)
|
assert.Equal(t, http.StatusNotFound, response.StatusCode)
|
||||||
})
|
})
|
||||||
t.Run("DeleteGroupNodesExist", func(t *testing.T) {
|
t.Run("NodesExist", func(t *testing.T) {
|
||||||
t.Skip()
|
t.Skip()
|
||||||
})
|
})
|
||||||
|
//Create Group for follow-on tests
|
||||||
|
createGroup(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestCreateAccessKey(t *testing.T) {
|
||||||
|
key := models.AccessKey{}
|
||||||
|
key.Name = "skynet"
|
||||||
|
key.Uses = 10
|
||||||
|
t.Run("MultiUse", func(t *testing.T) {
|
||||||
|
response, err := api(t, key, http.MethodPost, "http://localhost:8081/api/groups/skynet/keys", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
message, err := ioutil.ReadAll(response.Body)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.NotNil(t, message, message)
|
||||||
|
returnedkey := getKey(t, key.Name)
|
||||||
|
assert.Equal(t, key.Name, returnedkey.Name)
|
||||||
|
assert.Equal(t, key.Uses, returnedkey.Uses)
|
||||||
|
})
|
||||||
|
deleteKey(t, "skynet", "skynet")
|
||||||
|
t.Run("ZeroUse", func(t *testing.T) {
|
||||||
|
//t.Skip()
|
||||||
|
key.Uses = 0
|
||||||
|
response, err := api(t, key, http.MethodPost, "http://localhost:8081/api/groups/skynet/keys", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
message, err := ioutil.ReadAll(response.Body)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.NotNil(t, message, message)
|
||||||
|
returnedkey := getKey(t, key.Name)
|
||||||
|
assert.Equal(t, key.Name, returnedkey.Name)
|
||||||
|
assert.Equal(t, 1, returnedkey.Uses)
|
||||||
|
})
|
||||||
|
t.Run("DuplicateAccessKey", func(t *testing.T) {
|
||||||
|
//t.Skip()
|
||||||
|
//this will fail
|
||||||
|
response, err := api(t, key, http.MethodPost, "http://localhost:8081/api/groups/skynet/keys", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnprocessableEntity, response.StatusCode)
|
||||||
|
deleteKey(t, key.Name, "skynet")
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("InvalidToken", func(t *testing.T) {
|
||||||
|
response, err := api(t, key, http.MethodPost, "http://localhost:8081/api/groups/skynet/keys", "badkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
||||||
|
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
||||||
|
})
|
||||||
|
t.Run("BadGroup", func(t *testing.T) {
|
||||||
|
response, err := api(t, key, http.MethodPost, "http://localhost:8081/api/groups/badgroup/keys", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, "W1R3: This group does not exist.", message.Message)
|
||||||
|
assert.Equal(t, http.StatusNotFound, response.StatusCode)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDeleteKey(t *testing.T) {
|
||||||
|
t.Run("KeyValid", func(t *testing.T) {
|
||||||
|
//fails -- deletecount not returned
|
||||||
|
response, err := api(t, "", http.MethodDelete, "http://localhost:8081/api/groups/skynet/keys/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var message mongo.DeleteResult
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
assert.Equal(t, int64(1), message.DeletedCount)
|
||||||
|
})
|
||||||
|
t.Run("InValidKey", func(t *testing.T) {
|
||||||
|
//fails -- status message not returned
|
||||||
|
response, err := api(t, "", http.MethodDelete, "http://localhost:8081/api/groups/skynet/keys/badkey", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, "W1R3: This key does not exist.", message.Message)
|
||||||
|
assert.Equal(t, http.StatusNotFound, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("KeyInValidGroup", func(t *testing.T) {
|
||||||
|
response, err := api(t, "", http.MethodDelete, "http://localhost:8081/api/groups/badgroup/keys/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, "W1R3: This group does not exist.", message.Message)
|
||||||
|
assert.Equal(t, http.StatusNotFound, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("InvalidCredentials", func(t *testing.T) {
|
||||||
|
response, err := api(t, "", http.MethodDelete, "http://localhost:8081/api/groups/skynet/keys/skynet", "badkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
||||||
|
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetKeys(t *testing.T) {
|
||||||
|
createKey(t)
|
||||||
|
t.Run("Valid", func(t *testing.T) {
|
||||||
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/skynet/keys", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var keys []models.AccessKey
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&keys)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
})
|
||||||
|
//deletekeys
|
||||||
|
t.Run("InvalidGroup", func(t *testing.T) {
|
||||||
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/badgroup/keys", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, "W1R3: This group does not exist.", message.Message)
|
||||||
|
assert.Equal(t, http.StatusNotFound, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("InvalidCredentials", func(t *testing.T) {
|
||||||
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/skynet/keys", "badkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
||||||
|
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUpdateGroup(t *testing.T) {
|
||||||
|
var returnedGroup models.Group
|
||||||
|
t.Run("UpdateNameID", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
NameID string
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.NameID = "wirecat"
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, group.NameID, returnedGroup.NameID)
|
||||||
|
})
|
||||||
|
t.Run("NameIDInvalidCredentials", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
NameID string
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.NameID = "wirecat"
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "badkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, message.Code)
|
||||||
|
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
|
||||||
|
assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("InvalidGroup", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
NameID string
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.NameID = "wirecat"
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/badgroup", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusNotFound, message.Code)
|
||||||
|
assert.Equal(t, "W1R3: This group does not exist.", message.Message)
|
||||||
|
assert.Equal(t, http.StatusNotFound, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("UpdateNameIDTooLong", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
NameID string
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.NameID = "wirecat-skynet"
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnprocessableEntity, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("UpdateAddress", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
AddressRange string
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.AddressRange = "10.0.0.1/24"
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, group.AddressRange, returnedGroup.AddressRange)
|
||||||
|
})
|
||||||
|
t.Run("UpdateAddressInvalid", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
AddressRange string
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.AddressRange = "10.0.0.1/36"
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnprocessableEntity, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("UpdateDisplayName", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DisplayName string
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.DisplayName = "wirecat"
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, group.DisplayName, returnedGroup.DisplayName)
|
||||||
|
|
||||||
|
})
|
||||||
|
t.Run("UpdateDisplayNameInvalidName", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DisplayName string
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
//create name that is longer than 100 chars
|
||||||
|
name := ""
|
||||||
|
for i := 0; i < 101; i++ {
|
||||||
|
name = name + "a"
|
||||||
|
}
|
||||||
|
group.DisplayName = name
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnprocessableEntity, message.Code)
|
||||||
|
assert.Equal(t, "W1R3: Field validation for 'DisplayName' failed.", message.Message)
|
||||||
|
assert.Equal(t, http.StatusUnprocessableEntity, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("UpdateInterface", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DefaultInterface string
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.DefaultInterface = "netmaker"
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, group.DefaultInterface, returnedGroup.DefaultInterface)
|
||||||
|
|
||||||
|
})
|
||||||
|
t.Run("UpdateListenPort", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DefaultListenPort int32
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.DefaultListenPort = 6000
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, group.DefaultListenPort, returnedGroup.DefaultListenPort)
|
||||||
|
})
|
||||||
|
t.Run("UpdateListenPortInvalidPort", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DefaultListenPort int32
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.DefaultListenPort = 1023
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnprocessableEntity, message.Code)
|
||||||
|
assert.Equal(t, "W1R3: Field validation for 'DefaultListenPort' failed.", message.Message)
|
||||||
|
assert.Equal(t, http.StatusUnprocessableEntity, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("UpdatePostUP", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DefaultPostUp string
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.DefaultPostUp = "sudo wg add-conf wc-netmaker /etc/wireguard/peers/conf"
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, group.DefaultPostUp, returnedGroup.DefaultPostUp)
|
||||||
|
})
|
||||||
|
t.Run("UpdatePreUP", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DefaultPreUp string
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.DefaultPreUp = "test string"
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, group.DefaultPreUp, returnedGroup.DefaultPreUp)
|
||||||
|
})
|
||||||
|
t.Run("UpdateKeepAlive", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DefaultKeepalive int32
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.DefaultKeepalive = 60
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, group.DefaultKeepalive, returnedGroup.DefaultKeepalive)
|
||||||
|
})
|
||||||
|
t.Run("UpdateKeepAliveTooBig", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DefaultKeepAlive int32
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.DefaultKeepAlive = 1001
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnprocessableEntity, message.Code)
|
||||||
|
assert.Equal(t, "W1R3: Field validation for 'DefaultKeepAlive' failed.", message.Message)
|
||||||
|
assert.Equal(t, http.StatusUnprocessableEntity, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("UpdateSaveConfig", func(t *testing.T) {
|
||||||
|
//causes panic
|
||||||
|
t.Skip()
|
||||||
|
type Group struct {
|
||||||
|
DefaultSaveConfig *bool
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
value := false
|
||||||
|
group.DefaultSaveConfig = &value
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, *group.DefaultSaveConfig, *returnedGroup.DefaultSaveConfig)
|
||||||
|
})
|
||||||
|
t.Run("UpdateManualSignUP", func(t *testing.T) {
|
||||||
|
t.Skip()
|
||||||
|
type Group struct {
|
||||||
|
AllowManualSignUp *bool
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
value := true
|
||||||
|
group.AllowManualSignUp = &value
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, *group.AllowManualSignUp, *returnedGroup.AllowManualSignUp)
|
||||||
|
})
|
||||||
|
t.Run("DefaultCheckInterval", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DefaultCheckInInterval int32
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.DefaultCheckInInterval = 6000
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, group.DefaultCheckInInterval, returnedGroup.DefaultCheckInInterval)
|
||||||
|
})
|
||||||
|
t.Run("DefaultCheckIntervalTooBig", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DefaultCheckInInterval int32
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.DefaultCheckInInterval = 100001
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
var message models.ErrorResponse
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusUnprocessableEntity, message.Code)
|
||||||
|
assert.Equal(t, "W1R3: Field validation for 'DefaultCheckInInterval' failed.", message.Message)
|
||||||
|
assert.Equal(t, http.StatusUnprocessableEntity, response.StatusCode)
|
||||||
|
})
|
||||||
|
t.Run("MultipleFields", func(t *testing.T) {
|
||||||
|
type Group struct {
|
||||||
|
DisplayName string
|
||||||
|
DefaultListenPort int32
|
||||||
|
}
|
||||||
|
var group Group
|
||||||
|
group.DefaultListenPort = 7777
|
||||||
|
group.DisplayName = "multi"
|
||||||
|
response, err := api(t, group, http.MethodPut, "http://localhost:8081/api/groups/skynet", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&returnedGroup)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, group.DisplayName, returnedGroup.DisplayName)
|
||||||
|
assert.Equal(t, group.DefaultListenPort, returnedGroup.DefaultListenPort)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func getKey(t *testing.T, name string) models.AccessKey {
|
||||||
|
response, err := api(t, "", http.MethodGet, "http://localhost:8081/api/groups/skynet/keys", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
var keys []models.AccessKey
|
||||||
|
err = json.NewDecoder(response.Body).Decode(&keys)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
for _, key := range keys {
|
||||||
|
if key.Name == name {
|
||||||
|
return key
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return models.AccessKey{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func deleteKey(t *testing.T, key, group string) {
|
||||||
|
response, err := api(t, "", http.MethodDelete, "http://localhost:8081/api/groups/"+group+"/keys/"+key, "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
//api does not return Deleted Count at this time
|
||||||
|
//defer response.Body.Close()
|
||||||
|
//var message mongo.DeleteResult
|
||||||
|
//err = json.NewDecoder(response.Body).Decode(&message)
|
||||||
|
//assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
//assert.Equal(t, int64(1), message.DeletedCount)
|
||||||
|
}
|
||||||
|
|
||||||
|
func createKey(t *testing.T) {
|
||||||
|
key := models.AccessKey{}
|
||||||
|
key.Name = "skynet"
|
||||||
|
key.Uses = 10
|
||||||
|
response, err := api(t, key, http.MethodPost, "http://localhost:8081/api/groups/skynet/keys", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
|
defer response.Body.Close()
|
||||||
|
message, err := ioutil.ReadAll(response.Body)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.NotNil(t, message, message)
|
||||||
|
}
|
||||||
|
|
||||||
|
func createGroup(t *testing.T) {
|
||||||
|
group := models.Group{}
|
||||||
|
group.NameID = "skynet"
|
||||||
|
group.AddressRange = "10.71.0.0/16"
|
||||||
|
response, err := api(t, group, http.MethodPost, "http://localhost:8081/api/groups", "secretkey")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.Equal(t, http.StatusOK, response.StatusCode)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue