zishuo/netmaker
1
0
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-10-06 09:22:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix TestMain

Browse Source
This commit is contained in:
Matthew R Kasun
2023-03-07 14:45:44 -05:00
parent ed093a3bb6
commit 1de0c14b83
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
logic/host_test.go
View File

@@ -13,7 +13,7 @@ import (
"github.com/matryer/is" "github.com/matryer/is"
) )
func TestCheckPorts(t *testing.T) { func TestMain(m *testing.M) {
database.InitializeDatabase() database.InitializeDatabase()
defer database.CloseDB() defer database.CloseDB()
peerUpdate := make(chan *models.Node) peerUpdate := make(chan *models.Node)
@@ -28,6 +28,7 @@ func TestCheckPorts(t *testing.T) {
os.Exit(m.Run()) os.Exit(m.Run())
} }
func TestCheckPorts(t *testing.T) {
h := models.Host{ h := models.Host{
ID: uuid.New(), ID: uuid.New(),
EndpointIP: net.ParseIP("192.168.1.1"), EndpointIP: net.ParseIP("192.168.1.1"),
@@ -44,6 +45,8 @@ func TestCheckPorts(t *testing.T) {
t.Run("no change", func(t *testing.T) { t.Run("no change", func(t *testing.T) {
is := is.New(t) is := is.New(t)
CheckHostPorts(&testHost) CheckHostPorts(&testHost)
t.Log(testHost.ListenPort, testHost.ProxyListenPort)
t.Log(h.ListenPort, h.ProxyListenPort)
is.Equal(testHost.ListenPort, 51830) is.Equal(testHost.ListenPort, 51830)
is.Equal(testHost.ProxyListenPort, 51730) is.Equal(testHost.ProxyListenPort, 51730)
}) })
@@ -51,6 +54,8 @@ func TestCheckPorts(t *testing.T) {
is := is.New(t) is := is.New(t)
testHost.ListenPort = 51821 testHost.ListenPort = 51821
CheckHostPorts(&testHost) CheckHostPorts(&testHost)
t.Log(testHost.ListenPort, testHost.ProxyListenPort)
t.Log(h.ListenPort, h.ProxyListenPort)
is.Equal(testHost.ListenPort, 51822) is.Equal(testHost.ListenPort, 51822)
is.Equal(testHost.ProxyListenPort, 51730) is.Equal(testHost.ProxyListenPort, 51730)
}) })
@@ -58,6 +63,8 @@ func TestCheckPorts(t *testing.T) {
is := is.New(t) is := is.New(t)
testHost.ProxyListenPort = 65535 testHost.ProxyListenPort = 65535
CheckHostPorts(&testHost) CheckHostPorts(&testHost)
t.Log(testHost.ListenPort, testHost.ProxyListenPort)
t.Log(h.ListenPort, h.ProxyListenPort)
is.Equal(testHost.ListenPort, 51822) is.Equal(testHost.ListenPort, 51822)
is.Equal(testHost.ProxyListenPort, minPort) is.Equal(testHost.ProxyListenPort, minPort)
}) })
@@ -65,6 +72,8 @@ func TestCheckPorts(t *testing.T) {
is := is.New(t) is := is.New(t)
testHost.ListenPort = maxPort testHost.ListenPort = maxPort
CheckHostPorts(&testHost) CheckHostPorts(&testHost)
t.Log(testHost.ListenPort, testHost.ProxyListenPort)
t.Log(h.ListenPort, h.ProxyListenPort)
is.Equal(testHost.ListenPort, minPort) is.Equal(testHost.ListenPort, minPort)
is.Equal(testHost.ProxyListenPort, minPort+1) is.Equal(testHost.ProxyListenPort, minPort+1)
}) })
@@ -72,6 +81,8 @@ func TestCheckPorts(t *testing.T) {
is := is.New(t) is := is.New(t)
testHost.ProxyListenPort = 51821 testHost.ProxyListenPort = 51821
CheckHostPorts(&testHost) CheckHostPorts(&testHost)
t.Log(testHost.ListenPort, testHost.ProxyListenPort)
t.Log(h.ListenPort, h.ProxyListenPort)
is.Equal(testHost.ListenPort, minPort) is.Equal(testHost.ListenPort, minPort)
is.Equal(testHost.ProxyListenPort, 51822) is.Equal(testHost.ProxyListenPort, 51822)
}) })