blob: 1dd4fe0c80604bf39ff6f0b4c6a61b4dd4973cdd [file] [log] [blame]
// Licensed to the Apache Software Foundation (ASF) under one or more
// contributor license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright ownership.
// The ASF licenses this file to You 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 v1
import (
"context"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
corev1 "k8s.io/api/core/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
clientgoscheme "k8s.io/client-go/kubernetes/scheme"
"sigs.k8s.io/controller-runtime/pkg/client/fake"
gatewayv1 "sigs.k8s.io/gateway-api/apis/v1"
"github.com/apache/apisix-ingress-controller/internal/controller/config"
)
func buildHTTPRouteValidator(t *testing.T, objects ...runtime.Object) *HTTPRouteCustomValidator {
t.Helper()
scheme := runtime.NewScheme()
require.NoError(t, clientgoscheme.AddToScheme(scheme))
require.NoError(t, gatewayv1.Install(scheme))
managed := []runtime.Object{
&gatewayv1.GatewayClass{
ObjectMeta: metav1.ObjectMeta{Name: "apisix-gateway-class"},
Spec: gatewayv1.GatewayClassSpec{
ControllerName: gatewayv1.GatewayController(config.ControllerConfig.ControllerName),
},
},
&gatewayv1.Gateway{
ObjectMeta: metav1.ObjectMeta{Name: "test-gateway", Namespace: "default"},
Spec: gatewayv1.GatewaySpec{
GatewayClassName: gatewayv1.ObjectName("apisix-gateway-class"),
},
},
}
allObjects := append(managed, objects...)
builder := fake.NewClientBuilder().WithScheme(scheme).WithRuntimeObjects(allObjects...)
return NewHTTPRouteCustomValidator(builder.Build())
}
func TestHTTPRouteCustomValidator_WarnsForMissingReferences(t *testing.T) {
route := &gatewayv1.HTTPRoute{
ObjectMeta: metav1.ObjectMeta{Name: "demo", Namespace: "default"},
Spec: gatewayv1.HTTPRouteSpec{
CommonRouteSpec: gatewayv1.CommonRouteSpec{
ParentRefs: []gatewayv1.ParentReference{{
Name: gatewayv1.ObjectName("test-gateway"),
}},
},
Rules: []gatewayv1.HTTPRouteRule{{
BackendRefs: []gatewayv1.HTTPBackendRef{{
BackendRef: gatewayv1.BackendRef{
BackendObjectReference: gatewayv1.BackendObjectReference{
Name: gatewayv1.ObjectName("missing-svc"),
},
},
}},
Filters: []gatewayv1.HTTPRouteFilter{{
Type: gatewayv1.HTTPRouteFilterRequestMirror,
RequestMirror: &gatewayv1.HTTPRequestMirrorFilter{
BackendRef: gatewayv1.BackendObjectReference{
Name: gatewayv1.ObjectName("mirror-svc"),
},
},
}},
}},
},
}
validator := buildHTTPRouteValidator(t)
warnings, err := validator.ValidateCreate(context.Background(), route)
require.NoError(t, err)
assert.ElementsMatch(t, []string{
"Referenced Service 'default/mirror-svc' not found",
"Referenced Service 'default/missing-svc' not found",
}, warnings)
}
func TestHTTPRouteCustomValidator_NoWarningsWhenResourcesExist(t *testing.T) {
objects := []runtime.Object{
&corev1.Service{ObjectMeta: metav1.ObjectMeta{Name: "primary", Namespace: "default"}},
&corev1.Service{ObjectMeta: metav1.ObjectMeta{Name: "mirror", Namespace: "default"}},
}
validator := buildHTTPRouteValidator(t, objects...)
route := &gatewayv1.HTTPRoute{
ObjectMeta: metav1.ObjectMeta{Name: "demo", Namespace: "default"},
Spec: gatewayv1.HTTPRouteSpec{
CommonRouteSpec: gatewayv1.CommonRouteSpec{
ParentRefs: []gatewayv1.ParentReference{{
Name: gatewayv1.ObjectName("test-gateway"),
}},
},
Rules: []gatewayv1.HTTPRouteRule{{
BackendRefs: []gatewayv1.HTTPBackendRef{{
BackendRef: gatewayv1.BackendRef{
BackendObjectReference: gatewayv1.BackendObjectReference{
Name: gatewayv1.ObjectName("primary"),
},
},
}},
Filters: []gatewayv1.HTTPRouteFilter{{
Type: gatewayv1.HTTPRouteFilterRequestMirror,
RequestMirror: &gatewayv1.HTTPRequestMirrorFilter{
BackendRef: gatewayv1.BackendObjectReference{
Name: gatewayv1.ObjectName("mirror"),
},
},
}},
}},
},
}
warnings, err := validator.ValidateCreate(context.Background(), route)
require.NoError(t, err)
assert.Empty(t, warnings)
}