github.com/jmitchell/nomad@v0.1.3-0.20151007230021-7ab84c2862d8/command/run_test.go (about) 1 package command 2 3 import ( 4 "io/ioutil" 5 "os" 6 "strings" 7 "testing" 8 9 "github.com/mitchellh/cli" 10 ) 11 12 func TestRunCommand_Implements(t *testing.T) { 13 var _ cli.Command = &RunCommand{} 14 } 15 16 func TestRunCommand_Fails(t *testing.T) { 17 ui := new(cli.MockUi) 18 cmd := &RunCommand{Meta: Meta{Ui: ui}} 19 20 // Fails on misuse 21 if code := cmd.Run([]string{"some", "bad", "args"}); code != 1 { 22 t.Fatalf("expected exit code 1, got: %d", code) 23 } 24 if out := ui.ErrorWriter.String(); !strings.Contains(out, cmd.Help()) { 25 t.Fatalf("expected help output, got: %s", out) 26 } 27 ui.ErrorWriter.Reset() 28 29 // Fails when specified file does not exist 30 if code := cmd.Run([]string{"/unicorns/leprechauns"}); code != 1 { 31 t.Fatalf("expect exit 1, got: %d", code) 32 } 33 if out := ui.ErrorWriter.String(); !strings.Contains(out, "Error parsing") { 34 t.Fatalf("expect parsing error, got: %s", out) 35 } 36 ui.ErrorWriter.Reset() 37 38 // Fails on invalid HCL 39 fh1, err := ioutil.TempFile("", "nomad") 40 if err != nil { 41 t.Fatalf("err: %s", err) 42 } 43 defer os.Remove(fh1.Name()) 44 if _, err := fh1.WriteString("nope"); err != nil { 45 t.Fatalf("err: %s", err) 46 } 47 if code := cmd.Run([]string{fh1.Name()}); code != 1 { 48 t.Fatalf("expect exit 1, got: %d", code) 49 } 50 if out := ui.ErrorWriter.String(); !strings.Contains(out, "Error parsing") { 51 t.Fatalf("expect parsing error, got: %s", err) 52 } 53 ui.ErrorWriter.Reset() 54 55 // Fails on invalid job spec 56 fh2, err := ioutil.TempFile("", "nomad") 57 if err != nil { 58 t.Fatalf("err: %s", err) 59 } 60 defer os.Remove(fh2.Name()) 61 if _, err := fh2.WriteString(`job "job1" {}`); err != nil { 62 t.Fatalf("err: %s", err) 63 } 64 if code := cmd.Run([]string{fh2.Name()}); code != 1 { 65 t.Fatalf("expect exit 1, got: %d", code) 66 } 67 if out := ui.ErrorWriter.String(); !strings.Contains(out, "Error validating") { 68 t.Fatalf("expect validation error, got: %s", out) 69 } 70 ui.ErrorWriter.Reset() 71 72 // Fails on connection failure (requires a valid job) 73 fh3, err := ioutil.TempFile("", "nomad") 74 if err != nil { 75 t.Fatalf("err: %s", err) 76 } 77 defer os.Remove(fh3.Name()) 78 _, err = fh3.WriteString(` 79 job "job1" { 80 datacenters = [ "dc1" ] 81 group "group1" { 82 count = 1 83 task "task1" { 84 driver = "exec" 85 resources = { 86 cpu = 1000 87 mem = 512 88 } 89 } 90 } 91 }`) 92 if err != nil { 93 t.Fatalf("err: %s", err) 94 } 95 if code := cmd.Run([]string{"-address=nope", fh3.Name()}); code != 1 { 96 t.Fatalf("expected exit code 1, got: %d", code) 97 } 98 if out := ui.ErrorWriter.String(); !strings.Contains(out, "Error submitting job") { 99 t.Fatalf("expected failed query error, got: %s", out) 100 } 101 }