Move codes to pkg (#46)

pkg/ingress/kube/annotations/annotations_test.go

@@ -0,0 +1,182 @@
+// Copyright (c) 2022 Alibaba Group Holding Ltd.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package annotations
+
+import "testing"
+
+func TestNeedRegexMatch(t *testing.T) {
+	testCases := []struct {
+		input  *Ingress
+		expect bool
+	}{
+		{
+			input:  &Ingress{},
+			expect: false,
+		},
+		{
+			input: &Ingress{
+				Rewrite: &RewriteConfig{},
+			},
+			expect: false,
+		},
+		{
+			input: &Ingress{
+				Rewrite: &RewriteConfig{
+					RewriteTarget: "/test",
+				},
+			},
+			expect: true,
+		},
+		{
+			input: &Ingress{
+				Rewrite: &RewriteConfig{
+					UseRegex: true,
+				},
+			},
+			expect: true,
+		},
+	}
+
+	for _, testCase := range testCases {
+		t.Run("", func(t *testing.T) {
+			if testCase.input.NeedRegexMatch() != testCase.expect {
+				t.Fatalf("Should be %t, but actual is %t", testCase.expect, testCase.input.NeedRegexMatch())
+			}
+		})
+	}
+}
+
+func TestIsCanary(t *testing.T) {
+	testCases := []struct {
+		input  *Ingress
+		expect bool
+	}{
+		{
+			input:  &Ingress{},
+			expect: false,
+		},
+		{
+			input: &Ingress{
+				Canary: &CanaryConfig{},
+			},
+			expect: false,
+		},
+		{
+			input: &Ingress{
+				Canary: &CanaryConfig{
+					Enabled: true,
+				},
+			},
+			expect: true,
+		},
+	}
+
+	for _, testCase := range testCases {
+		t.Run("", func(t *testing.T) {
+			if testCase.input.IsCanary() != testCase.expect {
+				t.Fatalf("Should be %t, but actual is %t", testCase.expect, testCase.input.IsCanary())
+			}
+		})
+	}
+}
+
+func TestCanaryKind(t *testing.T) {
+	testCases := []struct {
+		input    *Ingress
+		byHeader bool
+		byWeight bool
+	}{
+		{
+			input:    &Ingress{},
+			byHeader: false,
+			byWeight: false,
+		},
+		{
+			input: &Ingress{
+				Canary: &CanaryConfig{},
+			},
+			byHeader: false,
+			byWeight: false,
+		},
+		{
+			input: &Ingress{
+				Canary: &CanaryConfig{
+					Enabled: true,
+				},
+			},
+			byHeader: false,
+			byWeight: true,
+		},
+		{
+			input: &Ingress{
+				Canary: &CanaryConfig{
+					Enabled: true,
+					Header:  "test",
+				},
+			},
+			byHeader: true,
+			byWeight: false,
+		},
+		{
+			input: &Ingress{
+				Canary: &CanaryConfig{
+					Enabled: true,
+					Cookie:  "test",
+				},
+			},
+			byHeader: true,
+			byWeight: false,
+		},
+		{
+			input: &Ingress{
+				Canary: &CanaryConfig{
+					Enabled: true,
+					Weight:  2,
+				},
+			},
+			byHeader: false,
+			byWeight: true,
+		},
+	}
+
+	for _, testCase := range testCases {
+		t.Run("", func(t *testing.T) {
+			byHeader, byWeight := testCase.input.CanaryKind()
+			if byHeader != testCase.byHeader {
+				t.Fatalf("Should be %t, but actual is %t", testCase.byHeader, byHeader)
+			}
+
+			if byWeight != testCase.byWeight {
+				t.Fatalf("Should be %t, but actual is %t", testCase.byWeight, byWeight)
+			}
+		})
+	}
+}
+
+func TestNeedTrafficPolicy(t *testing.T) {
+	config1 := &Ingress{}
+	if config1.NeedTrafficPolicy() {
+		t.Fatal("should be false")
+	}
+
+	config2 := &Ingress{
+		UpstreamTLS: &UpstreamTLSConfig{
+			BackendProtocol: defaultBackendProtocol,
+		},
+	}
+	if !config2.NeedTrafficPolicy() {
+		t.Fatal("should be true")
+	}
+}