mirror of
https://github.com/gravitl/netmaker.git
synced 2025-02-25 16:44:01 +08:00
add missing GetUserInternal and GetUsers tests
This commit is contained in:
parent
67aa4cb565
commit
2f24acbaf1
2 changed files with 43 additions and 27 deletions
|
@ -153,6 +153,49 @@ func TestGetUser(t *testing.T) {
|
|||
})
|
||||
}
|
||||
|
||||
func TestGetUserInternal(t *testing.T) {
|
||||
database.InitializeDatabase()
|
||||
deleteAllUsers()
|
||||
t.Run("NonExistantUser", func(t *testing.T) {
|
||||
admin, err := GetUserInternal("admin")
|
||||
assert.EqualError(t, err, "could not find any records")
|
||||
assert.Equal(t, "", admin.UserName)
|
||||
})
|
||||
t.Run("UserExisits", func(t *testing.T) {
|
||||
user := models.User{"admin", "password", nil, true}
|
||||
CreateUser(user)
|
||||
admin, err := GetUserInternal("admin")
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, user.UserName, admin.UserName)
|
||||
})
|
||||
}
|
||||
|
||||
func TestGetUsers(t *testing.T) {
|
||||
database.InitializeDatabase()
|
||||
deleteAllUsers()
|
||||
t.Run("NonExistantUser", func(t *testing.T) {
|
||||
admin, err := 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 := models.User{"admin", "password", nil, true}
|
||||
CreateUser(user)
|
||||
admins, err := GetUsers()
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, user.UserName, admins[0].UserName)
|
||||
})
|
||||
t.Run("MulipleUsers", func(t *testing.T) {
|
||||
user := models.User{"user", "password", nil, true}
|
||||
CreateUser(user)
|
||||
admins, err := GetUsers()
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "admin", admins[0].UserName)
|
||||
assert.Equal(t, user.UserName, admins[1].UserName)
|
||||
})
|
||||
|
||||
}
|
||||
|
||||
func TestUpdateUser(t *testing.T) {
|
||||
database.InitializeDatabase()
|
||||
deleteAllUsers()
|
||||
|
|
|
@ -1,27 +0,0 @@
|
|||
package tests
|
||||
|
||||
import "github.com/gravitl/netmaker/models"
|
||||
|
||||
func DeleteAllNetworks() {
|
||||
deleteAllNodes()
|
||||
nets, _ := models.GetNetworks()
|
||||
for _, net := range nets {
|
||||
DeleteNetwork(net.NetID)
|
||||
}
|
||||
}
|
||||
|
||||
func CreateNet() {
|
||||
var network models.Network
|
||||
network.NetID = "skynet"
|
||||
network.AddressRange = "10.0.0.1/24"
|
||||
network.DisplayName = "mynetwork"
|
||||
_, err := GetNetwork("skynet")
|
||||
if err != nil {
|
||||
CreateNetwork(network)
|
||||
}
|
||||
}
|
||||
|
||||
func GetNet() models.Network {
|
||||
network, _ := GetNetwork("skynet")
|
||||
return network
|
||||
}
|
Loading…
Reference in a new issue