Files
nomad/command/operator_autopilot_set_test.go
2024-12-19 15:41:11 +00:00

62 lines
1.5 KiB
Go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: BUSL-1.1
package command
import (
"strings"
"testing"
"time"
"github.com/hashicorp/cli"
"github.com/hashicorp/nomad/ci"
"github.com/shoenig/test/must"
)
func TestOperator_Autopilot_SetConfig_Implements(t *testing.T) {
ci.Parallel(t)
var _ cli.Command = &OperatorRaftListCommand{}
}
func TestOperatorAutopilotSetConfigCommand(t *testing.T) {
ci.Parallel(t)
s, _, addr := testServer(t, false, nil)
defer s.Shutdown()
ui := cli.NewMockUi()
c := &OperatorAutopilotSetCommand{Meta: Meta{Ui: ui}}
args := []string{
"-address=" + addr,
"-cleanup-dead-servers=false",
"-max-trailing-logs=99",
"-min-quorum=3",
"-last-contact-threshold=123ms",
"-server-stabilization-time=123ms",
"-enable-redundancy-zones=true",
"-disable-upgrade-migration=true",
"-enable-custom-upgrades=true",
}
code := c.Run(args)
must.Zero(t, code)
output := strings.TrimSpace(ui.OutputWriter.String())
must.StrContains(t, output, "Configuration updated")
client, err := c.Client()
must.NoError(t, err)
conf, _, err := client.Operator().AutopilotGetConfiguration(nil)
must.NoError(t, err)
must.False(t, conf.CleanupDeadServers)
must.Eq(t, 99, conf.MaxTrailingLogs)
must.Eq(t, 3, conf.MinQuorum)
must.Eq(t, 123*time.Millisecond, conf.LastContactThreshold)
must.Eq(t, 123*time.Millisecond, conf.ServerStabilizationTime)
must.True(t, conf.EnableRedundancyZones)
must.True(t, conf.DisableUpgradeMigration)
must.True(t, conf.EnableCustomUpgrades)
}