mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-06 01:07:41 +08:00
Delete User Test
This commit is contained in:
17
controllers/config/environments/dev.yaml
Normal file
17
controllers/config/environments/dev.yaml
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
server:
|
||||||
|
host: "localhost"
|
||||||
|
apiport: "8081"
|
||||||
|
grpcport: "50051"
|
||||||
|
masterkey: "secretkey"
|
||||||
|
allowedorigin: "*"
|
||||||
|
restbackend: true
|
||||||
|
agentbackend: true
|
||||||
|
defaultnetname: "default"
|
||||||
|
defaultnetrange: "10.10.10.0/24"
|
||||||
|
createdefault: true
|
||||||
|
mongoconn:
|
||||||
|
user: "mongoadmin"
|
||||||
|
pass: "mongopass"
|
||||||
|
host: "localhost"
|
||||||
|
port: "27017"
|
||||||
|
opts: '/?authSource=admin'
|
@@ -195,8 +195,9 @@ func HasAdmin() (bool, error) {
|
|||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if err == mongo.ErrNoDocuments {
|
if err == mongo.ErrNoDocuments {
|
||||||
return false, err
|
return false, nil
|
||||||
}
|
}
|
||||||
|
return false, err
|
||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
}
|
}
|
||||||
return true, err
|
return true, err
|
||||||
|
56
controllers/userHttpController_test.go
Normal file
56
controllers/userHttpController_test.go
Normal file
@@ -0,0 +1,56 @@
|
|||||||
|
package controller
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/gravitl/netmaker/models"
|
||||||
|
"github.com/gravitl/netmaker/mongoconn"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestMain(m *testing.M) {
|
||||||
|
mongoconn.ConnectDatabase()
|
||||||
|
//var waitgroup sync.WaitGroup
|
||||||
|
//waitgroup.Add(1)
|
||||||
|
//go controller.HandleRESTRequests(&waitgroup)
|
||||||
|
var gconf models.GlobalConfig
|
||||||
|
gconf.ServerGRPC = "localhost:8081"
|
||||||
|
gconf.PortGRPC = "50051"
|
||||||
|
//err := SetGlobalConfig(gconf)
|
||||||
|
collection := mongoconn.Client.Database("netmaker").Collection("config")
|
||||||
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||||
|
defer cancel()
|
||||||
|
//create, _, err := functions.GetGlobalConfig()
|
||||||
|
_, err := collection.InsertOne(ctx, gconf)
|
||||||
|
if err != nil {
|
||||||
|
panic("could not create config store")
|
||||||
|
}
|
||||||
|
|
||||||
|
//wait for http server to start
|
||||||
|
//time.Sleep(time.Second * 1)
|
||||||
|
os.Exit(m.Run())
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDeleteUser(t *testing.T) {
|
||||||
|
hasadmin, err := HasAdmin()
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
if !hasadmin {
|
||||||
|
//if !adminExists() {
|
||||||
|
user := models.User{"admin", "admin", true}
|
||||||
|
_, err := CreateUser(user)
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
}
|
||||||
|
t.Run("ExistingUser", func(t *testing.T) {
|
||||||
|
deleted, err := DeleteUser("admin")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.True(t, deleted)
|
||||||
|
})
|
||||||
|
t.Run("NonExistantUser", func(t *testing.T) {
|
||||||
|
deleted, err := DeleteUser("admin")
|
||||||
|
assert.Nil(t, err, err)
|
||||||
|
assert.False(t, deleted)
|
||||||
|
})
|
||||||
|
}
|
Reference in New Issue
Block a user