From 5931ff6453e81f4cd575cad614dc7403076e80b4 Mon Sep 17 00:00:00 2001
From: kahirokunn <okinakahiro@gmail.com>
Date: Mon, 16 Dec 2024 13:00:50 +0900
Subject: [PATCH] test(gateway-api): add tests for custom backend filters

Signed-off-by: kahirokunn <okinakahiro@gmail.com>
---
 pkg/router/gateway_api_test.go | 61 ++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/pkg/router/gateway_api_test.go b/pkg/router/gateway_api_test.go
index fb621f215..cdb09ad0d 100644
--- a/pkg/router/gateway_api_test.go
+++ b/pkg/router/gateway_api_test.go
@@ -276,6 +276,67 @@ func TestGatewayAPIRouter_Routes(t *testing.T) {
 		assert.Len(t, hr.Spec.Rules, 1)
 		assert.Len(t, hr.Spec.Rules[0].Filters, 0)
 	})
+
+	t.Run("custom backend filters", func(t *testing.T) {
+		canary := mocks.canary.DeepCopy()
+		primaryHostName := v1.PreciseHostname("primary.example.com")
+		canary.Spec.Service.Primary = &flaggerv1.CustomBackend{
+			Backend: &flaggerv1.HTTPBackendRefTemplate{
+				Filters: []v1.HTTPRouteFilter{
+					{
+						Type: v1.HTTPRouteFilterURLRewrite,
+						URLRewrite: &v1.HTTPURLRewriteFilter{
+							Hostname: &primaryHostName,
+						},
+					},
+				},
+			},
+		}
+
+		name := v1.ObjectName("canary")
+		unmanagedSvcNamespace := "kube-system"
+		namespace := v1.Namespace(unmanagedSvcNamespace)
+		port := v1.PortNumber(30080)
+		objRef := v1.BackendObjectReference{
+			Name:      name,
+			Namespace: &namespace,
+			Port:      &port,
+		}
+
+		canary.Spec.Service.Canary = &flaggerv1.CustomBackend{
+			Backend: &flaggerv1.HTTPBackendRefTemplate{
+				BackendObjectReference: &objRef,
+			},
+		}
+		err = router.SetRoutes(canary, 50, 50, false)
+		require.NoError(t, err)
+
+		httpRoute, err := router.gatewayAPIClient.GatewayapiV1().HTTPRoutes("default").Get(context.TODO(), "podinfo", metav1.GetOptions{})
+		require.NoError(t, err)
+
+		primary := httpRoute.Spec.Rules[0].BackendRefs[0]
+		assert.Equal(t, int32(50), *primary.Weight)
+
+		canaryBackend := httpRoute.Spec.Rules[0].BackendRefs[1]
+		assert.Equal(t, canaryBackend.Name, name)
+		assert.Equal(t, canaryBackend.Namespace, &namespace)
+		assert.Equal(t, canaryBackend.Port, &port)
+
+		primaryBackend := httpRoute.Spec.Rules[0].BackendRefs[0].Filters[0].URLRewrite
+		assert.Equal(t, primaryBackend.Hostname, &primaryHostName)
+
+		err = router.Reconcile(canary)
+		require.NoError(t, err)
+
+		referenceGrant, err := router.gatewayAPIClient.GatewayapiV1beta1().ReferenceGrants(unmanagedSvcNamespace).Get(context.TODO(), canary.Name, metav1.GetOptions{})
+		require.NoError(t, err)
+		assert.Equal(t, unmanagedSvcNamespace, string(referenceGrant.Namespace))
+		assert.Equal(t, "HTTPRoute", string(referenceGrant.Spec.From[0].Kind))
+		assert.Equal(t, canary.Namespace, string(referenceGrant.Spec.From[0].Namespace))
+		assert.Equal(t, "Service", string(referenceGrant.Spec.To[0].Kind))
+		assert.Equal(t, "", string(referenceGrant.Spec.To[0].Group))
+		assert.Equal(t, string(name), string(*referenceGrant.Spec.To[0].Name))
+	})
 }
 
 func TestGatewayAPIRouter_getSessionAffinityRouteRules(t *testing.T) {