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 }