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

     1  package command
     2  
     3  import (
     4  	"encoding/base64"
     5  	"testing"
     6  
     7  	"github.com/mitchellh/cli"
     8  )
     9  
    10  func TestKeygenCommand(t *testing.T) {
    11  	ui := new(cli.MockUi)
    12  	c := &KeygenCommand{Meta: Meta{Ui: ui}}
    13  	code := c.Run(nil)
    14  	if code != 0 {
    15  		t.Fatalf("bad: %d", code)
    16  	}
    17  
    18  	output := ui.OutputWriter.String()
    19  	result, err := base64.StdEncoding.DecodeString(output)
    20  	if err != nil {
    21  		t.Fatalf("err: %s", err)
    22  	}
    23  
    24  	if len(result) != 16 {
    25  		t.Fatalf("bad: %#v", result)
    26  	}
    27  }