validate from stdin

This commit is contained in:
Alex Dadgar 2016-07-22 13:48:14 -07:00
parent 34759dab86
commit 818d671407
2 changed files with 73 additions and 5 deletions

View File

@ -2,6 +2,8 @@ package command
import (
"fmt"
"io"
"os"
"strings"
"github.com/hashicorp/nomad/jobspec"
@ -9,6 +11,9 @@ import (
type ValidateCommand struct {
Meta
// The fields below can be overwritten for tests
testStdin io.Reader
}
func (c *ValidateCommand) Help() string {
@ -18,6 +23,9 @@ Usage: nomad validate [options] <file>
Checks if a given HCL job file has a valid specification. This can be used to
check for any syntax errors or validation problems with a job.
If the supplied path is "-", the jobfile is read from stdin. Otherwise
it is read from the file at the supplied path.
`
return strings.TrimSpace(helpText)
}
@ -39,12 +47,33 @@ func (c *ValidateCommand) Run(args []string) int {
c.Ui.Error(c.Help())
return 1
}
file := args[0]
// Parse the job file
job, err := jobspec.ParseFile(file)
// Read the Jobfile
path := args[0]
var f io.Reader
switch path {
case "-":
if c.testStdin != nil {
f = c.testStdin
} else {
f = os.Stdin
}
path = "stdin"
default:
file, err := os.Open(path)
defer file.Close()
if err != nil {
c.Ui.Error(fmt.Sprintf("Error opening file %q: %v", path, err))
return 1
}
f = file
}
// Parse the JobFile
job, err := jobspec.Parse(f)
if err != nil {
c.Ui.Error(fmt.Sprintf("Error parsing job file %s: %s", file, err))
c.Ui.Error(fmt.Sprintf("Error parsing job file from %s: %v", path, err))
return 1
}

View File

@ -63,7 +63,7 @@ func TestValidateCommand_Fails(t *testing.T) {
if code := cmd.Run([]string{"/unicorns/leprechauns"}); code != 1 {
t.Fatalf("expect exit 1, got: %d", code)
}
if out := ui.ErrorWriter.String(); !strings.Contains(out, "Error parsing") {
if out := ui.ErrorWriter.String(); !strings.Contains(out, "Error opening") {
t.Fatalf("expect parsing error, got: %s", out)
}
ui.ErrorWriter.Reset()
@ -102,3 +102,42 @@ func TestValidateCommand_Fails(t *testing.T) {
}
ui.ErrorWriter.Reset()
}
func TestValidateCommand_From_STDIN(t *testing.T) {
stdinR, stdinW, err := os.Pipe()
if err != nil {
t.Fatalf("err: %s", err)
}
ui := new(cli.MockUi)
cmd := &ValidateCommand{
Meta: Meta{Ui: ui},
testStdin: stdinR,
}
go func() {
stdinW.WriteString(`
job "job1" {
type = "service"
datacenters = [ "dc1" ]
group "group1" {
count = 1
task "task1" {
driver = "exec"
resources = {
cpu = 1000
disk = 150
memory = 512
}
}
}
}`)
stdinW.Close()
}()
args := []string{"-"}
if code := cmd.Run(args); code != 0 {
t.Fatalf("expected exit code 0, got %d: %q", code, ui.ErrorWriter.String())
}
ui.ErrorWriter.Reset()
}