mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-13 07:16:31 +08:00
126 lines
3.1 KiB
Go
126 lines
3.1 KiB
Go
package controller
|
|
|
|
import (
|
|
"github.com/gravitl/netmaker/logic/nodeacls"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/gravitl/netmaker/database"
|
|
"github.com/gravitl/netmaker/logic"
|
|
"github.com/gravitl/netmaker/models"
|
|
"github.com/stretchr/testify/assert"
|
|
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
|
)
|
|
|
|
var nonLinuxHost models.Host
|
|
var linuxHost models.Host
|
|
|
|
func TestGetNetworkNodes(t *testing.T) {
|
|
deleteAllNetworks()
|
|
createNet()
|
|
t.Run("BadNet", func(t *testing.T) {
|
|
node, err := logic.GetNetworkNodes("badnet")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, []models.Node{}, node)
|
|
})
|
|
t.Run("NoNodes", func(t *testing.T) {
|
|
node, err := logic.GetNetworkNodes("skynet")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, []models.Node{}, node)
|
|
})
|
|
t.Run("Success", func(t *testing.T) {
|
|
createTestNode()
|
|
node, err := logic.GetNetworkNodes("skynet")
|
|
assert.Nil(t, err)
|
|
assert.NotEqual(t, []models.LegacyNode(nil), node)
|
|
})
|
|
|
|
}
|
|
|
|
func TestValidateEgressGateway(t *testing.T) {
|
|
var gateway models.EgressGatewayRequest
|
|
|
|
t.Run("Success", func(t *testing.T) {
|
|
gateway.Ranges = []string{"10.100.100.0/24"}
|
|
err := logic.ValidateEgressGateway(gateway)
|
|
assert.Nil(t, err)
|
|
})
|
|
}
|
|
|
|
func TestNodeACLs(t *testing.T) {
|
|
deleteAllNodes()
|
|
node1 := createNodeWithParams("", "10.0.0.50/32")
|
|
node2 := createNodeWithParams("", "10.0.0.100/32")
|
|
logic.AssociateNodeToHost(node1, &linuxHost)
|
|
logic.AssociateNodeToHost(node2, &linuxHost)
|
|
|
|
t.Run("node acls correct after add new node not allowed", func(t *testing.T) {
|
|
node3 := createNodeWithParams("", "10.0.0.100/32")
|
|
createNodeHosts()
|
|
n, e := logic.GetNetwork(node3.Network)
|
|
assert.Nil(t, e)
|
|
n.DefaultACL = "no"
|
|
e = logic.SaveNetwork(&n)
|
|
assert.Nil(t, e)
|
|
err := logic.AssociateNodeToHost(node3, &linuxHost)
|
|
assert.Nil(t, err)
|
|
})
|
|
t.Run("node acls removed", func(t *testing.T) {
|
|
err := nodeacls.RemoveNodeACL(node1.Network, node1.ID.String())
|
|
assert.Nil(t, err)
|
|
})
|
|
deleteAllNodes()
|
|
}
|
|
|
|
func deleteAllNodes() {
|
|
database.DeleteAllRecords(database.NODES_TABLE_NAME)
|
|
}
|
|
|
|
func createTestNode() *models.Node {
|
|
createNodeHosts()
|
|
n := createNodeWithParams("skynet", "")
|
|
_ = logic.AssociateNodeToHost(n, &linuxHost)
|
|
return n
|
|
}
|
|
|
|
func createNodeWithParams(network, address string) *models.Node {
|
|
_, ipnet, _ := net.ParseCIDR("10.0.0.1/32")
|
|
tmpCNode := models.CommonNode{
|
|
ID: uuid.New(),
|
|
Network: "skynet",
|
|
Address: *ipnet,
|
|
}
|
|
if len(network) > 0 {
|
|
tmpCNode.Network = network
|
|
}
|
|
if len(address) > 0 {
|
|
_, ipnet2, _ := net.ParseCIDR(address)
|
|
tmpCNode.Address = *ipnet2
|
|
}
|
|
createnode := models.Node{
|
|
CommonNode: tmpCNode,
|
|
}
|
|
return &createnode
|
|
}
|
|
|
|
func createNodeHosts() {
|
|
k, _ := wgtypes.ParseKey("DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34=")
|
|
linuxHost = models.Host{
|
|
ID: uuid.New(),
|
|
PublicKey: k.PublicKey(),
|
|
HostPass: "password",
|
|
OS: "linux",
|
|
Name: "linuxhost",
|
|
}
|
|
_ = logic.CreateHost(&linuxHost)
|
|
nonLinuxHost = models.Host{
|
|
ID: uuid.New(),
|
|
OS: "windows",
|
|
PublicKey: k.PublicKey(),
|
|
Name: "windowshost",
|
|
HostPass: "password",
|
|
}
|
|
|
|
_ = logic.CreateHost(&nonLinuxHost)
|
|
}
|