K8s: Test for changes to aggregator's postStartHooks (#91771)

2026-08-26 05:17:26 -05:00 · 2024-08-10 20:32:57 -04:00
parent 8bb548e17b
commit 75f0b3a228
1 changed files with 70 additions and 0 deletions
@@ -0,0 +1,70 @@
+package aggregator_test
+
+import (
+	"sort"
+	"testing"
+	"time"
+
+	"github.com/grafana/grafana/pkg/storage/unified/apistore"
+	"github.com/stretchr/testify/require"
+	openapinamer "k8s.io/apiserver/pkg/endpoints/openapi"
+	genericapiserver "k8s.io/apiserver/pkg/server"
+	"k8s.io/apiserver/pkg/storage/storagebackend"
+	utilversion "k8s.io/apiserver/pkg/util/version"
+	"k8s.io/client-go/informers"
+	"k8s.io/client-go/kubernetes/fake"
+	clientrest "k8s.io/client-go/rest"
+	"k8s.io/kube-aggregator/pkg/apiserver"
+	aggregatorscheme "k8s.io/kube-aggregator/pkg/apiserver/scheme"
+	aggregatoropenapi "k8s.io/kube-aggregator/pkg/generated/openapi"
+)
+
+// TestAggregatorPostStartHooks tests that the kube-aggregator server has the expected default post start hooks enabled.
+func TestAggregatorPostStartHooks(t *testing.T) {
+	cfg := apiserver.Config{
+		GenericConfig: genericapiserver.NewRecommendedConfig(aggregatorscheme.Codecs),
+		ExtraConfig:   apiserver.ExtraConfig{},
+	}
+
+	cfg.GenericConfig.ExternalAddress = "127.0.0.1:6443"
+	cfg.GenericConfig.EffectiveVersion = utilversion.DefaultBuildEffectiveVersion()
+	cfg.GenericConfig.LoopbackClientConfig = &clientrest.Config{}
+	cfg.GenericConfig.MergedResourceConfig = apiserver.DefaultAPIResourceConfigSource()
+
+	// Add OpenAPI config, which depends on builders
+	namer := openapinamer.NewDefinitionNamer(aggregatorscheme.Scheme)
+	cfg.GenericConfig.OpenAPIV3Config = genericapiserver.DefaultOpenAPIV3Config(aggregatoropenapi.GetOpenAPIDefinitions, namer)
+	cfg.GenericConfig.OpenAPIV3Config.Info.Title = "Kubernetes"
+	cfg.GenericConfig.OpenAPIConfig = genericapiserver.DefaultOpenAPIConfig(aggregatoropenapi.GetOpenAPIDefinitions, namer)
+	cfg.GenericConfig.OpenAPIConfig.Info.Title = "Kubernetes"
+	cfg.GenericConfig.SkipOpenAPIInstallation = true
+	cfg.GenericConfig.SharedInformerFactory = informers.NewSharedInformerFactory(fake.NewSimpleClientset(), 10*time.Minute)
+
+	// override the RESTOptionsGetter to use the in memory storage options
+	restOptionsGetter, err := apistore.NewRESTOptionsGetterMemory(*storagebackend.NewDefaultConfig("memory", nil))
+	require.NoError(t, err)
+	cfg.GenericConfig.RESTOptionsGetter = restOptionsGetter
+
+	complete := cfg.Complete()
+
+	server, err := complete.NewWithDelegate(genericapiserver.NewEmptyDelegate())
+	require.NoError(t, err)
+
+	actual := make([]string, 0, len(server.GenericAPIServer.PostStartHooks()))
+	for k := range server.GenericAPIServer.PostStartHooks() {
+		actual = append(actual, k)
+	}
+	sort.Strings(actual)
+	expected := []string{
+		"apiservice-discovery-controller",
+		"generic-apiserver-start-informers",
+		"max-in-flight-filter",
+		"storage-object-count-tracker-hook",
+		"start-kube-aggregator-informers",
+		"apiservice-status-local-available-controller",
+		"apiservice-status-remote-available-controller",
+		"apiservice-registration-controller",
+	}
+	sort.Strings(expected)
+	require.Equal(t, expected, actual)
+}