mirror of
https://github.com/kemko/nomad.git
synced 2026-01-01 16:05:42 +03:00
2f30205102
The Nomad client will persist its own identity within its state store for restart persistence. The added benefit of using it over the filesystem is that it supports transactions. This is useful when considering the identity will be renewed periodically.
54 lines
1.5 KiB
Go
54 lines
1.5 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: BUSL-1.1
|
|
|
|
package command
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/hashicorp/cli"
|
|
"github.com/hashicorp/nomad/ci"
|
|
"github.com/hashicorp/nomad/client/state"
|
|
"github.com/hashicorp/nomad/helper/testlog"
|
|
"github.com/hashicorp/nomad/nomad/structs"
|
|
"github.com/shoenig/test/must"
|
|
)
|
|
|
|
func TestOperatorClientStateCommand(t *testing.T) {
|
|
ci.Parallel(t)
|
|
ui := cli.NewMockUi()
|
|
cmd := &OperatorClientStateCommand{Meta: Meta{Ui: ui}}
|
|
|
|
failedCode := cmd.Run([]string{"some", "bad", "args"})
|
|
must.Eq(t, 1, failedCode)
|
|
out := ui.ErrorWriter.String()
|
|
must.StrContains(t, out, commandErrorText(cmd), must.Sprint("expected help output"))
|
|
ui.ErrorWriter.Reset()
|
|
|
|
dir := t.TempDir()
|
|
|
|
// run against an empty client state directory
|
|
code := cmd.Run([]string{dir})
|
|
must.Eq(t, 0, code)
|
|
must.StrContains(t, ui.OutputWriter.String(), "{}")
|
|
|
|
// create a minimal client state db
|
|
db, err := state.NewBoltStateDB(testlog.HCLogger(t), dir)
|
|
must.NoError(t, err)
|
|
alloc := structs.MockAlloc()
|
|
err = db.PutAllocation(alloc)
|
|
must.NoError(t, err)
|
|
|
|
// Write a node identity to the DB, so we can test that the command reads
|
|
// this data.
|
|
must.NoError(t, db.PutNodeIdentity("mynodeidentity"))
|
|
|
|
must.NoError(t, db.Close())
|
|
|
|
// run against an incomplete client state directory
|
|
code = cmd.Run([]string{dir})
|
|
must.Eq(t, 0, code)
|
|
must.StrContains(t, ui.OutputWriter.String(), alloc.ID)
|
|
must.StrContains(t, ui.OutputWriter.String(), "NodeIdentity\":\"mynodeidentity")
|
|
}
|