diff --git a/conformance/utils/suite/suite.go b/conformance/utils/suite/suite.go index 53d914e1e8..af891aa41c 100644 --- a/conformance/utils/suite/suite.go +++ b/conformance/utils/suite/suite.go @@ -35,6 +35,7 @@ import ( clientset "k8s.io/client-go/kubernetes" "k8s.io/client-go/rest" "sigs.k8s.io/controller-runtime/pkg/client" + clicfg "sigs.k8s.io/controller-runtime/pkg/client/config" "sigs.k8s.io/gateway-api/apis/v1beta1" confv1 "sigs.k8s.io/gateway-api/conformance/apis/v1" @@ -307,6 +308,10 @@ func NewConformanceTestSuite(options ConformanceOptions) (*ConformanceTestSuite, // Conformance Test Suite - Public Methods // ----------------------------------------------------------------------------- +const ( + testSuiteUserAgentPrefix = "gateway-api-conformance.test" +) + // Setup ensures the base resources required for conformance tests are installed // in the cluster. It also ensures that all relevant resources are ready. func (suite *ConformanceTestSuite) Setup(t *testing.T, tests []ConformanceTest) { @@ -374,6 +379,35 @@ func (suite *ConformanceTestSuite) Setup(t *testing.T, tests []ConformanceTest) } } +func (suite *ConformanceTestSuite) setClientsetForTest(test ConformanceTest) error { + featureNames := []string{} + for _, v := range test.Features { + featureNames = append(featureNames, string(v)) + } + if len(test.Features) == 0 { + featureNames = []string{"unknownFeature"} + } + suite.RestConfig.UserAgent = strings.Join( + []string{ + testSuiteUserAgentPrefix, + suite.apiVersion, + test.ShortName, + strings.Join(featureNames, ","), + }, + "::") + client, err := client.New(suite.RestConfig, client.Options{}) + if err != nil { + return err + } + clientset, err := clientset.NewForConfig(suite.RestConfig) + if err != nil { + return err + } + suite.Client = client + suite.Clientset = clientset + return nil +} + // Run runs the provided set of conformance tests. func (suite *ConformanceTestSuite) Run(t *testing.T, tests []ConformanceTest) error { // verify that the test suite isn't already running, don't start a new run @@ -410,6 +444,8 @@ func (suite *ConformanceTestSuite) Run(t *testing.T, tests []ConformanceTest) er } succeeded := t.Run(test.ShortName, func(t *testing.T) { + err := suite.setClientsetForTest(test) + require.NoError(t, err, "failed to create new clientset for test") test.Run(t, suite) }) if !succeeded {