Function TestNodeUpdate

cli/command/node/update_test.go:76–183

(t *testing.T)

}

func TestNodeUpdate(t *testing.T) {
	testCases := []struct {
		args            []string
		flags           map[string]string
		nodeInspectFunc func() (client.NodeInspectResult, error)
		nodeUpdateFunc  func(nodeID string, options client.NodeUpdateOptions) (client.NodeUpdateResult, error)
	}{
		{
			args: []string{"nodeID"},
			flags: map[string]string{
				"role": "manager",
			},
			nodeInspectFunc: func() (client.NodeInspectResult, error) {
				return client.NodeInspectResult{
					Node: *builders.Node(),
				}, nil
			},
			nodeUpdateFunc: func(nodeID string, options client.NodeUpdateOptions) (client.NodeUpdateResult, error) {
				if options.Spec.Role != swarm.NodeRoleManager {
					return client.NodeUpdateResult{}, errors.New("expected role manager, got " + string(options.Spec.Role))
				}
				return client.NodeUpdateResult{}, nil
			},
		},
		{
			args: []string{"nodeID"},
			flags: map[string]string{
				"availability": "drain",
			},
			nodeInspectFunc: func() (client.NodeInspectResult, error) {
				return client.NodeInspectResult{
					Node: *builders.Node(),
				}, nil
			},
			nodeUpdateFunc: func(nodeID string, options client.NodeUpdateOptions) (client.NodeUpdateResult, error) {
				if options.Spec.Availability != swarm.NodeAvailabilityDrain {
					return client.NodeUpdateResult{}, errors.New("expected drain availability, got " + string(options.Spec.Availability))
				}
				return client.NodeUpdateResult{}, nil
			},
		},
		{
			args: []string{"nodeID"},
			flags: map[string]string{
				"label-add": "lbl",
			},
			nodeInspectFunc: func() (client.NodeInspectResult, error) {
				return client.NodeInspectResult{
					Node: *builders.Node(),
				}, nil
			},
			nodeUpdateFunc: func(nodeID string, options client.NodeUpdateOptions) (client.NodeUpdateResult, error) {
				if _, present := options.Spec.Annotations.Labels["lbl"]; !present {
					return client.NodeUpdateResult{}, fmt.Errorf("expected 'lbl' label, got %v", options.Spec.Annotations.Labels)
				}
				return client.NodeUpdateResult{}, nil
			},
		},