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) {
|
func TestUpdateUser(t *testing.T) {
|
||||||
database.InitializeDatabase()
|
database.InitializeDatabase()
|
||||||
deleteAllUsers()
|
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