mirror of
https://github.com/gravitl/netmaker.git
synced 2025-09-11 23:54:22 +08:00
Delete User Test
This commit is contained in:
parent
3a702ab3f5
commit
97119f86f1
3 changed files with 75 additions and 1 deletions
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 == mongo.ErrNoDocuments {
|
||||
return false, err
|
||||
return false, nil
|
||||
}
|
||||
return false, err
|
||||
fmt.Println(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)
|
||||
})
|
||||
}
|
Loading…
Add table
Reference in a new issue