1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
|
package unit
import (
"context"
"fmt"
"testing"
"github.com/jarcoal/httpmock"
"github.com/linode/linodego"
"github.com/stretchr/testify/assert"
)
func TestFirewallDevice_List(t *testing.T) {
fixtureData, err := fixtures.GetFixture("firewall_device_list")
assert.NoError(t, err)
var base ClientBaseCase
base.SetUp(t)
defer base.TearDown(t)
firewallID := 123
base.MockGet(formatMockAPIPath("networking/firewalls/%d/devices", firewallID), fixtureData)
firewallDevices, err := base.Client.ListFirewallDevices(context.Background(), firewallID, &linodego.ListOptions{})
assert.NoError(t, err)
assert.Equal(t, 2, len(firewallDevices))
for _, device := range firewallDevices {
assert.NotNil(t, device.Entity)
assert.NotEqual(t, 0, device.ID)
assert.NotNil(t, device.Created)
assert.NotNil(t, device.Updated)
switch device.Entity.Type {
case "linode":
assert.Equal(t, 123, device.Entity.ID)
assert.Equal(t, "my-linode", device.Entity.Label)
assert.Equal(t, "/v4/linode/instances/123", device.Entity.URL)
case "nodebalancer":
assert.Equal(t, 321, device.Entity.ID)
assert.Equal(t, "my-nodebalancer", device.Entity.Label)
assert.Equal(t, "/v4/nodebalancers/123", device.Entity.URL)
default:
t.Fatalf("Unexpected device type: %s", device.Entity.Type)
}
}
}
func TestFirewallDevice_Get(t *testing.T) {
fixtureData, err := fixtures.GetFixture("firewall_device_get")
assert.NoError(t, err)
var base ClientBaseCase
base.SetUp(t)
defer base.TearDown(t)
firewallID := 123
deviceID := 123
base.MockGet(formatMockAPIPath("networking/firewalls/%d/devices/%d", firewallID, deviceID), fixtureData)
firewallDevice, err := base.Client.GetFirewallDevice(context.Background(), firewallID, deviceID)
assert.NoError(t, err)
assert.NotNil(t, firewallDevice)
assert.Equal(t, deviceID, firewallDevice.ID)
assert.NotNil(t, firewallDevice.Entity)
assert.Equal(t, 123, firewallDevice.Entity.ID)
assert.Equal(t, "my-linode", firewallDevice.Entity.Label)
assert.Equal(t, linodego.FirewallDeviceType("linode"), firewallDevice.Entity.Type)
assert.Equal(t, "/v4/linode/instances/123", firewallDevice.Entity.URL)
assert.NotNil(t, firewallDevice.Created)
assert.NotNil(t, firewallDevice.Updated)
}
func TestFirewallDevice_Create(t *testing.T) {
fixtureData, err := fixtures.GetFixture("firewall_device_create")
assert.NoError(t, err)
var base ClientBaseCase
base.SetUp(t)
defer base.TearDown(t)
firewallID := 123
requestData := linodego.FirewallDeviceCreateOptions{
ID: 123,
Type: "linode",
}
base.MockPost(formatMockAPIPath("networking/firewalls/%d/devices", firewallID), fixtureData)
firewallDevice, err := base.Client.CreateFirewallDevice(context.Background(), firewallID, requestData)
assert.NoError(t, err)
assert.NotNil(t, firewallDevice)
assert.NotNil(t, firewallDevice.Entity)
assert.Equal(t, 123, firewallDevice.Entity.ID)
assert.Equal(t, "my-linode", firewallDevice.Entity.Label)
assert.Equal(t, linodego.FirewallDeviceType("linode"), firewallDevice.Entity.Type)
assert.Equal(t, "/v4/linode/instances/123", firewallDevice.Entity.URL)
assert.NotNil(t, firewallDevice.Created)
assert.NotNil(t, firewallDevice.Updated)
}
func TestFirewallDevice_Delete(t *testing.T) {
client := createMockClient(t)
firewallID := 123
deviceID := 123
httpmock.RegisterRegexpResponder("DELETE", mockRequestURL(t, fmt.Sprintf("networking/firewalls/%d/devices/%d", firewallID, deviceID)),
httpmock.NewStringResponder(200, "{}"))
if err := client.DeleteFirewallDevice(context.Background(), firewallID, deviceID); err != nil {
t.Fatal(err)
}
}
|