gravitl/netmaker
1
1
Fork 0
mirror of https://github.com/gravitl/netmaker.git synced 2025-09-11 23:54:22 +08:00
Code Issues Projects Releases Packages 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
Show stats 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/logger"
"github.com/gravitl/netmaker/logic" "github.com/gravitl/netmaker/logic"
"github.com/gravitl/netmaker/models" "github.com/gravitl/netmaker/models"
"github.com/stretchr/testify/assert"
) )
var ( var (
@ -46,36 +47,23 @@ func TestMain(m *testing.M) {
func TestNetworkExists(t *testing.T) { func TestNetworkExists(t *testing.T) {
database.DeleteRecord(database.NETWORKS_TABLE_NAME, testNetwork.NetID) database.DeleteRecord(database.NETWORKS_TABLE_NAME, testNetwork.NetID)
defer database.CloseDB()
exists, err := logic.NetworkExists(testNetwork.NetID) exists, err := logic.NetworkExists(testNetwork.NetID)
if err == nil { assert.NotNil(t, err)
t.Fatalf("expected error, received nil") assert.False(t, exists)
}
if exists {
t.Fatalf("expected false")
}
err = logic.SaveNetwork(testNetwork) err = logic.SaveNetwork(testNetwork)
if err != nil { assert.Nil(t, err)
t.Fatalf("failed to save test network in databse: %s", err)
}
exists, err = logic.NetworkExists(testNetwork.NetID) exists, err = logic.NetworkExists(testNetwork.NetID)
if err != nil { assert.Nil(t, err)
t.Fatalf("expected nil, received err: %s", err) assert.NotNil(t, exists)
}
if !exists {
t.Fatalf("expected network to exist in database")
}
err = database.DeleteRecord(database.NETWORKS_TABLE_NAME, testNetwork.NetID) err = database.DeleteRecord(database.NETWORKS_TABLE_NAME, testNetwork.NetID)
if err != nil { assert.Nil(t, err)
t.Fatalf("expected nil, failed to delete test network: %s", err)
}
} }
func TestGetAllExtClients(t *testing.T) { func TestGetAllExtClients(t *testing.T) {
defer database.CloseDB() err := database.DeleteRecord(database.EXT_CLIENT_TABLE_NAME, testExternalClient.ClientID)
database.DeleteRecord(database.EXT_CLIENT_TABLE_NAME, testExternalClient.ClientID) assert.Nil(t, err)
extClients, err := GetAllExtClients() extClients, err := GetAllExtClients()
if err == nil { if err == nil {