mirror of
https://github.com/gravitl/netmaker.git
synced 2025-09-04 04:04:17 +08:00
Net 1227 v1 (#3061)
* generalise smtp config * copy over smtp vars * env new line * fix master key api access * comment user tests
This commit is contained in:
parent
aa64e50374
commit
d41521607f
3 changed files with 450 additions and 451 deletions
|
@ -81,49 +81,59 @@ func upgradeHost(w http.ResponseWriter, r *http.Request) {
|
|||
func getHosts(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
currentHosts := []models.Host{}
|
||||
username := r.Header.Get("user")
|
||||
user, err := logic.GetUser(username)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
userPlatformRole, err := logic.GetRole(user.PlatformRoleID)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
respHostsMap := make(map[string]struct{})
|
||||
if !userPlatformRole.FullAccess {
|
||||
nodes, err := logic.GetAllNodes()
|
||||
if err != nil {
|
||||
logger.Log(0, "error fetching all nodes info: ", err.Error())
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
filteredNodes := logic.GetFilteredNodesByUserAccess(*user, nodes)
|
||||
if len(filteredNodes) > 0 {
|
||||
currentHostsMap, err := logic.GetHostsMap()
|
||||
if err != nil {
|
||||
logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
for _, node := range filteredNodes {
|
||||
if _, ok := respHostsMap[node.HostID.String()]; ok {
|
||||
continue
|
||||
}
|
||||
if host, ok := currentHostsMap[node.HostID.String()]; ok {
|
||||
currentHosts = append(currentHosts, host)
|
||||
respHostsMap[host.ID.String()] = struct{}{}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
} else {
|
||||
var err error
|
||||
if r.Header.Get("ismaster") == "yes" {
|
||||
currentHosts, err = logic.GetAllHosts()
|
||||
if err != nil {
|
||||
logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
} else {
|
||||
username := r.Header.Get("user")
|
||||
user, err := logic.GetUser(username)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
userPlatformRole, err := logic.GetRole(user.PlatformRoleID)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
respHostsMap := make(map[string]struct{})
|
||||
if !userPlatformRole.FullAccess {
|
||||
nodes, err := logic.GetAllNodes()
|
||||
if err != nil {
|
||||
logger.Log(0, "error fetching all nodes info: ", err.Error())
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
filteredNodes := logic.GetFilteredNodesByUserAccess(*user, nodes)
|
||||
if len(filteredNodes) > 0 {
|
||||
currentHostsMap, err := logic.GetHostsMap()
|
||||
if err != nil {
|
||||
logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
for _, node := range filteredNodes {
|
||||
if _, ok := respHostsMap[node.HostID.String()]; ok {
|
||||
continue
|
||||
}
|
||||
if host, ok := currentHostsMap[node.HostID.String()]; ok {
|
||||
currentHosts = append(currentHosts, host)
|
||||
respHostsMap[host.ID.String()] = struct{}{}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
} else {
|
||||
currentHosts, err = logic.GetAllHosts()
|
||||
if err != nil {
|
||||
logger.Log(0, r.Header.Get("user"), "failed to fetch hosts: ", err.Error())
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
apiHosts := logic.GetAllHostsAPI(currentHosts[:])
|
||||
|
|
|
@ -268,56 +268,59 @@ func getNetworkNodes(w http.ResponseWriter, r *http.Request) {
|
|||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
username := r.Header.Get("user")
|
||||
user, err := logic.GetUser(username)
|
||||
if err != nil {
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
userPlatformRole, err := logic.GetRole(user.PlatformRoleID)
|
||||
if err != nil {
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
filteredNodes := []models.Node{}
|
||||
if !userPlatformRole.FullAccess {
|
||||
nodesMap := make(map[string]struct{})
|
||||
networkRoles := user.NetworkRoles[models.NetworkID(networkName)]
|
||||
for networkRoleID := range networkRoles {
|
||||
userPermTemplate, err := logic.GetRole(networkRoleID)
|
||||
if err != nil {
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
if userPermTemplate.FullAccess {
|
||||
break
|
||||
}
|
||||
if rsrcPerms, ok := userPermTemplate.NetworkLevelAccess[models.RemoteAccessGwRsrc]; ok {
|
||||
if _, ok := rsrcPerms[models.AllRemoteAccessGwRsrcID]; ok {
|
||||
for _, node := range nodes {
|
||||
if _, ok := nodesMap[node.ID.String()]; ok {
|
||||
continue
|
||||
if r.Header.Get("ismaster") != "yes" {
|
||||
username := r.Header.Get("user")
|
||||
user, err := logic.GetUser(username)
|
||||
if err != nil {
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
userPlatformRole, err := logic.GetRole(user.PlatformRoleID)
|
||||
if err != nil {
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
|
||||
if !userPlatformRole.FullAccess {
|
||||
nodesMap := make(map[string]struct{})
|
||||
networkRoles := user.NetworkRoles[models.NetworkID(networkName)]
|
||||
for networkRoleID := range networkRoles {
|
||||
userPermTemplate, err := logic.GetRole(networkRoleID)
|
||||
if err != nil {
|
||||
logic.ReturnErrorResponse(w, r, logic.FormatError(err, "internal"))
|
||||
return
|
||||
}
|
||||
if userPermTemplate.FullAccess {
|
||||
break
|
||||
}
|
||||
if rsrcPerms, ok := userPermTemplate.NetworkLevelAccess[models.RemoteAccessGwRsrc]; ok {
|
||||
if _, ok := rsrcPerms[models.AllRemoteAccessGwRsrcID]; ok {
|
||||
for _, node := range nodes {
|
||||
if _, ok := nodesMap[node.ID.String()]; ok {
|
||||
continue
|
||||
}
|
||||
if node.IsIngressGateway {
|
||||
nodesMap[node.ID.String()] = struct{}{}
|
||||
filteredNodes = append(filteredNodes, node)
|
||||
}
|
||||
}
|
||||
if node.IsIngressGateway {
|
||||
nodesMap[node.ID.String()] = struct{}{}
|
||||
filteredNodes = append(filteredNodes, node)
|
||||
}
|
||||
}
|
||||
} else {
|
||||
for gwID, scope := range rsrcPerms {
|
||||
if _, ok := nodesMap[gwID.String()]; ok {
|
||||
continue
|
||||
}
|
||||
if scope.Read {
|
||||
gwNode, err := logic.GetNodeByID(gwID.String())
|
||||
if err == nil && gwNode.IsIngressGateway {
|
||||
filteredNodes = append(filteredNodes, gwNode)
|
||||
} else {
|
||||
for gwID, scope := range rsrcPerms {
|
||||
if _, ok := nodesMap[gwID.String()]; ok {
|
||||
continue
|
||||
}
|
||||
if scope.Read {
|
||||
gwNode, err := logic.GetNodeByID(gwID.String())
|
||||
if err == nil && gwNode.IsIngressGateway {
|
||||
filteredNodes = append(filteredNodes, gwNode)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
if len(filteredNodes) > 0 {
|
||||
|
@ -348,18 +351,19 @@ func getAllNodes(w http.ResponseWriter, r *http.Request) {
|
|||
return
|
||||
}
|
||||
username := r.Header.Get("user")
|
||||
user, err := logic.GetUser(username)
|
||||
if err != nil {
|
||||
return
|
||||
if r.Header.Get("ismaster") == "no" {
|
||||
user, err := logic.GetUser(username)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
userPlatformRole, err := logic.GetRole(user.PlatformRoleID)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if !userPlatformRole.FullAccess {
|
||||
nodes = logic.GetFilteredNodesByUserAccess(*user, nodes)
|
||||
}
|
||||
}
|
||||
userPlatformRole, err := logic.GetRole(user.PlatformRoleID)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
if !userPlatformRole.FullAccess {
|
||||
nodes = logic.GetFilteredNodesByUserAccess(*user, nodes)
|
||||
}
|
||||
|
||||
// return all the nodes in JSON/API format
|
||||
apiNodes := logic.GetAllNodesAPI(nodes[:])
|
||||
logger.Log(3, r.Header.Get("user"), "fetched all nodes they have access to")
|
||||
|
|
|
@ -1,376 +1,361 @@
|
|||
package controller
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
// TODO: Need Update Tests for New User Mgmt
|
||||
// func deleteAllUsers(t *testing.T) {
|
||||
// t.Helper()
|
||||
// users, _ := logic.GetUsers()
|
||||
// for _, user := range users {
|
||||
// if _, err := logic.DeleteUser(user.UserName); err != nil {
|
||||
// t.Fatal(err)
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
|
||||
"github.com/go-jose/go-jose/v3/json"
|
||||
"github.com/gorilla/mux"
|
||||
// func TestGetUserNoHashedPassword(t *testing.T) {
|
||||
// // prepare existing user base
|
||||
// user := models.User{UserName: "freddie", Password: "password"}
|
||||
// haveOnlyOneUser(t, user)
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
// // prepare request
|
||||
// rec, req := prepareUserRequest(t, models.User{}, user.UserName)
|
||||
|
||||
"github.com/gravitl/netmaker/logic"
|
||||
"github.com/gravitl/netmaker/models"
|
||||
)
|
||||
// // test response
|
||||
// getUser(rec, req)
|
||||
// assertUserNameButNoPassword(t, rec.Body, user.UserName)
|
||||
// }
|
||||
|
||||
func deleteAllUsers(t *testing.T) {
|
||||
t.Helper()
|
||||
users, _ := logic.GetUsers()
|
||||
for _, user := range users {
|
||||
if _, err := logic.DeleteUser(user.UserName); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
// func TestCreateAdminNoHashedPassword(t *testing.T) {
|
||||
// // prepare existing user base
|
||||
// deleteAllUsers(t)
|
||||
|
||||
func TestGetUserNoHashedPassword(t *testing.T) {
|
||||
// prepare existing user base
|
||||
user := models.User{UserName: "freddie", Password: "password"}
|
||||
haveOnlyOneUser(t, user)
|
||||
// // prepare request
|
||||
// user := models.User{UserName: "jonathan", Password: "password"}
|
||||
// rec, req := prepareUserRequest(t, user, "")
|
||||
|
||||
// prepare request
|
||||
rec, req := prepareUserRequest(t, models.User{}, user.UserName)
|
||||
// // test response
|
||||
// createSuperAdmin(rec, req)
|
||||
// assertUserNameButNoPassword(t, rec.Body, user.UserName)
|
||||
// }
|
||||
|
||||
// test response
|
||||
getUser(rec, req)
|
||||
assertUserNameButNoPassword(t, rec.Body, user.UserName)
|
||||
}
|
||||
// func prepareUserRequest(t *testing.T, userForBody models.User, userNameForParam string) (*httptest.ResponseRecorder, *http.Request) {
|
||||
// bits, err := json.Marshal(userForBody)
|
||||
// assert.Nil(t, err)
|
||||
// body := bytes.NewReader(bits)
|
||||
// rec := httptest.NewRecorder()
|
||||
// req := httptest.NewRequest("ANY", "https://example.com", body) // only the body matters here
|
||||
// req = mux.SetURLVars(req, map[string]string{"username": userNameForParam})
|
||||
// req.Header.Set("user", userForBody.UserName)
|
||||
// return rec, req
|
||||
// }
|
||||
|
||||
func TestCreateAdminNoHashedPassword(t *testing.T) {
|
||||
// prepare existing user base
|
||||
deleteAllUsers(t)
|
||||
// func haveOnlyOneUser(t *testing.T, user models.User) {
|
||||
// deleteAllUsers(t)
|
||||
// var err error
|
||||
// if user.PlatformRoleID == models.SuperAdminRole {
|
||||
// err = logic.CreateSuperAdmin(&user)
|
||||
// } else {
|
||||
// err = logic.CreateUser(&user)
|
||||
// }
|
||||
// assert.Nil(t, err)
|
||||
// }
|
||||
|
||||
// prepare request
|
||||
user := models.User{UserName: "jonathan", Password: "password"}
|
||||
rec, req := prepareUserRequest(t, user, "")
|
||||
// func assertUserNameButNoPassword(t *testing.T, r io.Reader, userName string) {
|
||||
// var resp models.User
|
||||
// err := json.NewDecoder(r).Decode(&resp)
|
||||
// assert.Nil(t, err)
|
||||
// assert.Equal(t, userName, resp.UserName)
|
||||
// assert.Empty(t, resp.Password)
|
||||
// }
|
||||
|
||||
// test response
|
||||
createSuperAdmin(rec, req)
|
||||
assertUserNameButNoPassword(t, rec.Body, user.UserName)
|
||||
}
|
||||
|
||||
func prepareUserRequest(t *testing.T, userForBody models.User, userNameForParam string) (*httptest.ResponseRecorder, *http.Request) {
|
||||
bits, err := json.Marshal(userForBody)
|
||||
assert.Nil(t, err)
|
||||
body := bytes.NewReader(bits)
|
||||
rec := httptest.NewRecorder()
|
||||
req := httptest.NewRequest("ANY", "https://example.com", body) // only the body matters here
|
||||
req = mux.SetURLVars(req, map[string]string{"username": userNameForParam})
|
||||
req.Header.Set("user", userForBody.UserName)
|
||||
return rec, req
|
||||
}
|
||||
|
||||
func haveOnlyOneUser(t *testing.T, user models.User) {
|
||||
deleteAllUsers(t)
|
||||
var err error
|
||||
if user.PlatformRoleID == models.SuperAdminRole {
|
||||
err = logic.CreateSuperAdmin(&user)
|
||||
} else {
|
||||
err = logic.CreateUser(&user)
|
||||
}
|
||||
assert.Nil(t, err)
|
||||
}
|
||||
|
||||
func assertUserNameButNoPassword(t *testing.T, r io.Reader, userName string) {
|
||||
var resp models.User
|
||||
err := json.NewDecoder(r).Decode(&resp)
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, userName, resp.UserName)
|
||||
assert.Empty(t, resp.Password)
|
||||
}
|
||||
|
||||
func TestHasSuperAdmin(t *testing.T) {
|
||||
// delete all current users
|
||||
users, _ := logic.GetUsers()
|
||||
for _, user := range users {
|
||||
success, err := logic.DeleteUser(user.UserName)
|
||||
assert.Nil(t, err)
|
||||
assert.True(t, success)
|
||||
}
|
||||
t.Run("NoUser", func(t *testing.T) {
|
||||
found, err := logic.HasSuperAdmin()
|
||||
assert.Nil(t, err)
|
||||
assert.False(t, found)
|
||||
})
|
||||
t.Run("No superadmin user", func(t *testing.T) {
|
||||
var user = models.User{UserName: "nosuperadmin", Password: "password"}
|
||||
err := logic.CreateUser(&user)
|
||||
assert.Nil(t, err)
|
||||
found, err := logic.HasSuperAdmin()
|
||||
assert.Nil(t, err)
|
||||
assert.False(t, found)
|
||||
})
|
||||
t.Run("superadmin user", func(t *testing.T) {
|
||||
var user = models.User{UserName: "superadmin", Password: "password", PlatformRoleID: models.SuperAdminRole}
|
||||
err := logic.CreateUser(&user)
|
||||
assert.Nil(t, err)
|
||||
found, err := logic.HasSuperAdmin()
|
||||
assert.Nil(t, err)
|
||||
assert.True(t, found)
|
||||
})
|
||||
t.Run("multiple superadmins", func(t *testing.T) {
|
||||
var user = models.User{UserName: "superadmin1", Password: "password", PlatformRoleID: models.SuperAdminRole}
|
||||
err := logic.CreateUser(&user)
|
||||
assert.Nil(t, err)
|
||||
found, err := logic.HasSuperAdmin()
|
||||
assert.Nil(t, err)
|
||||
assert.True(t, found)
|
||||
})
|
||||
}
|
||||
|
||||
func TestCreateUser(t *testing.T) {
|
||||
deleteAllUsers(t)
|
||||
user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
t.Run("NoUser", func(t *testing.T) {
|
||||
err := logic.CreateUser(&user)
|
||||
assert.Nil(t, err)
|
||||
})
|
||||
t.Run("UserExists", func(t *testing.T) {
|
||||
err := logic.CreateUser(&user)
|
||||
assert.NotNil(t, err)
|
||||
assert.EqualError(t, err, "user exists")
|
||||
})
|
||||
}
|
||||
|
||||
func TestCreateSuperAdmin(t *testing.T) {
|
||||
deleteAllUsers(t)
|
||||
logic.ClearSuperUserCache()
|
||||
var user models.User
|
||||
t.Run("NoSuperAdmin", func(t *testing.T) {
|
||||
user.UserName = "admin"
|
||||
user.Password = "password"
|
||||
err := logic.CreateSuperAdmin(&user)
|
||||
assert.Nil(t, err)
|
||||
})
|
||||
t.Run("SuperAdminExists", func(t *testing.T) {
|
||||
user.UserName = "admin2"
|
||||
user.Password = "password1"
|
||||
err := logic.CreateSuperAdmin(&user)
|
||||
assert.EqualError(t, err, "superadmin user already exists")
|
||||
})
|
||||
}
|
||||
|
||||
func TestDeleteUser(t *testing.T) {
|
||||
deleteAllUsers(t)
|
||||
t.Run("NonExistent User", func(t *testing.T) {
|
||||
deleted, err := logic.DeleteUser("admin")
|
||||
assert.EqualError(t, err, "user does not exist")
|
||||
assert.False(t, deleted)
|
||||
})
|
||||
t.Run("Existing User", func(t *testing.T) {
|
||||
user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
if err := logic.CreateUser(&user); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
deleted, err := logic.DeleteUser("admin")
|
||||
assert.Nil(t, err)
|
||||
assert.True(t, deleted)
|
||||
})
|
||||
}
|
||||
|
||||
func TestValidateUser(t *testing.T) {
|
||||
var user models.User
|
||||
t.Run("Valid Create", func(t *testing.T) {
|
||||
user.UserName = "admin"
|
||||
user.Password = "validpass"
|
||||
err := logic.ValidateUser(&user)
|
||||
assert.Nil(t, err)
|
||||
})
|
||||
t.Run("Valid Update", func(t *testing.T) {
|
||||
user.UserName = "admin"
|
||||
user.Password = "password"
|
||||
err := logic.ValidateUser(&user)
|
||||
assert.Nil(t, err)
|
||||
})
|
||||
t.Run("Invalid UserName", func(t *testing.T) {
|
||||
t.Skip()
|
||||
user.UserName = "*invalid"
|
||||
err := logic.ValidateUser(&user)
|
||||
assert.Error(t, err)
|
||||
// assert.Contains(t, err.Error(), "Field validation for 'UserName' failed")
|
||||
})
|
||||
t.Run("Short UserName", func(t *testing.T) {
|
||||
t.Skip()
|
||||
user.UserName = "1"
|
||||
err := logic.ValidateUser(&user)
|
||||
assert.NotNil(t, err)
|
||||
// assert.Contains(t, err.Error(), "Field validation for 'UserName' failed")
|
||||
})
|
||||
t.Run("Empty UserName", func(t *testing.T) {
|
||||
t.Skip()
|
||||
user.UserName = ""
|
||||
err := logic.ValidateUser(&user)
|
||||
assert.EqualError(t, err, "some string")
|
||||
// assert.Contains(t, err.Error(), "Field validation for 'UserName' failed")
|
||||
})
|
||||
t.Run("EmptyPassword", func(t *testing.T) {
|
||||
user.Password = ""
|
||||
err := logic.ValidateUser(&user)
|
||||
assert.EqualError(t, err, "Key: 'User.Password' Error:Field validation for 'Password' failed on the 'required' tag")
|
||||
})
|
||||
t.Run("ShortPassword", func(t *testing.T) {
|
||||
user.Password = "123"
|
||||
err := logic.ValidateUser(&user)
|
||||
assert.EqualError(t, err, "Key: 'User.Password' Error:Field validation for 'Password' failed on the 'min' tag")
|
||||
})
|
||||
}
|
||||
|
||||
func TestGetUser(t *testing.T) {
|
||||
deleteAllUsers(t)
|
||||
|
||||
user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
|
||||
t.Run("NonExistantUser", func(t *testing.T) {
|
||||
admin, err := logic.GetUser("admin")
|
||||
assert.EqualError(t, err, "could not find any records")
|
||||
assert.Equal(t, "", admin.UserName)
|
||||
})
|
||||
t.Run("UserExisits", func(t *testing.T) {
|
||||
if err := logic.CreateUser(&user); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
admin, err := logic.GetUser("admin")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, user.UserName, admin.UserName)
|
||||
})
|
||||
}
|
||||
|
||||
func TestGetUsers(t *testing.T) {
|
||||
deleteAllUsers(t)
|
||||
|
||||
adminUser := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
|
||||
t.Run("NonExistantUser", func(t *testing.T) {
|
||||
admin, err := logic.GetUsers()
|
||||
assert.EqualError(t, err, "could not find any records")
|
||||
assert.Equal(t, []models.ReturnUser(nil), admin)
|
||||
})
|
||||
t.Run("UserExisits", func(t *testing.T) {
|
||||
user.UserName = "anotheruser"
|
||||
if err := logic.CreateUser(&adminUser); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
admins, err := logic.GetUsers()
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, adminUser.UserName, admins[0].UserName)
|
||||
})
|
||||
t.Run("MulipleUsers", func(t *testing.T) {
|
||||
if err := logic.CreateUser(&user); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
admins, err := logic.GetUsers()
|
||||
assert.Nil(t, err)
|
||||
for _, u := range admins {
|
||||
if u.UserName == "admin" {
|
||||
assert.Equal(t, true, u.IsAdmin)
|
||||
} else {
|
||||
assert.Equal(t, user.UserName, u.UserName)
|
||||
assert.Equal(t, user.PlatformRoleID, u.PlatformRoleID)
|
||||
}
|
||||
}
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
func TestUpdateUser(t *testing.T) {
|
||||
deleteAllUsers(t)
|
||||
user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
newuser := models.User{UserName: "hello", Password: "world", PlatformRoleID: models.AdminRole}
|
||||
t.Run("NonExistantUser", func(t *testing.T) {
|
||||
admin, err := logic.UpdateUser(&newuser, &user)
|
||||
assert.EqualError(t, err, "could not find any records")
|
||||
assert.Equal(t, "", admin.UserName)
|
||||
})
|
||||
|
||||
t.Run("UserExists", func(t *testing.T) {
|
||||
if err := logic.CreateUser(&user); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
admin, err := logic.UpdateUser(&newuser, &user)
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, newuser.UserName, admin.UserName)
|
||||
})
|
||||
}
|
||||
|
||||
// func TestValidateUserToken(t *testing.T) {
|
||||
// t.Run("EmptyToken", func(t *testing.T) {
|
||||
// err := ValidateUserToken("", "", false)
|
||||
// assert.NotNil(t, err)
|
||||
// assert.Equal(t, "Missing Auth Token.", err.Error())
|
||||
// })
|
||||
// t.Run("InvalidToken", func(t *testing.T) {
|
||||
// err := ValidateUserToken("Bearer: badtoken", "", false)
|
||||
// assert.NotNil(t, err)
|
||||
// assert.Equal(t, "Error Verifying Auth Token", err.Error())
|
||||
// })
|
||||
// t.Run("InvalidUser", func(t *testing.T) {
|
||||
// t.Skip()
|
||||
// err := ValidateUserToken("Bearer: secretkey", "baduser", false)
|
||||
// assert.NotNil(t, err)
|
||||
// assert.Equal(t, "Error Verifying Auth Token", err.Error())
|
||||
// //need authorization
|
||||
// })
|
||||
// t.Run("ValidToken", func(t *testing.T) {
|
||||
// err := ValidateUserToken("Bearer: secretkey", "", true)
|
||||
// func TestHasSuperAdmin(t *testing.T) {
|
||||
// // delete all current users
|
||||
// users, _ := logic.GetUsers()
|
||||
// for _, user := range users {
|
||||
// success, err := logic.DeleteUser(user.UserName)
|
||||
// assert.Nil(t, err)
|
||||
// assert.True(t, success)
|
||||
// }
|
||||
// t.Run("NoUser", func(t *testing.T) {
|
||||
// found, err := logic.HasSuperAdmin()
|
||||
// assert.Nil(t, err)
|
||||
// assert.False(t, found)
|
||||
// })
|
||||
// t.Run("No superadmin user", func(t *testing.T) {
|
||||
// var user = models.User{UserName: "nosuperadmin", Password: "password"}
|
||||
// err := logic.CreateUser(&user)
|
||||
// assert.Nil(t, err)
|
||||
// found, err := logic.HasSuperAdmin()
|
||||
// assert.Nil(t, err)
|
||||
// assert.False(t, found)
|
||||
// })
|
||||
// t.Run("superadmin user", func(t *testing.T) {
|
||||
// var user = models.User{UserName: "superadmin", Password: "password", PlatformRoleID: models.SuperAdminRole}
|
||||
// err := logic.CreateUser(&user)
|
||||
// assert.Nil(t, err)
|
||||
// found, err := logic.HasSuperAdmin()
|
||||
// assert.Nil(t, err)
|
||||
// assert.True(t, found)
|
||||
// })
|
||||
// t.Run("multiple superadmins", func(t *testing.T) {
|
||||
// var user = models.User{UserName: "superadmin1", Password: "password", PlatformRoleID: models.SuperAdminRole}
|
||||
// err := logic.CreateUser(&user)
|
||||
// assert.Nil(t, err)
|
||||
// found, err := logic.HasSuperAdmin()
|
||||
// assert.Nil(t, err)
|
||||
// assert.True(t, found)
|
||||
// })
|
||||
// }
|
||||
|
||||
func TestVerifyAuthRequest(t *testing.T) {
|
||||
deleteAllUsers(t)
|
||||
user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
var authRequest models.UserAuthParams
|
||||
t.Run("EmptyUserName", func(t *testing.T) {
|
||||
authRequest.UserName = ""
|
||||
authRequest.Password = "Password"
|
||||
jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
assert.Equal(t, "", jwt)
|
||||
assert.EqualError(t, err, "username can't be empty")
|
||||
})
|
||||
t.Run("EmptyPassword", func(t *testing.T) {
|
||||
authRequest.UserName = "admin"
|
||||
authRequest.Password = ""
|
||||
jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
assert.Equal(t, "", jwt)
|
||||
assert.EqualError(t, err, "password can't be empty")
|
||||
})
|
||||
t.Run("NonExistantUser", func(t *testing.T) {
|
||||
authRequest.UserName = "admin"
|
||||
authRequest.Password = "password"
|
||||
jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
assert.Equal(t, "", jwt)
|
||||
assert.EqualError(t, err, "incorrect credentials")
|
||||
})
|
||||
t.Run("Non-Admin", func(t *testing.T) {
|
||||
user.PlatformRoleID = models.ServiceUser
|
||||
user.Password = "somepass"
|
||||
user.UserName = "nonadmin"
|
||||
if err := logic.CreateUser(&user); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
authRequest := models.UserAuthParams{UserName: "nonadmin", Password: "somepass"}
|
||||
jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
assert.NotNil(t, jwt)
|
||||
assert.Nil(t, err)
|
||||
})
|
||||
t.Run("WrongPassword", func(t *testing.T) {
|
||||
user := models.User{UserName: "admin", Password: "password"}
|
||||
if err := logic.CreateUser(&user); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
authRequest := models.UserAuthParams{UserName: "admin", Password: "badpass"}
|
||||
jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
assert.Equal(t, "", jwt)
|
||||
assert.EqualError(t, err, "incorrect credentials")
|
||||
})
|
||||
t.Run("Success", func(t *testing.T) {
|
||||
authRequest := models.UserAuthParams{UserName: "admin", Password: "password"}
|
||||
jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
assert.Nil(t, err)
|
||||
assert.NotNil(t, jwt)
|
||||
})
|
||||
}
|
||||
// func TestCreateUser(t *testing.T) {
|
||||
// deleteAllUsers(t)
|
||||
// user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
// t.Run("NoUser", func(t *testing.T) {
|
||||
// err := logic.CreateUser(&user)
|
||||
// assert.Nil(t, err)
|
||||
// })
|
||||
// t.Run("UserExists", func(t *testing.T) {
|
||||
// err := logic.CreateUser(&user)
|
||||
// assert.NotNil(t, err)
|
||||
// assert.EqualError(t, err, "user exists")
|
||||
// })
|
||||
// }
|
||||
|
||||
// func TestCreateSuperAdmin(t *testing.T) {
|
||||
// deleteAllUsers(t)
|
||||
// logic.ClearSuperUserCache()
|
||||
// var user models.User
|
||||
// t.Run("NoSuperAdmin", func(t *testing.T) {
|
||||
// user.UserName = "admin"
|
||||
// user.Password = "password"
|
||||
// err := logic.CreateSuperAdmin(&user)
|
||||
// assert.Nil(t, err)
|
||||
// })
|
||||
// t.Run("SuperAdminExists", func(t *testing.T) {
|
||||
// user.UserName = "admin2"
|
||||
// user.Password = "password1"
|
||||
// err := logic.CreateSuperAdmin(&user)
|
||||
// assert.EqualError(t, err, "superadmin user already exists")
|
||||
// })
|
||||
// }
|
||||
|
||||
// func TestDeleteUser(t *testing.T) {
|
||||
// deleteAllUsers(t)
|
||||
// t.Run("NonExistent User", func(t *testing.T) {
|
||||
// deleted, err := logic.DeleteUser("admin")
|
||||
// assert.EqualError(t, err, "user does not exist")
|
||||
// assert.False(t, deleted)
|
||||
// })
|
||||
// t.Run("Existing User", func(t *testing.T) {
|
||||
// user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
// if err := logic.CreateUser(&user); err != nil {
|
||||
// t.Fatal(err)
|
||||
// }
|
||||
// deleted, err := logic.DeleteUser("admin")
|
||||
// assert.Nil(t, err)
|
||||
// assert.True(t, deleted)
|
||||
// })
|
||||
// }
|
||||
|
||||
// func TestValidateUser(t *testing.T) {
|
||||
// var user models.User
|
||||
// t.Run("Valid Create", func(t *testing.T) {
|
||||
// user.UserName = "admin"
|
||||
// user.Password = "validpass"
|
||||
// err := logic.ValidateUser(&user)
|
||||
// assert.Nil(t, err)
|
||||
// })
|
||||
// t.Run("Valid Update", func(t *testing.T) {
|
||||
// user.UserName = "admin"
|
||||
// user.Password = "password"
|
||||
// err := logic.ValidateUser(&user)
|
||||
// assert.Nil(t, err)
|
||||
// })
|
||||
// t.Run("Invalid UserName", func(t *testing.T) {
|
||||
// t.Skip()
|
||||
// user.UserName = "*invalid"
|
||||
// err := logic.ValidateUser(&user)
|
||||
// assert.Error(t, err)
|
||||
// // assert.Contains(t, err.Error(), "Field validation for 'UserName' failed")
|
||||
// })
|
||||
// t.Run("Short UserName", func(t *testing.T) {
|
||||
// t.Skip()
|
||||
// user.UserName = "1"
|
||||
// err := logic.ValidateUser(&user)
|
||||
// assert.NotNil(t, err)
|
||||
// // assert.Contains(t, err.Error(), "Field validation for 'UserName' failed")
|
||||
// })
|
||||
// t.Run("Empty UserName", func(t *testing.T) {
|
||||
// t.Skip()
|
||||
// user.UserName = ""
|
||||
// err := logic.ValidateUser(&user)
|
||||
// assert.EqualError(t, err, "some string")
|
||||
// // assert.Contains(t, err.Error(), "Field validation for 'UserName' failed")
|
||||
// })
|
||||
// t.Run("EmptyPassword", func(t *testing.T) {
|
||||
// user.Password = ""
|
||||
// err := logic.ValidateUser(&user)
|
||||
// assert.EqualError(t, err, "Key: 'User.Password' Error:Field validation for 'Password' failed on the 'required' tag")
|
||||
// })
|
||||
// t.Run("ShortPassword", func(t *testing.T) {
|
||||
// user.Password = "123"
|
||||
// err := logic.ValidateUser(&user)
|
||||
// assert.EqualError(t, err, "Key: 'User.Password' Error:Field validation for 'Password' failed on the 'min' tag")
|
||||
// })
|
||||
// }
|
||||
|
||||
// func TestGetUser(t *testing.T) {
|
||||
// deleteAllUsers(t)
|
||||
|
||||
// user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
|
||||
// t.Run("NonExistantUser", func(t *testing.T) {
|
||||
// admin, err := logic.GetUser("admin")
|
||||
// assert.EqualError(t, err, "could not find any records")
|
||||
// assert.Equal(t, "", admin.UserName)
|
||||
// })
|
||||
// t.Run("UserExisits", func(t *testing.T) {
|
||||
// if err := logic.CreateUser(&user); err != nil {
|
||||
// t.Error(err)
|
||||
// }
|
||||
// admin, err := logic.GetUser("admin")
|
||||
// assert.Nil(t, err)
|
||||
// assert.Equal(t, user.UserName, admin.UserName)
|
||||
// })
|
||||
// }
|
||||
|
||||
// func TestGetUsers(t *testing.T) {
|
||||
// deleteAllUsers(t)
|
||||
|
||||
// adminUser := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
// user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
|
||||
// t.Run("NonExistantUser", func(t *testing.T) {
|
||||
// admin, err := logic.GetUsers()
|
||||
// assert.EqualError(t, err, "could not find any records")
|
||||
// assert.Equal(t, []models.ReturnUser(nil), admin)
|
||||
// })
|
||||
// t.Run("UserExisits", func(t *testing.T) {
|
||||
// user.UserName = "anotheruser"
|
||||
// if err := logic.CreateUser(&adminUser); err != nil {
|
||||
// t.Error(err)
|
||||
// }
|
||||
// admins, err := logic.GetUsers()
|
||||
// assert.Nil(t, err)
|
||||
// assert.Equal(t, adminUser.UserName, admins[0].UserName)
|
||||
// })
|
||||
// t.Run("MulipleUsers", func(t *testing.T) {
|
||||
// if err := logic.CreateUser(&user); err != nil {
|
||||
// t.Error(err)
|
||||
// }
|
||||
// admins, err := logic.GetUsers()
|
||||
// assert.Nil(t, err)
|
||||
// for _, u := range admins {
|
||||
// if u.UserName == "admin" {
|
||||
// assert.Equal(t, true, u.IsAdmin)
|
||||
// } else {
|
||||
// assert.Equal(t, user.UserName, u.UserName)
|
||||
// assert.Equal(t, user.PlatformRoleID, u.PlatformRoleID)
|
||||
// }
|
||||
// }
|
||||
// })
|
||||
|
||||
// }
|
||||
|
||||
// func TestUpdateUser(t *testing.T) {
|
||||
// deleteAllUsers(t)
|
||||
// user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
// newuser := models.User{UserName: "hello", Password: "world", PlatformRoleID: models.AdminRole}
|
||||
// t.Run("NonExistantUser", func(t *testing.T) {
|
||||
// admin, err := logic.UpdateUser(&newuser, &user)
|
||||
// assert.EqualError(t, err, "could not find any records")
|
||||
// assert.Equal(t, "", admin.UserName)
|
||||
// })
|
||||
|
||||
// t.Run("UserExists", func(t *testing.T) {
|
||||
// if err := logic.CreateUser(&user); err != nil {
|
||||
// t.Error(err)
|
||||
// }
|
||||
// admin, err := logic.UpdateUser(&newuser, &user)
|
||||
// assert.Nil(t, err)
|
||||
// assert.Equal(t, newuser.UserName, admin.UserName)
|
||||
// })
|
||||
// }
|
||||
|
||||
// // func TestValidateUserToken(t *testing.T) {
|
||||
// // t.Run("EmptyToken", func(t *testing.T) {
|
||||
// // err := ValidateUserToken("", "", false)
|
||||
// // assert.NotNil(t, err)
|
||||
// // assert.Equal(t, "Missing Auth Token.", err.Error())
|
||||
// // })
|
||||
// // t.Run("InvalidToken", func(t *testing.T) {
|
||||
// // err := ValidateUserToken("Bearer: badtoken", "", false)
|
||||
// // assert.NotNil(t, err)
|
||||
// // assert.Equal(t, "Error Verifying Auth Token", err.Error())
|
||||
// // })
|
||||
// // t.Run("InvalidUser", func(t *testing.T) {
|
||||
// // t.Skip()
|
||||
// // err := ValidateUserToken("Bearer: secretkey", "baduser", false)
|
||||
// // assert.NotNil(t, err)
|
||||
// // assert.Equal(t, "Error Verifying Auth Token", err.Error())
|
||||
// // //need authorization
|
||||
// // })
|
||||
// // t.Run("ValidToken", func(t *testing.T) {
|
||||
// // err := ValidateUserToken("Bearer: secretkey", "", true)
|
||||
// // assert.Nil(t, err)
|
||||
// // })
|
||||
// // }
|
||||
|
||||
// func TestVerifyAuthRequest(t *testing.T) {
|
||||
// deleteAllUsers(t)
|
||||
// user := models.User{UserName: "admin", Password: "password", PlatformRoleID: models.AdminRole}
|
||||
// var authRequest models.UserAuthParams
|
||||
// t.Run("EmptyUserName", func(t *testing.T) {
|
||||
// authRequest.UserName = ""
|
||||
// authRequest.Password = "Password"
|
||||
// jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
// assert.Equal(t, "", jwt)
|
||||
// assert.EqualError(t, err, "username can't be empty")
|
||||
// })
|
||||
// t.Run("EmptyPassword", func(t *testing.T) {
|
||||
// authRequest.UserName = "admin"
|
||||
// authRequest.Password = ""
|
||||
// jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
// assert.Equal(t, "", jwt)
|
||||
// assert.EqualError(t, err, "password can't be empty")
|
||||
// })
|
||||
// t.Run("NonExistantUser", func(t *testing.T) {
|
||||
// authRequest.UserName = "admin"
|
||||
// authRequest.Password = "password"
|
||||
// jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
// assert.Equal(t, "", jwt)
|
||||
// assert.EqualError(t, err, "incorrect credentials")
|
||||
// })
|
||||
// t.Run("Non-Admin", func(t *testing.T) {
|
||||
// user.PlatformRoleID = models.ServiceUser
|
||||
// user.Password = "somepass"
|
||||
// user.UserName = "nonadmin"
|
||||
// if err := logic.CreateUser(&user); err != nil {
|
||||
// t.Error(err)
|
||||
// }
|
||||
// authRequest := models.UserAuthParams{UserName: "nonadmin", Password: "somepass"}
|
||||
// jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
// assert.NotNil(t, jwt)
|
||||
// assert.Nil(t, err)
|
||||
// })
|
||||
// t.Run("WrongPassword", func(t *testing.T) {
|
||||
// user := models.User{UserName: "admin", Password: "password"}
|
||||
// if err := logic.CreateUser(&user); err != nil {
|
||||
// t.Error(err)
|
||||
// }
|
||||
// authRequest := models.UserAuthParams{UserName: "admin", Password: "badpass"}
|
||||
// jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
// assert.Equal(t, "", jwt)
|
||||
// assert.EqualError(t, err, "incorrect credentials")
|
||||
// })
|
||||
// t.Run("Success", func(t *testing.T) {
|
||||
// authRequest := models.UserAuthParams{UserName: "admin", Password: "password"}
|
||||
// jwt, err := logic.VerifyAuthRequest(authRequest)
|
||||
// assert.Nil(t, err)
|
||||
// assert.NotNil(t, jwt)
|
||||
// })
|
||||
// }
|
||||
|
|
Loading…
Add table
Reference in a new issue