Skip to content

Commit

Permalink
Add conformance tests for EndpointSlices
Browse files Browse the repository at this point in the history
Look for an MCS EndpointSlice using the criteria whereby the
"endpointslice.kubernetes.io/managed-by" label is not equal to
endpointslice-controller.k8s.io" or any MCS label is present.
If not found then fail the test as non-conformant. Otherwise
verify the required MCS labels are correctly set.

The tests are labeled with "EndpointSlice" so, if EndpointSlices
aren't supported by the implmentation, the user can use the
"EndpointSlice" label to filter out the tests.

Fixes #70

Signed-off-by: Tom Pantelis <[email protected]>
  • Loading branch information
tpantelis committed Oct 25, 2024
1 parent 959a5dd commit 25ba403
Show file tree
Hide file tree
Showing 3 changed files with 140 additions and 4 deletions.
23 changes: 19 additions & 4 deletions conformance/conformance_suite.go
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ import (
. "github.com/onsi/ginkgo/v2"
. "github.com/onsi/gomega"
corev1 "k8s.io/api/core/v1"
discoveryv1 "k8s.io/api/discovery/v1"
apierrors "k8s.io/apimachinery/pkg/api/errors"
"k8s.io/apimachinery/pkg/api/meta"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
Expand All @@ -49,10 +50,13 @@ type clusterClients struct {
rest *rest.Config
}

var contexts string
var clients []clusterClients
var loadingRules *clientcmd.ClientConfigLoadingRules
var ctx = context.TODO()
var (
contexts string
clients []clusterClients
loadingRules *clientcmd.ClientConfigLoadingRules
skipVerifyEndpointSliceManagedBy bool
ctx = context.TODO()
)

func TestConformance(t *testing.T) {
RegisterFailHandler(Fail)
Expand All @@ -64,6 +68,12 @@ func init() {
loadingRules.DefaultClientConfig = &clientcmd.DefaultClientConfig
flag.StringVar(&loadingRules.ExplicitPath, "kubeconfig", "", "absolute path(s) to the kubeconfig file(s)")
flag.StringVar(&contexts, "contexts", "", "comma-separated list of contexts to use")
flag.BoolVar(&skipVerifyEndpointSliceManagedBy, "skip-verify-eps-managed-by", false,
fmt.Sprintf("The MSC spec states that any EndpointSlice created by an mcs-controller must be marked as managed by "+
"the mcs-controller. By default, the conformance test verifies that the %q label on MCS EndpointSlices is not equal to %q. "+
"However with some implementations, MCS EndpointSlices may be created and managed by K8s. If this flag is set to true, "+
"the test only verifies the presence of the label.",
discoveryv1.LabelManagedBy, K8sEndpointSliceManagedByName))
}

var _ = BeforeSuite(func() {
Expand Down Expand Up @@ -166,6 +176,11 @@ func (t *testDriver) createServiceExport(c *clusterClients) {
Expect(err).ToNot(HaveOccurred())
}

func (t *testDriver) deleteServiceExport(c *clusterClients) {
Expect(c.mcs.MulticlusterV1alpha1().ServiceExports(t.namespace).Delete(ctx, helloServiceName,
metav1.DeleteOptions{})).ToNot(HaveOccurred())
}

func (t *testDriver) deployHelloService(c *clusterClients, service *corev1.Service) {
_, err := c.k8s.AppsV1().Deployments(t.namespace).Create(ctx, newHelloDeployment(), metav1.CreateOptions{})
Expect(err).ToNot(HaveOccurred())
Expand Down
120 changes: 120 additions & 0 deletions conformance/endpoint_slice.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,120 @@
/*
Copyright 2024 The Kubernetes Authors.
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 conformance

import (
"context"
"fmt"
"time"

. "github.com/onsi/ginkgo/v2"
. "github.com/onsi/gomega"
discoveryv1 "k8s.io/api/discovery/v1"
apierrors "k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/util/wait"
"sigs.k8s.io/mcs-api/pkg/apis/v1alpha1"
)

const K8sEndpointSliceManagedByName = "endpointslice-controller.k8s.io"

var _ = Describe("", Label(OptionalLabel, EndpointSliceLabel), func() {
t := newTestDriver()

JustBeforeEach(func() {
t.createServiceExport(&clients[0])
})

Specify("Exporting a service should create an MCS EndpointSlice in the service's namespace in each cluster with the "+
"required MCS labels. Unexporting should delete the EndpointSlice.", func() {
AddReportEntry(SpecRefReportEntry, "https://github.com/kubernetes/enhancements/tree/master/keps/sig-multicluster/1645-multi-cluster-services-api#using-endpointslice-objects-to-track-endpoints")

endpointSlices := make([]*discoveryv1.EndpointSlice, len(clients))

for i, client := range clients {
eps := t.awaitMCSEndpointSlice(&client)
Expect(eps).ToNot(BeNil(), reportNonConformant(fmt.Sprintf(
"an MCS EndpointSlice was not found on cluster %q. An MCS EndpointSlice is identified by the presence "+
"of at least one of the required MCS labels, whose names are prefixed with \"multicluster.kubernetes.io\". "+
"If the MCS implementation does not use MCS EndpointSlices, you can specify a Ginkgo label filter using "+
"the %q label where appropriate to skip this test.",
client.name, EndpointSliceLabel)))

endpointSlices[i] = eps

Expect(eps.Labels).To(HaveKeyWithValue(v1alpha1.LabelServiceName, t.helloService.Name),
reportNonConformant(fmt.Sprintf("the MCS EndpointSlice %q does not contain the %q label referencing the service name",
eps.Name, v1alpha1.LabelServiceName)))

Expect(eps.Labels).To(HaveKey(v1alpha1.LabelSourceCluster),
reportNonConformant(fmt.Sprintf("the MCS EndpointSlice %q does not contain the %q label",
eps.Name, v1alpha1.LabelSourceCluster)))

Expect(eps.Labels).To(HaveKey(discoveryv1.LabelManagedBy),
reportNonConformant(fmt.Sprintf("the MCS EndpointSlice %q does not contain the %q label",
eps.Name, discoveryv1.LabelManagedBy)))

if !skipVerifyEndpointSliceManagedBy {
Expect(eps.Labels[discoveryv1.LabelManagedBy]).ToNot(Equal(K8sEndpointSliceManagedByName),
reportNonConformant(fmt.Sprintf("the MCS EndpointSlice's %q label must not reference %q",
discoveryv1.LabelManagedBy, K8sEndpointSliceManagedByName)))
}
}

By("Unexporting the service")

t.deleteServiceExport(&clients[0])

for i, client := range clients {
Eventually(func() bool {
_, err := client.k8s.DiscoveryV1().EndpointSlices(t.namespace).Get(ctx, endpointSlices[i].Name, metav1.GetOptions{})
return apierrors.IsNotFound(err)
}, 20*time.Second, 100*time.Millisecond).Should(BeTrue(),
reportNonConformant(fmt.Sprintf("the EndpointSlice was not deleted on unexport from cluster %d", i+1)))
}
})
})

func (t *testDriver) awaitMCSEndpointSlice(c *clusterClients) *discoveryv1.EndpointSlice {
var endpointSlice *discoveryv1.EndpointSlice

hasLabel := func(eps *discoveryv1.EndpointSlice, label string) bool {
_, exists := eps.Labels[label]
return exists
}

_ = wait.PollUntilContextTimeout(ctx, 100*time.Millisecond,
20*time.Second, true, func(ctx context.Context) (bool, error) {
defer GinkgoRecover()

list, err := c.k8s.DiscoveryV1().EndpointSlices(t.namespace).List(ctx, metav1.ListOptions{})
Expect(err).ToNot(HaveOccurred(), "Error retrieving EndpointSlices")

for i := range list.Items {
eps := &list.Items[i]

if hasLabel(eps, v1alpha1.LabelServiceName) || hasLabel(eps, v1alpha1.LabelSourceCluster) {
endpointSlice = eps
return true, nil
}
}

return false, nil
})

return endpointSlice
}
1 change: 1 addition & 0 deletions conformance/report.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ const (
RequiredLabel = "Required"
DNSLabel = "DNS"
ClusterIPLabel = "ClusterIP"
EndpointSliceLabel = "EndpointSlice"
SpecRefReportEntry = "spec-ref"
NonConformantReportEntry = "non-conformant"
)
Expand Down

0 comments on commit 25ba403

Please sign in to comment.