test(fips): validate keystore cmd is not enabled in fips mode (#15707) (#15723)

update test to validate fips behaviour

(cherry picked from commit e3d61f4)

Co-authored-by: kruskall <[email protected]>

Author: mergify[bot]

x-pack/apm-server/root_test.go x-pack/apm-server/root_fips_test.gox-pack/apm-server/root_test.go renamed to x-pack/apm-server/root_fips_test.go

@@ -2,6 +2,8 @@
 // or more contributor license agreements. Licensed under the Elastic License 2.0;
 // you may not use this file except in compliance with the Elastic License 2.0.
 
+//go:build requirefips
+
 package main
 
 import (
@@ -23,7 +25,6 @@ func TestSubCommands(t *testing.T) {
 
 	assert.ElementsMatch(t, []string{
 		"export",
-		"keystore",
 		"run",
 		"test",
 		"version",

x-pack/apm-server/root_nofips_test.go

@@ -0,0 +1,33 @@
+// Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+// or more contributor license agreements. Licensed under the Elastic License 2.0;
+// you may not use this file except in compliance with the Elastic License 2.0.
+
+//go:build !requirefips
+
+package main
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	"github.com/elastic/apm-server/internal/beatcmd"
+)
+
+func TestSubCommands(t *testing.T) {
+	rootCmd := newXPackRootCommand(func(beatcmd.RunnerParams) (beatcmd.Runner, error) {
+		panic("unexpected call")
+	})
+	var commands []string
+	for _, cmd := range rootCmd.Commands() {
+		commands = append(commands, cmd.Name())
+	}
+
+	assert.ElementsMatch(t, []string{
+		"export",
+		"keystore",
+		"run",
+		"test",
+		"version",
+	}, commands)
+}