update TestMain

This commit is contained in:
Matthew R Kasun
2023-02-25 10:48:36 -05:00
parent 3244472481
commit 3836bd1d36
4 changed files with 9 additions and 0 deletions

View File

@@ -40,6 +40,8 @@ func TestMain(m *testing.M) {
logger.Log(3, "received node update", update.Action) logger.Log(3, "received node update", update.Action)
} }
}() }()
os.Exit(m.Run())
} }
func TestCreateNetwork(t *testing.T) { func TestCreateNetwork(t *testing.T) {

View File

@@ -3,6 +3,7 @@ package functions
import ( import (
"context" "context"
"encoding/json" "encoding/json"
"os"
"testing" "testing"
"github.com/gravitl/netmaker/database" "github.com/gravitl/netmaker/database"
@@ -39,6 +40,8 @@ func TestMain(m *testing.M) {
logger.Log(3, "received node update", update.Action) logger.Log(3, "received node update", update.Action)
} }
}() }()
os.Exit(m.Run())
} }
func TestNetworkExists(t *testing.T) { func TestNetworkExists(t *testing.T) {

View File

@@ -3,6 +3,7 @@ package logic
import ( import (
"context" "context"
"net" "net"
"os"
"testing" "testing"
"github.com/google/uuid" "github.com/google/uuid"
@@ -30,6 +31,7 @@ func TestMain(m *testing.M) {
logger.Log(3, "received node update", update.Action) logger.Log(3, "received node update", update.Action)
} }
}() }()
os.Exit(m.Run())
} }
func TestCheckPorts(t *testing.T) { func TestCheckPorts(t *testing.T) {

View File

@@ -1,6 +1,7 @@
package pro package pro
import ( import (
"os"
"testing" "testing"
"github.com/google/uuid" "github.com/google/uuid"
@@ -13,6 +14,7 @@ import (
func TestMain(m *testing.M) { func TestMain(m *testing.M) {
database.InitializeDatabase() database.InitializeDatabase()
defer database.CloseDB() defer database.CloseDB()
os.Exit(m.Run())
} }
func TestNetworkUserLogic(t *testing.T) { func TestNetworkUserLogic(t *testing.T) {