netmaker/logic/pro/networkuser_test.go

package pro

import (
	"os"
	"testing"

	"github.com/google/uuid"
	"github.com/gravitl/netmaker/database"
	"github.com/gravitl/netmaker/models"
	"github.com/gravitl/netmaker/models/promodels"
	"github.com/stretchr/testify/assert"
)

func TestMain(m *testing.M) {
	database.InitializeDatabase()
	defer database.CloseDB()
	os.Exit(m.Run())
}

func TestNetworkUserLogic(t *testing.T) {
	networkUser := promodels.NetworkUser{
		ID: "helloworld",
	}
	network := models.Network{
		NetID:        "skynet",
		AddressRange: "192.168.0.0/24",
	}
	tmpCNode := models.CommonNode{
		ID: uuid.New(),
	}
	tempNode := models.Node{}
	tempNode.CommonNode = tmpCNode
	nodes := []models.Node{
		tempNode,
	}

	clients := []models.ExtClient{
		{
			ClientID: "coolclient",
		},
	}
	AddProNetDefaults(&network)
	t.Run("Net Users initialized successfully", func(t *testing.T) {
		err := InitializeNetworkUsers(network.NetID)
		assert.Nil(t, err)
	})

	t.Run("Error when no network users", func(t *testing.T) {
		user, err := GetNetworkUser(network.NetID, networkUser.ID)
		assert.Nil(t, user)
		assert.NotNil(t, err)
	})

	t.Run("Successful net user create", func(t *testing.T) {
		DeleteNetworkUser(network.NetID, string(networkUser.ID))
		err := CreateNetworkUser(&network, &networkUser)
		assert.Nil(t, err)
		user, err := GetNetworkUser(network.NetID, networkUser.ID)
		assert.NotNil(t, user)
		assert.Nil(t, err)
		assert.Equal(t, 0, user.AccessLevel)
		assert.Equal(t, 0, user.ClientLimit)
	})

	t.Run("Successful net user update", func(t *testing.T) {
		networkUser.AccessLevel = 0
		networkUser.ClientLimit = 1
		err := UpdateNetworkUser(network.NetID, &networkUser)
		assert.Nil(t, err)
		user, err := GetNetworkUser(network.NetID, networkUser.ID)
		assert.NotNil(t, user)
		assert.Nil(t, err)
		assert.Equal(t, 0, user.AccessLevel)
		assert.Equal(t, 1, user.ClientLimit)
	})

	t.Run("Successful net user node isallowed", func(t *testing.T) {
		networkUser.Nodes = append(networkUser.Nodes, nodes[0].ID.String())
		err := UpdateNetworkUser(network.NetID, &networkUser)
		assert.Nil(t, err)
		isUserNodeAllowed := IsUserNodeAllowed(nodes[:], network.NetID, string(networkUser.ID), nodes[0].ID.String())
		assert.True(t, isUserNodeAllowed)
	})

	t.Run("Successful net user node not allowed", func(t *testing.T) {
		isUserNodeAllowed := IsUserNodeAllowed(nodes[:], network.NetID, string(networkUser.ID), "notanode")
		assert.False(t, isUserNodeAllowed)
	})

	t.Run("Successful net user client isallowed", func(t *testing.T) {
		networkUser.Clients = append(networkUser.Clients, "coolclient")
		err := UpdateNetworkUser(network.NetID, &networkUser)
		assert.Nil(t, err)
		isUserClientAllowed := IsUserClientAllowed(clients[:], network.NetID, string(networkUser.ID), "coolclient")
		assert.True(t, isUserClientAllowed)
	})

	t.Run("Successful net user client not allowed", func(t *testing.T) {
		isUserClientAllowed := IsUserClientAllowed(clients[:], network.NetID, string(networkUser.ID), "notaclient")
		assert.False(t, isUserClientAllowed)
	})

	t.Run("Successful net user delete", func(t *testing.T) {
		err := DeleteNetworkUser(network.NetID, string(networkUser.ID))
		assert.Nil(t, err)
		user, err := GetNetworkUser(network.NetID, networkUser.ID)
		assert.Nil(t, user)
		assert.NotNil(t, err)
	})
}