From 85c7bc3a7a066725340c7820db63d17306915b74 Mon Sep 17 00:00:00 2001
From: vshanthe <vshanthe@akamai.com>
Date: Fri, 27 Dec 2024 11:02:22 +0530
Subject: [PATCH] vlan_tests

---
 test/unit/fixtures/vlan_get_ipam_address.json | 15 +++++
 test/unit/fixtures/vlans_list.json            | 13 ++++
 test/unit/vlan_test.go                        | 65 +++++++++++++++++++
 3 files changed, 93 insertions(+)
 create mode 100644 test/unit/fixtures/vlan_get_ipam_address.json
 create mode 100644 test/unit/fixtures/vlans_list.json
 create mode 100644 test/unit/vlan_test.go

diff --git a/test/unit/fixtures/vlan_get_ipam_address.json b/test/unit/fixtures/vlan_get_ipam_address.json
new file mode 100644
index 000000000..d294d9f8f
--- /dev/null
+++ b/test/unit/fixtures/vlan_get_ipam_address.json
@@ -0,0 +1,15 @@
+{
+  "data": [
+    {
+      "interfaces": [
+        {
+          "label": "test-vlan",
+          "ipam_address": "10.0.0.1/24"
+        }
+      ]
+    }
+  ],
+  "page": 1,
+  "pages": 1,
+  "results": 1
+}
diff --git a/test/unit/fixtures/vlans_list.json b/test/unit/fixtures/vlans_list.json
new file mode 100644
index 000000000..5572e8074
--- /dev/null
+++ b/test/unit/fixtures/vlans_list.json
@@ -0,0 +1,13 @@
+{
+  "data": [
+    {
+      "label": "test-vlan",
+      "linodes": [12345],
+      "region": "us-east",
+      "created": "2024-12-01T12:00:00"
+    }
+  ],
+  "page": 1,
+  "pages": 1,
+  "results": 1
+}
diff --git a/test/unit/vlan_test.go b/test/unit/vlan_test.go
new file mode 100644
index 000000000..f8aca13a5
--- /dev/null
+++ b/test/unit/vlan_test.go
@@ -0,0 +1,65 @@
+package unit
+
+import (
+	"context"
+	"fmt"
+	"github.com/linode/linodego"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestVLAN_List(t *testing.T) {
+	// Load the fixture data for VLANs
+	fixtureData, err := fixtures.GetFixture("vlans_list")
+	assert.NoError(t, err)
+
+	var base ClientBaseCase
+	base.SetUp(t)
+	defer base.TearDown(t)
+
+	// Mock the GET request
+	base.MockGet("networking/vlans", fixtureData)
+
+	vlans, err := base.Client.ListVLANs(context.Background(), &linodego.ListOptions{})
+	assert.NoError(t, err)
+	assert.NotEmpty(t, vlans, "Expected non-empty VLAN list")
+
+	// Verify a specific VLAN exists in the list
+	var testVLAN *linodego.VLAN
+	for _, v := range vlans {
+		if v.Label == "test-vlan" {
+			testVLAN = &v
+			break
+		}
+	}
+
+	if testVLAN == nil {
+		t.Errorf("Expected VLAN 'test-vlan' to be in the response, but it was not found")
+	} else {
+		assert.Equal(t, "us-east", testVLAN.Region, "Expected region to be 'us-east'")
+		assert.Contains(t, testVLAN.Linodes, 12345, "Expected Linodes to include 12345")
+		assert.NotNil(t, testVLAN.Created, "Expected 'test-vlan' to have a created timestamp")
+	}
+}
+
+func TestVLAN_GetIPAMAddress(t *testing.T) {
+	// Load the fixture data for VLAN IPAM address
+	fixtureData, err := fixtures.GetFixture("vlan_get_ipam_address")
+	assert.NoError(t, err)
+
+	var base ClientBaseCase
+	base.SetUp(t)
+	defer base.TearDown(t)
+
+	linodeID := 12345
+	vlanLabel := "test-vlan"
+	// Mock the GET request
+	base.MockGet(fmt.Sprintf("linode/instances/%d/configs", linodeID), fixtureData)
+
+	ipamAddress, err := base.Client.GetVLANIPAMAddress(context.Background(), linodeID, vlanLabel)
+	assert.NoError(t, err)
+	assert.NotEmpty(t, ipamAddress, "Expected non-empty IPAM address")
+
+	// Verify the returned IPAM address
+	assert.Equal(t, "10.0.0.1/24", ipamAddress, "Expected IPAM address to be '10.0.0.1/24'")
+}