package node import ( "bytes" "fmt" "io/ioutil" "testing" "github.com/docker/docker/api/types" "github.com/docker/docker/api/types/swarm" "github.com/docker/docker/cli/internal/test" // Import builders to get the builder function as package function . "github.com/docker/docker/cli/internal/test/builders" "github.com/docker/docker/pkg/testutil/assert" ) func TestNodeListErrorOnAPIFailure(t *testing.T) { testCases := []struct { nodeListFunc func() ([]swarm.Node, error) infoFunc func() (types.Info, error) expectedError string }{ { nodeListFunc: func() ([]swarm.Node, error) { return []swarm.Node{}, fmt.Errorf("error listing nodes") }, expectedError: "error listing nodes", }, { nodeListFunc: func() ([]swarm.Node, error) { return []swarm.Node{ { ID: "nodeID", }, }, nil }, infoFunc: func() (types.Info, error) { return types.Info{}, fmt.Errorf("error asking for node info") }, expectedError: "error asking for node info", }, } for _, tc := range testCases { buf := new(bytes.Buffer) cmd := newListCommand( test.NewFakeCli(&fakeClient{ nodeListFunc: tc.nodeListFunc, infoFunc: tc.infoFunc, }, buf)) cmd.SetOutput(ioutil.Discard) assert.Error(t, cmd.Execute(), tc.expectedError) } } func TestNodeList(t *testing.T) { buf := new(bytes.Buffer) cmd := newListCommand( test.NewFakeCli(&fakeClient{ nodeListFunc: func() ([]swarm.Node, error) { return []swarm.Node{ *Node(NodeID("nodeID1"), Hostname("nodeHostname1"), Manager(Leader())), *Node(NodeID("nodeID2"), Hostname("nodeHostname2"), Manager()), *Node(NodeID("nodeID3"), Hostname("nodeHostname3")), }, nil }, infoFunc: func() (types.Info, error) { return types.Info{ Swarm: swarm.Info{ NodeID: "nodeID1", }, }, nil }, }, buf)) assert.NilError(t, cmd.Execute()) assert.Contains(t, buf.String(), `nodeID1 * nodeHostname1 Ready Active Leader`) assert.Contains(t, buf.String(), `nodeID2 nodeHostname2 Ready Active Reachable`) assert.Contains(t, buf.String(), `nodeID3 nodeHostname3 Ready Active`) } func TestNodeListQuietShouldOnlyPrintIDs(t *testing.T) { buf := new(bytes.Buffer) cmd := newListCommand( test.NewFakeCli(&fakeClient{ nodeListFunc: func() ([]swarm.Node, error) { return []swarm.Node{ *Node(), }, nil }, }, buf)) cmd.Flags().Set("quiet", "true") assert.NilError(t, cmd.Execute()) assert.Contains(t, buf.String(), "nodeID") } // Test case for #24090 func TestNodeListContainsHostname(t *testing.T) { buf := new(bytes.Buffer) cmd := newListCommand(test.NewFakeCli(&fakeClient{}, buf)) assert.NilError(t, cmd.Execute()) assert.Contains(t, buf.String(), "HOSTNAME") }