fixed tests

This commit is contained in:
Matthew R Kasun
2021-05-09 13:31:50 -04:00
parent 073be947c6
commit bb2bb0e1ee
8 changed files with 51 additions and 42 deletions

View File

@@ -0,0 +1,8 @@
skynet {
reload 15s
hosts /root/dnsconfig/netmaker.hosts {
fallthrough
}
forward . 8.8.8.8 8.8.4.4
log
}

View File

@@ -0,0 +1 @@
10.0.0.2 myhost.skynet

View File

@@ -1,7 +1,6 @@
package controller package controller
import ( import (
"io/ioutil"
"testing" "testing"
"github.com/gravitl/netmaker/models" "github.com/gravitl/netmaker/models"
@@ -66,18 +65,7 @@ func TestDeleteDNS(t *testing.T) {
}) })
} }
func TestWriteHosts(t *testing.T) {
err := WriteHosts()
assert.Nil(t, err)
files, err := ioutil.ReadDir("./config")
assert.Nil(t, err)
for _, file := range files {
if file.Name() == "netmaker.hosts" {
return
}
}
t.Fail()
}
func TestValidateDNSUpdate(t *testing.T) { func TestValidateDNSUpdate(t *testing.T) {
entry := models.DNSEntry{"10.0.0.2", "myhost", "skynet"} entry := models.DNSEntry{"10.0.0.2", "myhost", "skynet"}
_, _ = DeleteDNS("mynode", "skynet") _, _ = DeleteDNS("mynode", "skynet")

View File

@@ -604,14 +604,14 @@ func CreateAccessKey(accesskey models.AccessKey, network models.Network) (models
} }
} }
netID := params["networkname"] netID := network.NetID
address := servercfg.GetGRPCHost() + ":" + servercfg.GetGRPCPort() address := servercfg.GetGRPCHost() + ":" + servercfg.GetGRPCPort()
accessstringdec := address + "|" + netID + "|" + accesskey.Value + "|" + privAddr accessstringdec := address + "|" + netID + "|" + accesskey.Value + "|" + privAddr
accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(accessstringdec)) accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(accessstringdec))
//validate accesskey //validate accesskey
v := validator.New() v := validator.New()
err = v.Struct(accesskey) err := v.Struct(accesskey)
if err != nil { if err != nil {
for _, e := range err.(validator.ValidationErrors) { for _, e := range err.(validator.ValidationErrors) {
fmt.Println(e) fmt.Println(e)

View File

@@ -223,7 +223,7 @@ func (s *NodeServiceServer) UpdateNode(ctx context.Context, req *nodepb.UpdateNo
// Get the node data from the request // Get the node data from the request
data := req.GetNode() data := req.GetNode()
// Now we have to convert this into a NodeItem type to convert into BSON // Now we have to convert this into a NodeItem type to convert into BSON
nodechange := models.Node{ nodechange := models.NodeUpdate{
// ID: primitive.NilObjectID, // ID: primitive.NilObjectID,
MacAddress: data.GetMacaddress(), MacAddress: data.GetMacaddress(),
Name: data.GetName(), Name: data.GetName(),

View File

@@ -1,36 +1,39 @@
package functions package functions
import ( import (
"fmt" "fmt"
"path/filepath"
"log"
"os"
"io/ioutil" "io/ioutil"
"log"
"os"
) )
func FileExists(f string) bool { func FileExists(f string) bool {
info, err := os.Stat(f) info, err := os.Stat(f)
if os.IsNotExist(err) { if os.IsNotExist(err) {
return false return false
} }
return !info.IsDir() return !info.IsDir()
} }
func SetCorefile(domains string) error { func SetCorefile(domains string) error {
dir, err := filepath.Abs(filepath.Dir(os.Args[0])) //does not work when executing tests
//dir, err := filepath.Abs(filepath.Dir(os.Args[0]))
//if err != nil {
// return err
//}
dir, err := os.Getwd()
if err != nil { if err != nil {
return err return err
} }
_, err = os.Stat(dir + "/config/dnsconfig") _, err = os.Stat(dir + "/config/dnsconfig")
if os.IsNotExist(err) { if os.IsNotExist(err) {
os.Mkdir(dir +"/config/dnsconfig", 744) os.Mkdir(dir+"/config/dnsconfig", 744)
} else if err != nil { } else if err != nil {
fmt.Println("couldnt find or create /config/dnsconfig") fmt.Println("couldnt find or create /config/dnsconfig")
return err return err
} }
corefile := domains + ` { corefile := domains + ` {
reload 15s reload 15s
hosts /root/dnsconfig/netmaker.hosts { hosts /root/dnsconfig/netmaker.hosts {
fallthrough fallthrough
@@ -39,13 +42,13 @@ func SetCorefile(domains string) error {
log log
} }
` `
corebytes := []byte(corefile) corebytes := []byte(corefile)
err = ioutil.WriteFile(dir + "/config/dnsconfig/Corefile", corebytes, 0644) err = ioutil.WriteFile(dir+"/config/dnsconfig/Corefile", corebytes, 0644)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
log.Println("") log.Println("")
return err return err
} }
return err return err
} }

View File

@@ -0,0 +1,8 @@
skynet {
reload 15s
hosts /root/dnsconfig/netmaker.hosts {
fallthrough
}
forward . 8.8.8.8 8.8.4.4
log
}

View File

@@ -0,0 +1 @@
10.71.0.1 mynode.skynet