github.com/huiliang/nomad@v0.2.1-0.20151124023127-7a8b664699ff/command/client_config_test.go (about)

     1  package command
     2  
     3  import (
     4  	"strings"
     5  	"testing"
     6  
     7  	"github.com/hashicorp/nomad/testutil"
     8  	"github.com/mitchellh/cli"
     9  )
    10  
    11  func TestClientConfigCommand_Implements(t *testing.T) {
    12  	var _ cli.Command = &ClientConfigCommand{}
    13  }
    14  
    15  func TestClientConfigCommand_UpdateServers(t *testing.T) {
    16  	srv, _, url := testServer(t, func(c *testutil.TestServerConfig) {
    17  		c.Client.Enabled = true
    18  		c.Server.BootstrapExpect = 0
    19  	})
    20  	defer srv.Stop()
    21  
    22  	ui := new(cli.MockUi)
    23  	cmd := &ClientConfigCommand{Meta: Meta{Ui: ui}}
    24  
    25  	// Fails if trying to update with no servers
    26  	code := cmd.Run([]string{"-update-servers"})
    27  	if code != 1 {
    28  		t.Fatalf("expected exit 1, got: %d", code)
    29  	}
    30  	if out := ui.ErrorWriter.String(); !strings.Contains(out, cmd.Help()) {
    31  		t.Fatalf("expected help output, got: %s", out)
    32  	}
    33  	ui.ErrorWriter.Reset()
    34  
    35  	// Set the servers list
    36  	code = cmd.Run([]string{"-address=" + url, "-update-servers", "foo", "bar"})
    37  	if code != 0 {
    38  		t.Fatalf("expected exit 0, got: %d", code)
    39  	}
    40  
    41  	// Query the servers list
    42  	code = cmd.Run([]string{"-address=" + url, "-servers"})
    43  	if code != 0 {
    44  		t.Fatalf("expect exit 0, got: %d", code)
    45  	}
    46  	out := ui.OutputWriter.String()
    47  	if !strings.Contains(out, "foo") {
    48  		t.Fatalf("missing foo")
    49  	}
    50  	if !strings.Contains(out, "bar") {
    51  		t.Fatalf("missing bar")
    52  	}
    53  }
    54  
    55  func TestClientConfigCommand_Fails(t *testing.T) {
    56  	ui := new(cli.MockUi)
    57  	cmd := &ClientConfigCommand{Meta: Meta{Ui: ui}}
    58  
    59  	// Fails on misuse
    60  	if code := cmd.Run([]string{"some", "bad", "args"}); code != 1 {
    61  		t.Fatalf("expected exit code 1, got: %d", code)
    62  	}
    63  	if out := ui.ErrorWriter.String(); !strings.Contains(out, cmd.Help()) {
    64  		t.Fatalf("expected help output, got: %s", out)
    65  	}
    66  	ui.ErrorWriter.Reset()
    67  
    68  	// Fails if no valid flags given
    69  	if code := cmd.Run([]string{}); code != 1 {
    70  		t.Fatalf("expected exit code 1, got: %d", code)
    71  	}
    72  	if out := ui.ErrorWriter.String(); !strings.Contains(out, cmd.Help()) {
    73  		t.Fatalf("expected help output, got: %s", out)
    74  	}
    75  	ui.ErrorWriter.Reset()
    76  
    77  	// Fails on connection failure
    78  	if code := cmd.Run([]string{"-address=nope", "-servers"}); code != 1 {
    79  		t.Fatalf("expected exit code 1, got: %d", code)
    80  	}
    81  	if out := ui.ErrorWriter.String(); !strings.Contains(out, "Error querying server list") {
    82  		t.Fatalf("expected failed query error, got: %s", out)
    83  	}
    84  }