test logic update

This commit is contained in:
0xdcarns 2022-01-10 19:43:20 -05:00 committed by Matthew R Kasun
parent c6a2f2cb64
commit edbce5547a

View file

@ -19,12 +19,12 @@ func TestCreateEgressGateway(t *testing.T) {
t.Run("NoNodes", func(t *testing.T) {
node, err := logic.CreateEgressGateway(gateway)
assert.Equal(t, models.Node{}, node)
assert.EqualError(t, err, "unable to get record key")
assert.EqualError(t, err, "could not find any records")
})
t.Run("Success", func(t *testing.T) {
testnode := createTestNode()
gateway.NetID = "skynet"
gateway.NodeID = testnode.MacAddress
gateway.NodeID = testnode.ID
node, err := logic.CreateEgressGateway(gateway)
assert.Nil(t, err)
@ -43,7 +43,7 @@ func TestDeleteEgressGateway(t *testing.T) {
gateway.Interface = "eth0"
gateway.Ranges = []string{"10.100.100.0/24"}
gateway.NetID = "skynet"
gateway.NodeID = testnode.MacAddress
gateway.NodeID = testnode.ID
t.Run("Success", func(t *testing.T) {
node, err := logic.CreateEgressGateway(gateway)
assert.Nil(t, err)