fix TestGetNetworkNodes

This commit is contained in:
Matthew R Kasun 2023-03-08 17:50:21 -05:00
parent 8ab81ca2da
commit 3aa6571360
2 changed files with 3 additions and 6 deletions

View file

@ -327,10 +327,7 @@ func TestIpv6Network(t *testing.T) {
func deleteAllNetworks() { func deleteAllNetworks() {
deleteAllNodes() deleteAllNodes()
nets, _ := logic.GetNetworks() database.DeleteAllRecords(database.NETWORKS_TABLE_NAME)
for _, net := range nets {
logic.DeleteNetwork(net.NetID)
}
} }
func createNet() { func createNet() {

View file

@ -113,12 +113,12 @@ func TestGetNetworkNodes(t *testing.T) {
t.Run("BadNet", func(t *testing.T) { t.Run("BadNet", func(t *testing.T) {
node, err := logic.GetNetworkNodes("badnet") node, err := logic.GetNetworkNodes("badnet")
assert.Nil(t, err) assert.Nil(t, err)
assert.Nil(t, node) assert.Equal(t, []models.Node{}, node)
}) })
t.Run("NoNodes", func(t *testing.T) { t.Run("NoNodes", func(t *testing.T) {
node, err := logic.GetNetworkNodes("skynet") node, err := logic.GetNetworkNodes("skynet")
assert.Nil(t, err) assert.Nil(t, err)
assert.Nil(t, node) assert.Equal(t, []models.Node{}, node)
}) })
t.Run("Success", func(t *testing.T) { t.Run("Success", func(t *testing.T) {
createTestNode() createTestNode()