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
import (
"io/ioutil"
"testing"
"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) {
entry := models.DNSEntry{"10.0.0.2", "myhost", "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()
accessstringdec := address + "|" + netID + "|" + accesskey.Value + "|" + privAddr
accesskey.AccessString = base64.StdEncoding.EncodeToString([]byte(accessstringdec))
//validate accesskey
v := validator.New()
err = v.Struct(accesskey)
err := v.Struct(accesskey)
if err != nil {
for _, e := range err.(validator.ValidationErrors) {
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
data := req.GetNode()
// Now we have to convert this into a NodeItem type to convert into BSON
nodechange := models.Node{
nodechange := models.NodeUpdate{
// ID: primitive.NilObjectID,
MacAddress: data.GetMacaddress(),
Name: data.GetName(),

View file

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