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

fix helper tests

Browse Source
This commit is contained in:
Matthew R Kasun
2023-02-25 11:06:55 -05:00
parent b9126c7fa3
commit a39fa7e065
1 changed files with 9 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
functions/helpers_test.go
View File

@@ -10,6 +10,7 @@ import (
"github.com/gravitl/netmaker/logger"
"github.com/gravitl/netmaker/logic"
"github.com/gravitl/netmaker/models"
"github.com/stretchr/testify/assert"
)
var (
@@ -46,36 +47,23 @@ func TestMain(m *testing.M) {
func TestNetworkExists(t *testing.T) {
database.DeleteRecord(database.NETWORKS_TABLE_NAME, testNetwork.NetID)
defer database.CloseDB()
exists, err := logic.NetworkExists(testNetwork.NetID)
if err == nil {
t.Fatalf("expected error, received nil")
}
if exists {
t.Fatalf("expected false")
}
assert.NotNil(t, err)
assert.False(t, exists)
err = logic.SaveNetwork(testNetwork)
if err != nil {
t.Fatalf("failed to save test network in databse: %s", err)
}
assert.Nil(t, err)
exists, err = logic.NetworkExists(testNetwork.NetID)
if err != nil {
t.Fatalf("expected nil, received err: %s", err)
}
if !exists {
t.Fatalf("expected network to exist in database")
}
assert.Nil(t, err)
assert.NotNil(t, exists)
err = database.DeleteRecord(database.NETWORKS_TABLE_NAME, testNetwork.NetID)
if err != nil {
t.Fatalf("expected nil, failed to delete test network: %s", err)
}
assert.Nil(t, err)
}
func TestGetAllExtClients(t *testing.T) {
defer database.CloseDB()
database.DeleteRecord(database.EXT_CLIENT_TABLE_NAME, testExternalClient.ClientID)
err := database.DeleteRecord(database.EXT_CLIENT_TABLE_NAME, testExternalClient.ClientID)
assert.Nil(t, err)
extClients, err := GetAllExtClients()
if err == nil {