Ver código fonte

set dns dir when testing

Matthew R Kasun 2 anos atrás
pai
commit
13c379f751
1 arquivos alterados com 5 adições e 0 exclusões
  1. 5 0
      controllers/dns_test.go

+ 5 - 0
controllers/dns_test.go

@@ -9,6 +9,7 @@ import (
 	"github.com/stretchr/testify/assert"
 	"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
 
+	"github.com/gravitl/netmaker/functions"
 	"github.com/gravitl/netmaker/logic"
 	"github.com/gravitl/netmaker/models"
 )
@@ -50,6 +51,8 @@ func TestGetNodeDNS(t *testing.T) {
 	deleteAllNetworks()
 	createNet()
 	createHost()
+	err := functions.SetDNSDir()
+	assert.Nil(t, err)
 	t.Run("NoNodes", func(t *testing.T) {
 		dns, _ := logic.GetNodeDNS("skynet")
 		assert.Equal(t, []models.DNSEntry(nil), dns)
@@ -204,6 +207,8 @@ func TestSetDNS(t *testing.T) {
 	deleteAllNetworks()
 	etc, err := os.Stat("/etc/hosts")
 	assert.Nil(t, err)
+	err = functions.SetDNSDir()
+	assert.Nil(t, err)
 	t.Run("NoNetworks", func(t *testing.T) {
 		err := logic.SetDNS()
 		assert.Nil(t, err)