github.com/diptanu/nomad@v0.5.7-0.20170516172507-d72e86cbe3d9/command/check_test.go (about)

     1  package command
     2  
     3  import (
     4  	"fmt"
     5  	"testing"
     6  
     7  	"github.com/mitchellh/cli"
     8  )
     9  
    10  func TestAgentCheckCommand_ServerHealth(t *testing.T) {
    11  	srv, _, url := testServer(t, nil)
    12  	defer srv.Stop()
    13  
    14  	ui := new(cli.MockUi)
    15  	cmd := &AgentCheckCommand{Meta: Meta{Ui: ui}}
    16  	address := fmt.Sprintf("-address=%s", url)
    17  
    18  	code := cmd.Run([]string{address})
    19  	if code != HealthPass {
    20  		t.Fatalf("expected exit: %v, actual: %d", HealthPass, code)
    21  	}
    22  
    23  	minPeers := fmt.Sprintf("-min-peers=%v", 3)
    24  	code = cmd.Run([]string{address, minPeers})
    25  	if code != HealthCritical {
    26  		t.Fatalf("expected exitcode: %v, actual: %v", HealthCritical, code)
    27  	}
    28  
    29  }