mirror of
https://github.com/kemko/nomad.git
synced 2026-01-01 16:05:42 +03:00
e3c8700ded
No functional changes, just cleaning up deprecated usages that are removed in v2 and replace one call of .Slice with .ForEach to avoid making the intermediate copy.
90 lines
2.0 KiB
Go
90 lines
2.0 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package command
|
|
|
|
import (
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/go-set/v2"
|
|
"github.com/hashicorp/nomad/api"
|
|
"github.com/hashicorp/nomad/ci"
|
|
"github.com/mitchellh/cli"
|
|
"github.com/posener/complete"
|
|
"github.com/shoenig/test/must"
|
|
)
|
|
|
|
var (
|
|
// jsonOutputRaftIndexes is a regex that matches raft index fields in JSON
|
|
// strings. It can be used to remove them to make test results more
|
|
// consistent.
|
|
jsonOutputRaftIndexes = regexp.MustCompile(`(?m)\s*"(?:CreateIndex|ModifyIndex)".*`)
|
|
)
|
|
|
|
func TestNodePoolCommand_Implements(t *testing.T) {
|
|
ci.Parallel(t)
|
|
var _ cli.Command = &NodePoolCommand{}
|
|
}
|
|
|
|
func TestMeta_NodePoolPredictor(t *testing.T) {
|
|
ci.Parallel(t)
|
|
|
|
// Start test server.
|
|
srv, client, url := testServer(t, true, nil)
|
|
defer srv.Shutdown()
|
|
|
|
waitForNodes(t, client)
|
|
|
|
// Register some test node pools.
|
|
dev1 := &api.NodePool{Name: "dev-1"}
|
|
_, err := client.NodePools().Register(dev1, nil)
|
|
must.NoError(t, err)
|
|
|
|
dev2 := &api.NodePool{Name: "dev-2"}
|
|
_, err = client.NodePools().Register(dev2, nil)
|
|
must.NoError(t, err)
|
|
|
|
prod := &api.NodePool{Name: "prod"}
|
|
_, err = client.NodePools().Register(prod, nil)
|
|
must.NoError(t, err)
|
|
|
|
testCases := []struct {
|
|
name string
|
|
args complete.Args
|
|
filter *set.Set[string]
|
|
expected []string
|
|
}{
|
|
{
|
|
name: "find with prefix",
|
|
args: complete.Args{
|
|
Last: "de",
|
|
},
|
|
expected: []string{"default", "dev-1", "dev-2"},
|
|
},
|
|
{
|
|
name: "filter",
|
|
args: complete.Args{
|
|
Last: "de",
|
|
},
|
|
filter: set.From([]string{"default"}),
|
|
expected: []string{"dev-1", "dev-2"},
|
|
},
|
|
{
|
|
name: "find all",
|
|
args: complete.Args{
|
|
Last: "",
|
|
},
|
|
expected: []string{"all", "default", "dev-1", "dev-2", "prod"},
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
m := &Meta{flagAddress: url}
|
|
got := nodePoolPredictor(m.Client, tc.filter).Predict(tc.args)
|
|
must.SliceContainsAll(t, tc.expected, got)
|
|
})
|
|
}
|
|
}
|