test: LbaasV2 method getSubnetID

kubernetes · Oct 9, 2023 · 330dcb7 · 330dcb7
1 parent 31d4983
commit 330dcb7
Showing 1 changed file with 147 additions and 0 deletions.
diff --git a/pkg/openstack/loadbalancer_test.go b/pkg/openstack/loadbalancer_test.go
@@ -1,6 +1,7 @@
 package openstack
 
 import (
+	"fmt"
 	"reflect"
 	"sort"
 	"testing"
@@ -787,3 +788,149 @@ func Test_buildPoolCreateOpt(t *testing.T) {
 		})
 	}
 }
+
+func Test_getSubnetID(t *testing.T) {
+	type args struct {
+		svcConf *serviceConfig
+		service *corev1.Service
+		lbaasV2 *LbaasV2
+	}
+	tests := []struct {
+		name        string
+		args        args
+		want        string
+		expectedErr error
+	}{
+		{
+			name: "test get subnet from service annotation",
+			args: args{
+				svcConf: &serviceConfig{
+					keepClientIP:        true,
+					tlsContainerRef:     "tls-container-ref",
+					enableProxyProtocol: true,
+				},
+				lbaasV2: &LbaasV2{
+					LoadBalancer{
+						opts: LoadBalancerOpts{
+							LBClasses: map[string]*LBClass{
+								"test-class": {
+									SubnetID:           "test-id",
+									FloatingNetworkID:  "test-floating-network-id",
+									FloatingSubnetID:   "test-floating-subnet-id",
+									FloatingSubnet:     "test-floating-subnet",
+									NetworkID:          "test-network-id",
+									FloatingSubnetTags: "test-floating-subnet-tags",
+									MemberSubnetID:     "test-member-subnet-id",
+								},
+							},
+						},
+					},
+				},
+				service: &corev1.Service{
+					ObjectMeta: v1.ObjectMeta{
+						Annotations: map[string]string{
+							"loadbalancer.openstack.org/subnet-id": "test-id",
+							"loadbalancer.openstack.org/class":     "test-class",
+						},
+					},
+				},
+			},
+			want: "test-id",
+		},
+		{
+			name: "test get subnet from config class",
+			args: args{
+				svcConf: &serviceConfig{
+					keepClientIP:        true,
+					tlsContainerRef:     "tls-container-ref",
+					enableProxyProtocol: true,
+				},
+				lbaasV2: &LbaasV2{
+					LoadBalancer{
+						opts: LoadBalancerOpts{
+							LBClasses: map[string]*LBClass{
+								"test-class": {
+									SubnetID:           "test-class-subnet-id",
+									FloatingNetworkID:  "test-floating-network-id",
+									FloatingSubnetID:   "test-floating-subnet-id",
+									FloatingSubnet:     "test-floating-subnet",
+									NetworkID:          "test-network-id",
+									FloatingSubnetTags: "test-floating-subnet-tags",
+									MemberSubnetID:     "test-member-subnet-id",
+								},
+							},
+						},
+					},
+				},
+				service: &corev1.Service{
+					ObjectMeta: v1.ObjectMeta{
+						Annotations: map[string]string{
+							"loadbalancer.openstack.org/class": "test-class",
+						},
+					},
+				},
+			},
+			want:        "test-class-subnet-id",
+			expectedErr: nil,
+		},
+		{
+			name: "test get subnet from config class",
+			args: args{
+				svcConf: &serviceConfig{
+					keepClientIP:        true,
+					tlsContainerRef:     "tls-container-ref",
+					enableProxyProtocol: true,
+				},
+				lbaasV2: &LbaasV2{
+					LoadBalancer{
+						opts: LoadBalancerOpts{
+							SubnetID: "test-subnet-id",
+						},
+					},
+				},
+				service: &corev1.Service{},
+			},
+			want:        "test-subnet-id",
+			expectedErr: nil,
+		},
+		{
+			name: "test get subnet from config class with invalid loadbalancer class",
+			args: args{
+				svcConf: &serviceConfig{
+					keepClientIP:        true,
+					tlsContainerRef:     "tls-container-ref",
+					enableProxyProtocol: true,
+				},
+				lbaasV2: &LbaasV2{
+					LoadBalancer{
+						opts: LoadBalancerOpts{
+							SubnetID: "test-subnet-id",
+						},
+					},
+				},
+				service: &corev1.Service{
+					ObjectMeta: v1.ObjectMeta{
+						Annotations: map[string]string{
+							"loadbalancer.openstack.org/class": "test-class",
+						},
+					},
+				},
+			},
+			want:        "",
+			expectedErr: fmt.Errorf("invalid loadbalancer class %q", "test-class"),
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := tt.args.lbaasV2.getSubnetID(tt.args.service, tt.args.svcConf)
+			if tt.expectedErr != nil {
+				assert.EqualError(t, err, tt.expectedErr.Error())
+			}
+
+			if !reflect.DeepEqual(got, tt.want) {
+				t.Errorf("getSubnetID() = %v, expected %v", got, tt.want)
+			}
+		})
+	}
+}