add test for bgp custom id map

Author: Rahul Sharma

cloud/linode/cilium_loadbalancers_test.go

@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"net"
+	"os"
 	"testing"
 
 	k8sClient "github.com/cilium/cilium/pkg/k8s/client"
@@ -201,6 +202,7 @@ func createNewIpHolderInstance() linodego.Instance {
 func testNoBGPNodeLabel(t *testing.T, mc *mocks.MockClient) {
 	Options.BGPNodeSelector = ""
 	Options.IpHolderSuffix = "linodelb"
+	os.Setenv("BGP_PEER_PREFIX", "2600:3cef")
 	svc := createTestService()
 	newIpHolderInstance = createNewIpHolderInstance()
 
@@ -244,6 +246,7 @@ func testNoBGPNodeLabel(t *testing.T, mc *mocks.MockClient) {
 	if lbStatus == nil {
 		t.Fatal("expected non-nil lbStatus")
 	}
+	os.Unsetenv("BGP_PEER_PREFIX")
 }
 
 func testUnsupportedRegion(t *testing.T, mc *mocks.MockClient) {
@@ -257,11 +260,23 @@ func testUnsupportedRegion(t *testing.T, mc *mocks.MockClient) {
 
 	lbStatus, err := lb.EnsureLoadBalancer(context.TODO(), "linodelb", svc, nodes)
 	if err == nil {
-		t.Fatal("expected nil error")
+		t.Fatal("expected not nil error")
 	}
 	if lbStatus != nil {
 		t.Fatalf("expected a nil lbStatus, got %v", lbStatus)
 	}
+
+	// Use BGP custom id map
+	os.Setenv("BGP_CUSTOM_ID_MAP", "{'us-foobar': 2}")
+	lb = &loadbalancers{mc, zone, kubeClient, ciliumClient, ciliumLBType}
+	lbStatus, err = lb.EnsureLoadBalancer(context.TODO(), "linodelb", svc, nodes)
+	if err == nil {
+		t.Fatal("expected not nil error")
+	}
+	if lbStatus != nil {
+		t.Fatalf("expected a nil lbStatus, got %v", lbStatus)
+	}
+	os.Unsetenv("BGP_CUSTOM_ID_MAP")
 }
 
 func testCreateWithExistingIPHolderWithOldIpHolderNamingConvention(t *testing.T, mc *mocks.MockClient) {