File: vpc_test.go

package info (click to toggle)
golang-github-linode-linodego 1.55.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 13,112 kB
  • sloc: makefile: 96; sh: 52; python: 24
file content (122 lines) | stat: -rw-r--r-- 3,687 bytes parent folder | download
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
package unit

import (
	"context"
	"testing"
	"time"

	"github.com/linode/linodego"
	"github.com/stretchr/testify/assert"
)

func TestVPC_Create(t *testing.T) {
	var base ClientBaseCase
	base.SetUp(t)
	defer base.TearDown(t)

	base.MockPost("vpcs", linodego.VPC{
		ID:          123,
		Label:       "test-vpc",
		Description: "Test VPC description",
		Region:      "us-east",
		Subnets: []linodego.VPCSubnet{
			{ID: 1, Label: "subnet-1"},
			{ID: 2, Label: "subnet-2"},
		},
		Created: linodego.Pointer(time.Now()),
		Updated: linodego.Pointer(time.Now()),
	})

	vpc, err := base.Client.CreateVPC(context.Background(), linodego.VPCCreateOptions{
		Label:       "test-vpc",
		Description: "Test VPC description",
		Region:      "us-east",
		Subnets: []linodego.VPCSubnetCreateOptions{
			{Label: "subnet-1"},
			{Label: "subnet-2"},
		},
	})
	assert.NoError(t, err, "Expected no error when creating VPC")
	assert.NotNil(t, vpc, "Expected VPC to be created")
	assert.Equal(t, "test-vpc", vpc.Label, "Expected VPC label to match")
	assert.Equal(t, "us-east", vpc.Region, "Expected VPC region to match")
	assert.Len(t, vpc.Subnets, 2, "Expected VPC to have 2 subnets")
}

func TestVPC_Get(t *testing.T) {
	var base ClientBaseCase
	base.SetUp(t)
	defer base.TearDown(t)

	mockVPC := linodego.VPC{
		ID:    123,
		Label: "test-vpc",
	}
	base.MockGet("vpcs/123", mockVPC)

	vpc, err := base.Client.GetVPC(context.Background(), 123)
	assert.NoError(t, err, "Expected no error when getting VPC")
	assert.NotNil(t, vpc, "Expected non-nil VPC")
	assert.Equal(t, mockVPC.ID, vpc.ID, "Expected VPC ID to match")
	assert.Equal(t, mockVPC.Label, vpc.Label, "Expected VPC label to match")
}

func TestVPC_List(t *testing.T) {
	fixtureData, err := fixtures.GetFixture("vpc_list")
	assert.NoError(t, err)

	var base ClientBaseCase
	base.SetUp(t)
	defer base.TearDown(t)

	base.MockGet("vpcs", fixtureData)

	vpcs, err := base.Client.ListVPCs(context.Background(), &linodego.ListOptions{})
	assert.NoError(t, err)

	assert.NotEmpty(t, vpcs, "Expected non-empty VPC list")

	assert.Equal(t, 123, vpcs[0].ID, "Expected VPC ID to match")
	assert.Equal(t, "test-vpc", vpcs[0].Label, "Expected VPC label to match")
	assert.Equal(t, "Test VPC description", vpcs[0].Description, "Expected VPC description to match")
	assert.Equal(t, "us-east", vpcs[0].Region, "Expected VPC region to match")
	assert.NotEmpty(t, vpcs[0].Subnets, "Expected VPC to have subnets")
	assert.Equal(t, 456, vpcs[0].Subnets[0].ID, "Expected subnet ID to match")
	assert.Equal(t, "subnet-1", vpcs[0].Subnets[0].Label, "Expected subnet label to match")
	assert.Equal(t, "192.168.1.0/24", vpcs[0].Subnets[0].IPv4, "Expected subnet IPv4 to match")
}

func TestVPC_Update(t *testing.T) {
	var base ClientBaseCase
	base.SetUp(t)
	defer base.TearDown(t)

	updatedMockVPC := linodego.VPC{
		ID:          123,
		Label:       "updated-vpc",
		Description: "Updated description",
	}
	base.MockPut("vpcs/123", updatedMockVPC)

	opts := linodego.VPCUpdateOptions{
		Label:       "updated-vpc",
		Description: "Updated description",
	}

	vpc, err := base.Client.UpdateVPC(context.Background(), 123, opts)
	assert.NoError(t, err, "Expected no error when updating VPC")
	assert.NotNil(t, vpc, "Expected non-nil updated VPC")
	assert.Equal(t, "updated-vpc", vpc.Label, "Expected VPC label to match")
	assert.Equal(t, "Updated description", vpc.Description, "Expected VPC description to match")
}

func TestVPC_Delete(t *testing.T) {
	var base ClientBaseCase
	base.SetUp(t)
	defer base.TearDown(t)

	base.MockDelete("vpcs/123", nil)

	err := base.Client.DeleteVPC(context.Background(), 123)
	assert.NoError(t, err, "Expected no error when deleting VPC")
}