add missing GetUserInternal and GetUsers tests

This commit is contained in:
Matthew R Kasun 2021-10-16 11:47:57 -04:00
parent 67aa4cb565
commit 2f24acbaf1
2 changed files with 43 additions and 27 deletions

View file

@ -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()

View file

@ -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
}