fixed network tests that failed with nodes

This commit is contained in:
Matthew R Kasun
2021-04-24 18:03:26 -04:00
parent f28937c1f6
commit ae826002cd
6 changed files with 1165 additions and 721 deletions

View File

@@ -1,19 +1,19 @@
package controller package controller
import ( import (
"gopkg.in/go-playground/validator.v9"
"log"
"fmt"
"golang.org/x/crypto/bcrypt"
"github.com/gravitl/netmaker/mongoconn"
"github.com/gravitl/netmaker/functions"
"context" "context"
"go.mongodb.org/mongo-driver/bson" "fmt"
"time" "log"
"net" "net"
"github.com/gravitl/netmaker/models" "time"
"go.mongodb.org/mongo-driver/mongo/options"
"github.com/gravitl/netmaker/functions"
"github.com/gravitl/netmaker/models"
"github.com/gravitl/netmaker/mongoconn"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo/options"
"golang.org/x/crypto/bcrypt"
"gopkg.in/go-playground/validator.v9"
) )
func GetPeersList(networkName string) ([]models.PeersResponse, error) { func GetPeersList(networkName string) ([]models.PeersResponse, error) {
@@ -58,7 +58,6 @@ func GetPeersList(networkName string) ([]models.PeersResponse, error) {
return peers, err return peers, err
} }
func ValidateNode(operation string, networkName string, node models.Node) error { func ValidateNode(operation string, networkName string, node models.Node) error {
v := validator.New() v := validator.New()
@@ -67,16 +66,15 @@ func ValidateNode(operation string, networkName string, node models.Node) error
//var isFieldUnique bool = functions.IsFieldUnique(networkName, "endpoint", node.Endpoint) //var isFieldUnique bool = functions.IsFieldUnique(networkName, "endpoint", node.Endpoint)
isIpv4 := functions.IsIpv4Net(node.Endpoint) isIpv4 := functions.IsIpv4Net(node.Endpoint)
notEmptyCheck := node.Endpoint != "" notEmptyCheck := node.Endpoint != ""
return (notEmptyCheck && isIpv4) || operation == "update" return (notEmptyCheck && isIpv4)
}) })
_ = v.RegisterValidation("localaddress_check", func(fl validator.FieldLevel) bool { _ = v.RegisterValidation("localaddress_check", func(fl validator.FieldLevel) bool {
//var isFieldUnique bool = functions.IsFieldUnique(networkName, "endpoint", node.Endpoint) //var isFieldUnique bool = functions.IsFieldUnique(networkName, "endpoint", node.Endpoint)
isIpv4 := functions.IsIpv4Net(node.LocalAddress) isIpv4 := functions.IsIpv4Net(node.LocalAddress)
notEmptyCheck := node.LocalAddress != "" notEmptyCheck := node.LocalAddress != ""
return (notEmptyCheck && isIpv4) || operation == "update" return (notEmptyCheck && isIpv4)
}) })
_ = v.RegisterValidation("macaddress_unique", func(fl validator.FieldLevel) bool { _ = v.RegisterValidation("macaddress_unique", func(fl validator.FieldLevel) bool {
var isFieldUnique bool = functions.IsFieldUnique(networkName, "macaddress", node.MacAddress) var isFieldUnique bool = functions.IsFieldUnique(networkName, "macaddress", node.MacAddress)
return isFieldUnique || operation == "update" return isFieldUnique || operation == "update"
@@ -99,12 +97,12 @@ func ValidateNode(operation string, networkName string, node models.Node) error
_ = v.RegisterValidation("pubkey_check", func(fl validator.FieldLevel) bool { _ = v.RegisterValidation("pubkey_check", func(fl validator.FieldLevel) bool {
notEmptyCheck := node.PublicKey != "" notEmptyCheck := node.PublicKey != ""
isBase64 := functions.IsBase64(node.PublicKey) isBase64 := functions.IsBase64(node.PublicKey)
return (notEmptyCheck && isBase64) || operation == "update" return (notEmptyCheck && isBase64)
}) })
_ = v.RegisterValidation("password_check", func(fl validator.FieldLevel) bool { _ = v.RegisterValidation("password_check", func(fl validator.FieldLevel) bool {
notEmptyCheck := node.Password != "" notEmptyCheck := node.Password != ""
goodLength := len(node.Password) > 5 goodLength := len(node.Password) > 5
return (notEmptyCheck && goodLength) || operation == "update" return (notEmptyCheck && goodLength)
}) })
err := v.Struct(node) err := v.Struct(node)
@@ -117,7 +115,6 @@ func ValidateNode(operation string, networkName string, node models.Node) error
return err return err
} }
func UpdateNode(nodechange models.Node, node models.Node) (models.Node, error) { func UpdateNode(nodechange models.Node, node models.Node) (models.Node, error) {
//Question: Is there a better way of doing this than a bunch of "if" statements? probably... //Question: Is there a better way of doing this than a bunch of "if" statements? probably...
//Eventually, lets have a better way to check if any of the fields are filled out... //Eventually, lets have a better way to check if any of the fields are filled out...
@@ -156,7 +153,7 @@ func UpdateNode(nodechange models.Node, node models.Node) (models.Node, error) {
if nodechange.Endpoint != "" { if nodechange.Endpoint != "" {
node.Endpoint = nodechange.Endpoint node.Endpoint = nodechange.Endpoint
notifynetwork = true notifynetwork = true
} }
if nodechange.SaveConfig != nil { if nodechange.SaveConfig != nil {
node.SaveConfig = nodechange.SaveConfig node.SaveConfig = nodechange.SaveConfig
} }
@@ -221,6 +218,9 @@ func UpdateNode(nodechange models.Node, node models.Node) (models.Node, error) {
return nodeupdate, errN return nodeupdate, errN
} }
if nodechange.MacAddress != "" {
queryMac = nodechange.MacAddress
}
returnnode, errN := GetNode(queryMac, queryNetwork) returnnode, errN := GetNode(queryMac, queryNetwork)
defer cancel() defer cancel()
@@ -285,7 +285,6 @@ func CreateNode(node models.Node, networkName string) (models.Node, error) {
//set password to encrypted password //set password to encrypted password
node.Password = string(hash) node.Password = string(hash)
node.Network = networkName node.Network = networkName
//node.SetDefaults() //node.SetDefaults()
@@ -324,7 +323,6 @@ func CreateNode(node models.Node, networkName string) (models.Node, error) {
collection := mongoconn.Client.Database("netmaker").Collection("nodes") collection := mongoconn.Client.Database("netmaker").Collection("nodes")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
// insert our node to the node db. // insert our node to the node db.
result, err := collection.InsertOne(ctx, node) result, err := collection.InsertOne(ctx, node)
_ = result _ = result
@@ -351,18 +349,18 @@ func NodeCheckIn(node models.Node, networkName string) (models.CheckInResponse,
var response models.CheckInResponse var response models.CheckInResponse
parentnetwork, err := functions.GetParentNetwork(networkName) parentnetwork, err := functions.GetParentNetwork(networkName)
if err != nil{ if err != nil {
err = fmt.Errorf("%w; Couldnt retrieve Network " + networkName + ": ", err) err = fmt.Errorf("%w; Couldnt retrieve Network "+networkName+": ", err)
return response, err return response, err
} }
parentnode, err := functions.GetNodeByMacAddress(networkName, node.MacAddress) parentnode, err := functions.GetNodeByMacAddress(networkName, node.MacAddress)
if err != nil{ if err != nil {
err = fmt.Errorf("%w; Couldnt Get Node " + node.MacAddress, err) err = fmt.Errorf("%w; Couldnt Get Node "+node.MacAddress, err)
return response, err return response, err
} }
if parentnode.IsPending { if parentnode.IsPending {
err = fmt.Errorf("%w; Node checking in is still pending: " + node.MacAddress, err) err = fmt.Errorf("%w; Node checking in is still pending: "+node.MacAddress, err)
response.IsPending = true response.IsPending = true
return response, err return response, err
} }
@@ -394,7 +392,7 @@ func NodeCheckIn(node models.Node, networkName string) (models.CheckInResponse,
} else { } else {
err = TimestampNode(parentnode, true, false, false) err = TimestampNode(parentnode, true, false, false)
if err != nil{ if err != nil {
err = fmt.Errorf("%w; Couldnt Timestamp Node: ", err) err = fmt.Errorf("%w; Couldnt Timestamp Node: ", err)
return response, err return response, err
} }
@@ -433,7 +431,7 @@ func SetNetworkNodesLastModified(networkName string) error {
return nil return nil
} }
func TimestampNode(node models.Node, updatecheckin bool, updatepeers bool, updatelm bool) error{ func TimestampNode(node models.Node, updatecheckin bool, updatepeers bool, updatelm bool) error {
if updatelm { if updatelm {
node.SetLastModified() node.SetLastModified()
} }
@@ -466,5 +464,3 @@ func TimestampNode(node models.Node, updatecheckin bool, updatepeers bool, updat
return err return err
} }

View File

@@ -387,6 +387,12 @@ func checkIn(w http.ResponseWriter, r *http.Request) {
} }
//TODO: check node last modified vs network last modified //TODO: check node last modified vs network last modified
//Get Updated node to return
node, err = GetNode(params["macaddress"], params["network"])
if err != nil {
returnErrorResponse(w, r, formatError(err, "internal"))
return
}
w.WriteHeader(http.StatusOK) w.WriteHeader(http.StatusOK)
json.NewEncoder(w).Encode(node) json.NewEncoder(w).Encode(node)
@@ -660,6 +666,10 @@ func validateGateway(gateway models.GatewayRequest) error {
if empty || !isIpv4 { if empty || !isIpv4 {
err = errors.New("IP Range Not Valid") err = errors.New("IP Range Not Valid")
} }
empty = gateway.Interface == ""
if empty {
err = errors.New("Interface cannot be empty")
}
return err return err
} }

View File

@@ -5,20 +5,21 @@
package functions package functions
import ( import (
"fmt"
"errors"
"math/rand"
"time"
"context" "context"
"encoding/base64" "encoding/base64"
"strings" "errors"
"fmt"
"math/rand"
"net" "net"
"strings"
"time"
"github.com/gravitl/netmaker/models" "github.com/gravitl/netmaker/models"
"github.com/gravitl/netmaker/mongoconn" "github.com/gravitl/netmaker/mongoconn"
"go.mongodb.org/mongo-driver/bson" "go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive" "go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo/options"
"go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
) )
//Takes in an arbitrary field and value for field and checks to see if any other //Takes in an arbitrary field and value for field and checks to see if any other
@@ -48,7 +49,6 @@ func CreateServerToken(netID string) (string, error) {
privAddr = network.LocalRange privAddr = network.LocalRange
} }
fmt.Println("Token details:") fmt.Println("Token details:")
fmt.Println(" grpc address + port: " + address) fmt.Println(" grpc address + port: " + address)
fmt.Println(" network: " + netID) fmt.Println(" network: " + netID)
@@ -60,7 +60,6 @@ func CreateServerToken(netID string) (string, error) {
fmt.Println(" access string: " + accesskey.AccessString) fmt.Println(" access string: " + accesskey.AccessString)
network.AccessKeys = append(network.AccessKeys, accesskey) network.AccessKeys = append(network.AccessKeys, accesskey)
collection := mongoconn.Client.Database("netmaker").Collection("networks") collection := mongoconn.Client.Database("netmaker").Collection("networks")
@@ -108,7 +107,7 @@ func IsFieldUnique(network string, field string, value string) bool {
return isunique return isunique
} }
if (node.Name != "") { if node.Name != "" {
isunique = false isunique = false
} }
@@ -185,6 +184,7 @@ func UpdateNetworkNodeAddresses(networkName string) error {
return err return err
} }
//TODO TODO TODO!!!!! //TODO TODO TODO!!!!!
func UpdateNetworkPrivateAddresses(networkName string) error { func UpdateNetworkPrivateAddresses(networkName string) error {
@@ -232,7 +232,7 @@ func UpdateNetworkPrivateAddresses(networkName string) error {
} }
//Checks to see if any other networks have the same name (id) //Checks to see if any other networks have the same name (id)
func IsNetworkNameUnique(name string) (bool, error ){ func IsNetworkNameUnique(name string) (bool, error) {
isunique := true isunique := true
@@ -252,7 +252,7 @@ func IsNetworkNameUnique(name string) (bool, error ){
return isunique, nil return isunique, nil
} }
func IsNetworkDisplayNameUnique(name string) (bool, error){ func IsNetworkDisplayNameUnique(name string) (bool, error) {
isunique := true isunique := true
@@ -261,7 +261,6 @@ func IsNetworkDisplayNameUnique(name string) (bool, error){
return false, err return false, err
} }
for i := 0; i < len(dbs); i++ { for i := 0; i < len(dbs); i++ {
if name == dbs[i].DisplayName { if name == dbs[i].DisplayName {
@@ -272,9 +271,9 @@ func IsNetworkDisplayNameUnique(name string) (bool, error){
return isunique, nil return isunique, nil
} }
func GetNetworkNodeNumber(networkName string) (int, error){ func GetNetworkNodeNumber(networkName string) (int, error) {
collection := mongoconn.Client.Database("wirecat").Collection("nodes") collection := mongoconn.Client.Database("netmaker").Collection("nodes")
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second) ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
@@ -292,10 +291,9 @@ func GetNetworkNodeNumber(networkName string) (int, error){
return returncount, err return returncount, err
} }
//Kind of a weird name. Should just be GetNetworks I think. Consider changing. //Kind of a weird name. Should just be GetNetworks I think. Consider changing.
//Anyway, returns all the networks //Anyway, returns all the networks
func ListNetworks() ([]models.Network, error){ func ListNetworks() ([]models.Network, error) {
var networks []models.Network var networks []models.Network
@@ -334,7 +332,7 @@ func ListNetworks() ([]models.Network, error){
//Does so by checking against all keys and seeing if any have the same value //Does so by checking against all keys and seeing if any have the same value
//may want to hash values before comparing...consider this //may want to hash values before comparing...consider this
//TODO: No error handling!!!! //TODO: No error handling!!!!
func IsKeyValid(networkname string, keyvalue string) bool{ func IsKeyValid(networkname string, keyvalue string) bool {
network, _ := GetParentNetwork(networkname) network, _ := GetParentNetwork(networkname)
var key models.AccessKey var key models.AccessKey
@@ -342,7 +340,7 @@ func IsKeyValid(networkname string, keyvalue string) bool{
isvalid := false isvalid := false
for i := len(network.AccessKeys) - 1; i >= 0; i-- { for i := len(network.AccessKeys) - 1; i >= 0; i-- {
currentkey:= network.AccessKeys[i] currentkey := network.AccessKeys[i]
if currentkey.Value == keyvalue { if currentkey.Value == keyvalue {
key = currentkey key = currentkey
foundkey = true foundkey = true
@@ -355,6 +353,7 @@ func IsKeyValid(networkname string, keyvalue string) bool{
} }
return isvalid return isvalid
} }
//TODO: Contains a fatal error return. Need to change //TODO: Contains a fatal error return. Need to change
//This just gets a network object from a network name //This just gets a network object from a network name
//Should probably just be GetNetwork. kind of a dumb name. //Should probably just be GetNetwork. kind of a dumb name.
@@ -435,7 +434,7 @@ func GetNodeObj(id primitive.ObjectID) models.Node {
//This checks to make sure a network name is valid. //This checks to make sure a network name is valid.
//Switch to REGEX? //Switch to REGEX?
func NameInNetworkCharSet(name string) bool{ func NameInNetworkCharSet(name string) bool {
charset := "abcdefghijklmnopqrstuvwxyz1234567890-_" charset := "abcdefghijklmnopqrstuvwxyz1234567890-_"
@@ -447,7 +446,7 @@ func NameInNetworkCharSet(name string) bool{
return true return true
} }
func NameInNodeCharSet(name string) bool{ func NameInNodeCharSet(name string) bool {
charset := "abcdefghijklmnopqrstuvwxyz1234567890-" charset := "abcdefghijklmnopqrstuvwxyz1234567890-"
@@ -459,7 +458,6 @@ func NameInNodeCharSet(name string) bool{
return true return true
} }
//This returns a node based on its mac address. //This returns a node based on its mac address.
//The mac address acts as the Unique ID for nodes. //The mac address acts as the Unique ID for nodes.
//Is this a dumb thing to do? I thought it was cool but maybe it's dumb. //Is this a dumb thing to do? I thought it was cool but maybe it's dumb.
@@ -489,7 +487,7 @@ func GetNodeByMacAddress(network string, macaddress string) (models.Node, error)
//and checks against all nodes to see if it's taken, until it finds one. //and checks against all nodes to see if it's taken, until it finds one.
//TODO: We do not handle a case where we run out of addresses. //TODO: We do not handle a case where we run out of addresses.
//We will need to handle that eventually //We will need to handle that eventually
func UniqueAddress(networkName string) (string, error){ func UniqueAddress(networkName string) (string, error) {
var network models.Network var network models.Network
network, err := GetParentNetwork(networkName) network, err := GetParentNetwork(networkName)
@@ -509,7 +507,7 @@ func UniqueAddress(networkName string) (string, error){
offset = false offset = false
continue continue
} }
if IsIPUnique(networkName, ip.String()){ if IsIPUnique(networkName, ip.String()) {
return ip.String(), err return ip.String(), err
} }
} }
@@ -547,8 +545,6 @@ func GetGlobalConfig() (bool, models.GlobalConfig, error) {
return create, globalconf, err return create, globalconf, err
} }
//generate an access key value //generate an access key value
func GenKey() string { func GenKey() string {
@@ -606,7 +602,7 @@ func IsIPUnique(network string, ip string) bool {
return isunique return isunique
} }
if (node.Address == ip) { if node.Address == ip {
isunique = false isunique = false
} }
return isunique return isunique
@@ -656,6 +652,7 @@ func DecrimentKey(networkName string, keyvalue string) {
return return
} }
} }
//takes the logic from controllers.deleteKey //takes the logic from controllers.deleteKey
func DeleteKey(network models.Network, i int) { func DeleteKey(network models.Network, i int) {
@@ -684,9 +681,10 @@ func DeleteKey(network models.Network, i int) {
return return
} }
} }
//increments an IP over the previous //increments an IP over the previous
func Inc(ip net.IP) { func Inc(ip net.IP) {
for j := len(ip)-1; j>=0; j-- { for j := len(ip) - 1; j >= 0; j-- {
ip[j]++ ip[j]++
if ip[j] > 0 { if ip[j] > 0 {
break break

View File

@@ -216,22 +216,14 @@ func networkExists(t *testing.T) bool {
return false return false
} }
func TestJunk(t *testing.T) {
deleteNetworks(t)
}
func deleteNetworks(t *testing.T) { func deleteNetworks(t *testing.T) {
//delete all network nodes //delete all node
var nodes []models.ReturnNode deleteAllNodes(t)
response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/skynet", "secretkey") response, err := api(t, "", http.MethodGet, baseURL+"/api/networks", "secretkey")
assert.Nil(t, err, err)
if response.StatusCode == http.StatusOK {
err = json.NewDecoder(response.Body).Decode(&nodes)
response.Body.Close()
assert.Nil(t, err, err)
for _, node := range nodes {
resp, err := api(t, "", http.MethodDelete, baseURL+"/api/nodes/skynet/"+node.MacAddress, "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, resp.StatusCode)
}
}
response, err = api(t, "", http.MethodGet, baseURL+"/api/networks", "secretkey")
assert.Nil(t, err, err) assert.Nil(t, err, err)
if response.StatusCode == http.StatusOK { if response.StatusCode == http.StatusOK {
defer response.Body.Close() defer response.Body.Close()
@@ -260,12 +252,24 @@ func getNetworkNodes(t *testing.T) []models.ReturnNode {
return nodes return nodes
} }
func deleteNode(t *testing.T, node models.ReturnNode) { func deleteNode(t *testing.T) {
response, err := api(t, "", http.MethodDelete, baseURL+"/api/nodes/skynet/"+node.MacAddress, "secretkey") response, err := api(t, "", http.MethodDelete, baseURL+"/api/nodes/skynet/01:02:03:04:05:06", "secretkey")
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode) assert.Equal(t, http.StatusOK, response.StatusCode)
} }
func deleteAllNodes(t *testing.T) {
response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode)
var nodes []models.ReturnNode
defer response.Body.Close()
json.NewDecoder(response.Body).Decode(&nodes)
for _, node := range nodes {
resp, err := api(t, "", http.MethodDelete, baseURL+"/api/nodes/"+node.Network+"/"+node.MacAddress, "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, resp.StatusCode)
}
}
func createNode(t *testing.T) { func createNode(t *testing.T) {
var node models.Node var node models.Node
key := createAccessKey(t) key := createAccessKey(t)
@@ -291,3 +295,9 @@ func getNode(t *testing.T) models.Node {
assert.Nil(t, err, err) assert.Nil(t, err, err)
return node return node
} }
func setup(t *testing.T) {
deleteNetworks(t)
createNetwork(t)
createNode(t)
}

View File

@@ -8,7 +8,6 @@ import (
"github.com/gravitl/netmaker/models" "github.com/gravitl/netmaker/models"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"go.mongodb.org/mongo-driver/mongo"
) )
func TestCreateNetwork(t *testing.T) { func TestCreateNetwork(t *testing.T) {
@@ -118,6 +117,7 @@ func TestGetNetwork(t *testing.T) {
func TestDeleteNetwork(t *testing.T) { func TestDeleteNetwork(t *testing.T) {
t.Run("InvalidKey", func(t *testing.T) { t.Run("InvalidKey", func(t *testing.T) {
setup(t)
response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/skynet", "badkey") response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/skynet", "badkey")
assert.Nil(t, err, err) assert.Nil(t, err, err)
defer response.Body.Close() defer response.Body.Close()
@@ -128,17 +128,6 @@ func TestDeleteNetwork(t *testing.T) {
assert.Equal(t, http.StatusUnauthorized, message.Code) assert.Equal(t, http.StatusUnauthorized, message.Code)
assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message) assert.Equal(t, "W1R3: You are unauthorized to access this endpoint.", message.Message)
}) })
t.Run("ValidKey", func(t *testing.T) {
response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/skynet", "secretkey")
assert.Nil(t, err, err)
defer response.Body.Close()
var message mongo.DeleteResult
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode)
assert.Equal(t, int64(1), message.DeletedCount)
})
t.Run("Badnetwork", func(t *testing.T) { t.Run("Badnetwork", func(t *testing.T) {
response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/badnetwork", "secretkey") response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/badnetwork", "secretkey")
assert.Nil(t, err, err) assert.Nil(t, err, err)
@@ -150,7 +139,33 @@ func TestDeleteNetwork(t *testing.T) {
assert.Equal(t, http.StatusNotFound, response.StatusCode) assert.Equal(t, http.StatusNotFound, response.StatusCode)
}) })
t.Run("NodesExist", func(t *testing.T) { t.Run("NodesExist", func(t *testing.T) {
setup(t)
node := getNode(t)
t.Log(node)
response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/skynet", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusForbidden, response.StatusCode)
defer response.Body.Close()
var message models.ErrorResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Contains(t, message.Message, "Node check failed")
assert.Equal(t, http.StatusForbidden, message.Code)
})
t.Run("ValidKey", func(t *testing.T) {
type Message struct {
DeletedCount int64
}
setup(t)
deleteAllNodes(t)
response, err := api(t, "", http.MethodDelete, baseURL+"/api/networks/skynet", "secretkey")
assert.Nil(t, err, err)
defer response.Body.Close()
var message Message
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode)
assert.Equal(t, int64(1), message.DeletedCount)
}) })
} }

View File

@@ -4,23 +4,34 @@ import (
"encoding/json" "encoding/json"
"net/http" "net/http"
"testing" "testing"
"time"
"github.com/gravitl/netmaker/models" "github.com/gravitl/netmaker/models"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
"golang.org/x/crypto/bcrypt"
) )
func setup(t *testing.T) { func TestRainyDay(t *testing.T) {
deleteNetworks(t) t.Run("badkey", func(t *testing.T) {
createNetwork(t) response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes", "badkey")
createNode(t) assert.Nil(t, err, err)
} assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
})
func TestJunk(t *testing.T) { t.Run("badURL", func(t *testing.T) {
deleteNetworks(t) response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/adm/skynet/01:02:03:04:05:07", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusNotFound, response.StatusCode)
})
t.Run("NonExistentNetwork", func(t *testing.T) {
response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/badnet", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusNotFound, response.StatusCode)
})
} }
func TestGetAllNodes(t *testing.T) { func TestGetAllNodes(t *testing.T) {
setup(t) setup(t)
t.Run("NodesExist", func(t *testing.T) {
response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes", "secretkey") response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes", "secretkey")
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode) assert.Equal(t, http.StatusOK, response.StatusCode)
@@ -28,13 +39,24 @@ func TestGetAllNodes(t *testing.T) {
defer response.Body.Close() defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&nodes) err = json.NewDecoder(response.Body).Decode(&nodes)
assert.Nil(t, err, err) assert.Nil(t, err, err)
for _, node := range nodes { assert.NotEmpty(t, nodes)
assert.NotNil(t, node, "empty node") })
} t.Run("NodeDoesNotExist", func(t *testing.T) {
deleteNode(t)
response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode)
var nodes []models.ReturnNode
defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&nodes)
assert.Nil(t, err, err)
assert.Empty(t, nodes)
})
} }
func TestGetNetworkNodes(t *testing.T) { func TestGetNetworkNodes(t *testing.T) {
setup(t) setup(t)
t.Run("NodeExists", func(t *testing.T) {
response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/skynet", "secretkey") response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/skynet", "secretkey")
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode) assert.Equal(t, http.StatusOK, response.StatusCode)
@@ -42,13 +64,24 @@ func TestGetNetworkNodes(t *testing.T) {
defer response.Body.Close() defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&nodes) err = json.NewDecoder(response.Body).Decode(&nodes)
assert.Nil(t, err, err) assert.Nil(t, err, err)
for _, node := range nodes { assert.NotEmpty(t, nodes)
assert.NotNil(t, node, "empty node") })
} t.Run("NodeDoesNotExit", func(t *testing.T) {
deleteNode(t)
response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/skynet", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode)
var nodes []models.ReturnNode
defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&nodes)
assert.Nil(t, err, err)
assert.Empty(t, nodes)
})
} }
func TestGetNode(t *testing.T) { func TestGetNode(t *testing.T) {
setup(t) setup(t)
t.Run("NodeExists", func(t *testing.T) {
response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/skynet/01:02:03:04:05:06", "secretkey") response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/skynet/01:02:03:04:05:06", "secretkey")
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode) assert.Equal(t, http.StatusOK, response.StatusCode)
@@ -57,22 +90,128 @@ func TestGetNode(t *testing.T) {
err = json.NewDecoder(response.Body).Decode(&node) err = json.NewDecoder(response.Body).Decode(&node)
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.Equal(t, "01:02:03:04:05:06", node.MacAddress) assert.Equal(t, "01:02:03:04:05:06", node.MacAddress)
})
t.Run("NodeDoesNotExist", func(t *testing.T) {
deleteNode(t)
response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/skynet/01:02:03:04:05:06", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusInternalServerError, response.StatusCode)
})
} }
func TestUpdateNode(t *testing.T) { func TestUpdateNode(t *testing.T) {
var data models.Node
setup(t) setup(t)
var data struct {
Name string t.Run("UpdateMulti", func(t *testing.T) {
} data.MacAddress = "01:02:03:04:05:05"
data.Name = "NewName" data.Name = "NewName"
data.PublicKey = "DM5qhLAE20PG9BbfBCgfr+Ac9D2NDOwCtY1rbYDLf34="
data.Password = "newpass"
data.LocalAddress = "192.168.0.2"
data.Endpoint = "10.100.100.5"
response, err := api(t, data, http.MethodPut, baseURL+"/api/nodes/skynet/01:02:03:04:05:06", "secretkey") response, err := api(t, data, http.MethodPut, baseURL+"/api/nodes/skynet/01:02:03:04:05:06", "secretkey")
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode) assert.Equal(t, http.StatusOK, response.StatusCode)
var node models.Node
defer response.Body.Close() defer response.Body.Close()
var node models.Node
t.Log(response.Body)
err = json.NewDecoder(response.Body).Decode(&node) err = json.NewDecoder(response.Body).Decode(&node)
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.Equal(t, data.Name, node.Name) assert.Equal(t, data.Name, node.Name)
assert.Equal(t, data.PublicKey, node.PublicKey)
err = bcrypt.CompareHashAndPassword([]byte(node.Password), []byte(data.Password))
assert.Nil(t, err, err)
assert.Equal(t, data.LocalAddress, node.LocalAddress)
assert.Equal(t, data.Endpoint, node.Endpoint)
})
t.Run("InvalidMacAddress", func(t *testing.T) {
data.MacAddress = "10:11:12:13:14:15:16"
response, err := api(t, data, http.MethodPut, baseURL+"/api/nodes/skynet/01:02:03:04:05:05", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
var message models.ErrorResponse
defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Field validation for 'MacAddress' failed")
})
t.Run("InvalidEndpoint", func(t *testing.T) {
data.Endpoint = "10.10.10.300"
response, err := api(t, data, http.MethodPut, baseURL+"/api/nodes/skynet/01:02:03:04:05:05", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
var message models.ErrorResponse
defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Field validation for 'Endpoint' failed")
})
t.Run("InvalidLocalAddress", func(t *testing.T) {
data.LocalAddress = "10.10.10.300"
response, err := api(t, data, http.MethodPut, baseURL+"/api/nodes/skynet/01:02:03:04:05:05", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
var message models.ErrorResponse
defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Field validation for 'LocalAddress' failed")
})
t.Run("InvalidName", func(t *testing.T) {
var data struct {
Name string
}
data.Name = "New*Name"
response, err := api(t, data, http.MethodPut, baseURL+"/api/nodes/skynet/01:02:03:04:05:05", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
var message models.ErrorResponse
defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Field validation for 'Name' failed")
})
t.Run("InvalidPublicKey", func(t *testing.T) {
data.PublicKey = "xxx"
response, err := api(t, data, http.MethodPut, baseURL+"/api/nodes/skynet/01:02:03:04:05:05", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
var message models.ErrorResponse
defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Field validation for 'PublicKey' failed")
})
t.Run("InvalidPassword", func(t *testing.T) {
data.Password = "1234"
response, err := api(t, data, http.MethodPut, baseURL+"/api/nodes/skynet/01:02:03:04:05:05", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
var message models.ErrorResponse
defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Field validation for 'Password' failed")
})
t.Run("EmptyPassword", func(t *testing.T) {
data.Password = ""
response, err := api(t, data, http.MethodPut, baseURL+"/api/nodes/skynet/01:02:03:04:05:05", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
var message models.ErrorResponse
defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Field validation for 'Password' failed")
})
} }
func TestDeleteNode(t *testing.T) { func TestDeleteNode(t *testing.T) {
@@ -103,9 +242,10 @@ func TestDeleteNode(t *testing.T) {
} }
func TestCheckIn(t *testing.T) { func TestCheckIn(t *testing.T) {
//get node
//oldNode := getNode(t)
setup(t) setup(t)
oldNode := getNode(t)
time.Sleep(1 * time.Second)
t.Run("Valid", func(t *testing.T) {
response, err := api(t, "", http.MethodPost, baseURL+"/api/nodes/skynet/01:02:03:04:05:06/checkin", "secretkey") response, err := api(t, "", http.MethodPost, baseURL+"/api/nodes/skynet/01:02:03:04:05:06/checkin", "secretkey")
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode) assert.Equal(t, http.StatusOK, response.StatusCode)
@@ -113,13 +253,27 @@ func TestCheckIn(t *testing.T) {
defer response.Body.Close() defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&node) err = json.NewDecoder(response.Body).Decode(&node)
assert.Nil(t, err, err) assert.Nil(t, err, err)
//assert.Greater(t, node.LastCheckIn, oldNode.LastCheckin) assert.Greater(t, node.LastCheckIn, oldNode.LastCheckIn)
})
t.Run("NodeDoesNotExist", func(t *testing.T) {
deleteNode(t)
response, err := api(t, "", http.MethodPost, baseURL+"/api/nodes/skynet/01:02:03:04:05:06/checkin", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusInternalServerError, response.StatusCode)
var message models.ErrorResponse
defer response.Body.Close()
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusInternalServerError, message.Code)
assert.Equal(t, "mongo: no documents in result", message.Message)
})
} }
func TestCreateGateway(t *testing.T) { func TestCreateGateway(t *testing.T) {
setup(t) setup(t)
//assert.False(t, node.IsGateway) //assert.False(t, node.IsGateway)
var gateway models.GatewayRequest var gateway models.GatewayRequest
t.Run("Valid", func(t *testing.T) {
gateway.RangeString = "0.0.0.0/0" gateway.RangeString = "0.0.0.0/0"
gateway.Interface = "eth0" gateway.Interface = "eth0"
response, err := api(t, gateway, http.MethodPost, baseURL+"/api/nodes/skynet/01:02:03:04:05:06/creategateway", "secretkey") response, err := api(t, gateway, http.MethodPost, baseURL+"/api/nodes/skynet/01:02:03:04:05:06/creategateway", "secretkey")
@@ -130,6 +284,33 @@ func TestCreateGateway(t *testing.T) {
err = json.NewDecoder(response.Body).Decode(&message) err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.True(t, message.IsGateway) assert.True(t, message.IsGateway)
})
t.Run("BadRange", func(t *testing.T) {
gateway.RangeString = "0.0.0.0/36"
gateway.Interface = "eth0"
response, err := api(t, gateway, http.MethodPost, baseURL+"/api/nodes/skynet/01:02:03:04:05:06/creategateway", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusInternalServerError, response.StatusCode)
defer response.Body.Close()
var message models.ErrorResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusInternalServerError, message.Code)
assert.Equal(t, "IP Range Not Valid", message.Message)
})
t.Run("BadInterface", func(t *testing.T) {
gateway.RangeString = "0.0.0.0/0"
gateway.Interface = ""
response, err := api(t, gateway, http.MethodPost, baseURL+"/api/nodes/skynet/01:02:03:04:05:06/creategateway", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusInternalServerError, response.StatusCode)
defer response.Body.Close()
var message models.ErrorResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusInternalServerError, message.Code)
assert.Equal(t, "Interface cannot be empty", message.Message)
})
} }
func TestDeleteGateway(t *testing.T) { func TestDeleteGateway(t *testing.T) {
@@ -158,8 +339,173 @@ func TestUncordonNode(t *testing.T) {
} }
func TestCreateNode(t *testing.T) { func TestCreateNode(t *testing.T) {
setup(t)
key := createAccessKey(t)
t.Run("NodeExists", func(t *testing.T) {
var node models.Node
node.AccessKey = key.Value
node.MacAddress = "01:02:03:04:05:06"
node.Name = "myNode"
node.PublicKey = "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34="
node.Password = "tobedetermined"
node.LocalAddress = "192.168.0.1"
node.Endpoint = "10.100.100.4"
response, err := api(t, node, http.MethodPost, "http://localhost:8081:/api/nodes/skynet", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
defer response.Body.Close()
var message models.ErrorResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
assert.Contains(t, message.Message, "Field validation for 'MacAddress' failed on the 'macaddress_unique' tag")
})
t.Run("BadKey", func(t *testing.T) {
deleteNode(t)
var node models.Node
node.AccessKey = "badkey"
node.MacAddress = "01:02:03:04:05:06"
node.Name = "myNode"
node.PublicKey = "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34="
node.Password = "tobedetermined"
node.LocalAddress = "192.168.0.1"
node.Endpoint = "10.100.100.4"
response, err := api(t, node, http.MethodPost, "http://localhost:8081:/api/nodes/skynet", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
defer response.Body.Close()
var message models.ErrorResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusUnauthorized, response.StatusCode)
assert.Contains(t, message.Message, "W1R3: Key invalid, or none provided")
})
t.Run("BadMac", func(t *testing.T) {
var node models.Node
node.AccessKey = key.Value
node.MacAddress = "badmac"
node.Name = "myNode"
node.PublicKey = "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34="
node.Password = "tobedetermined"
node.LocalAddress = "192.168.0.1"
node.Endpoint = "10.100.100.4"
response, err := api(t, node, http.MethodPost, "http://localhost:8081:/api/nodes/skynet", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
defer response.Body.Close()
var message models.ErrorResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Field validation for 'MacAddress' failed on the 'macaddress_valid' tag")
})
t.Run("BadPublicKey", func(t *testing.T) {
var node models.Node
node.AccessKey = key.Value
node.MacAddress = "01:02:03:04:05:06"
node.Name = "myNode"
node.PublicKey = "xxx"
node.Password = "tobedetermined"
node.LocalAddress = "192.168.0.1"
node.Endpoint = "10.100.100.4"
response, err := api(t, node, http.MethodPost, "http://localhost:8081:/api/nodes/skynet", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
defer response.Body.Close()
var message models.ErrorResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Field validation for 'PublicKey' failed")
})
t.Run("BadPass", func(t *testing.T) {
var node models.Node
node.AccessKey = key.Value
node.MacAddress = "01:02:03:04:05:06"
node.Name = "myNode"
node.PublicKey = "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34="
node.Password = ""
node.LocalAddress = "192.168.0.1"
node.Endpoint = "10.100.100.4"
response, err := api(t, node, http.MethodPost, "http://localhost:8081:/api/nodes/skynet", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
defer response.Body.Close()
var message models.ErrorResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Error:Field validation for 'Password' failed")
})
t.Run("BadLocalAddress", func(t *testing.T) {
var node models.Node
node.AccessKey = key.Value
node.MacAddress = "01:02:03:04:05:06"
node.Name = "myNode"
node.PublicKey = "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34="
node.Password = "tobedetermined"
node.LocalAddress = "192.168.300.1"
node.Endpoint = "10.100.100.4"
response, err := api(t, node, http.MethodPost, "http://localhost:8081:/api/nodes/skynet", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
defer response.Body.Close()
var message models.ErrorResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Field validation for 'LocalAddress' failed")
})
t.Run("BadEndpoint", func(t *testing.T) {
var node models.Node
node.AccessKey = key.Value
node.MacAddress = "01:02:03:04:05:06"
node.Name = "myNode"
node.PublicKey = "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34="
node.Password = "tobedetermined"
node.LocalAddress = "192.168.0.1"
node.Endpoint = "10.100.300.4"
response, err := api(t, node, http.MethodPost, "http://localhost:8081:/api/nodes/skynet", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
defer response.Body.Close()
var message models.ErrorResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Contains(t, message.Message, "Field validation for 'Endpoint' failed")
})
t.Run("NetworkDoesNotExist", func(t *testing.T) {
deleteNetworks(t) deleteNetworks(t)
createNetwork(t) var node models.Node
node.AccessKey = key.Value
node.MacAddress = "01:02:03:04:05:06"
node.Name = "myNode"
node.PublicKey = "DM5qhLAE20PG9BbfBCger+Ac9D2NDOwCtY1rbYDLf34="
node.Password = "tobedetermined"
node.LocalAddress = "192.168.0.1"
node.Endpoint = "10.100.100.4"
response, err := api(t, node, http.MethodPost, "http://localhost:8081:/api/nodes/skynet", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusNotFound, response.StatusCode)
defer response.Body.Close()
var message models.ErrorResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusNotFound, message.Code)
assert.Contains(t, "W1R3: Network does not exist! ", message.Message)
})
t.Run("Valid", func(t *testing.T) {
setup(t)
deleteNode(t)
key := createAccessKey(t) key := createAccessKey(t)
var node models.Node var node models.Node
@@ -179,18 +525,30 @@ func TestCreateNode(t *testing.T) {
err = json.NewDecoder(response.Body).Decode(&message) err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.Equal(t, node.Name, message.Name) assert.Equal(t, node.Name, message.Name)
t.Log(message.Password) t.Log(message)
})
} }
func TestGetLastModified(t *testing.T) { func TestGetLastModified(t *testing.T) {
deleteNetworks(t)
createNetwork(t)
t.Run("Valid", func(t *testing.T) {
response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/adm/skynet/lastmodified", "secretkey") response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/adm/skynet/lastmodified", "secretkey")
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, response.StatusCode) assert.Equal(t, http.StatusOK, response.StatusCode)
assert.NotNil(t, response.Body, "no time returned") })
deleteNetworks(t)
t.Run("NoNetwork", func(t *testing.T) {
response, err := api(t, "", http.MethodGet, baseURL+"/api/nodes/adm/skynet/lastmodified", "secretkey")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusNotFound, response.StatusCode)
})
} }
func TestNodeAuthenticate(t *testing.T) { func TestNodeAuthenticate(t *testing.T) {
setup(t) setup(t)
t.Run("Valid", func(t *testing.T) {
var authRequest models.AuthParams var authRequest models.AuthParams
authRequest.MacAddress = "01:02:03:04:05:06" authRequest.MacAddress = "01:02:03:04:05:06"
authRequest.Password = "tobedetermined" authRequest.Password = "tobedetermined"
@@ -203,4 +561,61 @@ func TestNodeAuthenticate(t *testing.T) {
assert.Nil(t, err, err) assert.Nil(t, err, err)
assert.Equal(t, http.StatusOK, message.Code) assert.Equal(t, http.StatusOK, message.Code)
assert.Equal(t, "W1R3: Device 01:02:03:04:05:06 Authorized", message.Message) assert.Equal(t, "W1R3: Device 01:02:03:04:05:06 Authorized", message.Message)
})
t.Run("MacEmpty", func(t *testing.T) {
var authRequest models.AuthParams
authRequest.MacAddress = ""
authRequest.Password = "tobedetermined"
response, err := api(t, authRequest, http.MethodPost, "http://localhost:8081:/api/nodes/adm/skynet/authenticate", "")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
defer response.Body.Close()
var message models.SuccessResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Equal(t, "W1R3: MacAddress can't be empty", message.Message)
})
t.Run("EmptyPass", func(t *testing.T) {
var authRequest models.AuthParams
authRequest.MacAddress = "01:02:03:04:05:06"
authRequest.Password = ""
response, err := api(t, authRequest, http.MethodPost, "http://localhost:8081:/api/nodes/adm/skynet/authenticate", "")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
defer response.Body.Close()
var message models.SuccessResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Equal(t, "W1R3: Password can't be empty", message.Message)
})
t.Run("BadPass", func(t *testing.T) {
var authRequest models.AuthParams
authRequest.MacAddress = "01:02:03:04:05:06"
authRequest.Password = "badpass"
response, err := api(t, authRequest, http.MethodPost, "http://localhost:8081:/api/nodes/adm/skynet/authenticate", "")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
defer response.Body.Close()
var message models.SuccessResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Equal(t, "crypto/bcrypt: hashedPassword is not the hash of the given password", message.Message)
})
t.Run("BadMac", func(t *testing.T) {
var authRequest models.AuthParams
authRequest.MacAddress = "01:02:03:04:05:07"
authRequest.Password = "tobedetermined"
response, err := api(t, authRequest, http.MethodPost, "http://localhost:8081:/api/nodes/adm/skynet/authenticate", "")
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, response.StatusCode)
defer response.Body.Close()
var message models.SuccessResponse
err = json.NewDecoder(response.Body).Decode(&message)
assert.Nil(t, err, err)
assert.Equal(t, http.StatusBadRequest, message.Code)
assert.Equal(t, "mongo: no documents in result", message.Message)
})
} }