apmbench: support api key

systemtest/benchtest/clients.go

@@ -111,6 +111,7 @@ func NewEventHandler(tb testing.TB, p string, l *rate.Limiter) *eventhandler.Han
 		Path:              p,
 		URL:               serverCfg.ServerURL.String(),
 		Token:             serverCfg.SecretToken,
+		APIKey:            serverCfg.APIKey,
 		Limiter:           l,
 		RewriteIDs:        serverCfg.RewriteIDs,
 		RewriteTimestamps: serverCfg.RewriteTimestamps,
@@ -132,6 +133,7 @@ func NewFSEventHandler(tb testing.TB, p string, l *rate.Limiter, fs fs.FS) *even
 		Path:              p,
 		URL:               serverCfg.ServerURL.String(),
 		Token:             serverCfg.SecretToken,
+		APIKey:            serverCfg.APIKey,
 		Limiter:           l,
 		RewriteIDs:        serverCfg.RewriteIDs,
 		RewriteTimestamps: serverCfg.RewriteTimestamps,

systemtest/benchtest/main.go

@@ -220,7 +220,8 @@ func Run(allBenchmarks ...BenchmarkFunc) error {
 		agents := agentsList[0]
 		serverURL := loadgencfg.Config.ServerURL.String()
 		secretToken := loadgencfg.Config.SecretToken
-		if err := warmup(agents, benchConfig.WarmupTime, serverURL, secretToken); err != nil {
+		apiKey := loadgencfg.Config.APIKey
+		if err := warmup(agents, benchConfig.WarmupTime, serverURL, secretToken, apiKey); err != nil {
 			return fmt.Errorf("warm-up failed with %d agents: %v", agents, err)
 		}
 	}
@@ -255,14 +256,15 @@ func Run(allBenchmarks ...BenchmarkFunc) error {
 
 // warmup sends events to the remote APM Server using the specified number of
 // agents for the specified duration.
-func warmup(agents int, duration time.Duration, url, token string) error {
+func warmup(agents int, duration time.Duration, url, token, apiKey string) error {
 	rl := loadgen.GetNewLimiter(loadgencfg.Config.EventRate.Burst, loadgencfg.Config.EventRate.Interval)
 	h, err := loadgen.NewEventHandler(loadgen.EventHandlerParams{
 		Logger:   zap.NewNop(),
 		Protocol: "apm/http",
 		Path:     `apm-*.ndjson`,
 		URL:      url,
 		Token:    token,
+		APIKey:   apiKey,
 		Limiter:  rl,
 	})
 	if err != nil {

systemtest/benchtest/main_test.go

@@ -111,7 +111,7 @@ func Test_warmup(t *testing.T) {
 			}))
 
 			t.Cleanup(srv.Close)
-			err := warmup(c.agents, c.duration, srv.URL, "")
+			err := warmup(c.agents, c.duration, srv.URL, "", "")
 			assert.NoError(t, err)
 			assert.Greater(t, received.Load(), uint64(c.agents))
 		})