131 lines
2.8 KiB
Go
131 lines
2.8 KiB
Go
|
package jobspec2
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/hashicorp/nomad/jobspec"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestEquivalentToHCL1(t *testing.T) {
|
||
|
|
hclSpecDir := "../jobspec/test-fixtures/"
|
||
|
|
fis, err := ioutil.ReadDir(hclSpecDir)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
for _, fi := range fis {
|
||
|
|
name := fi.Name()
|
||
|
|
|
||
|
|
t.Run(name, func(t *testing.T) {
|
||
|
|
f, err := os.Open(hclSpecDir + name)
|
||
|
|
require.NoError(t, err)
|
||
|
|
defer f.Close()
|
||
|
|
|
||
|
|
job1, err := jobspec.Parse(f)
|
||
|
|
if err != nil {
|
||
|
|
t.Skip("file is not parsable in v1")
|
||
|
|
}
|
||
|
|
|
||
|
|
f.Seek(0, 0)
|
||
|
|
|
||
|
|
job2, err := Parse(name, f)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
require.Equal(t, job1, job2)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParse_VarsAndFunctions(t *testing.T) {
|
||
|
|
hcl := `
|
||
|
|
job "example" {
|
||
|
|
datacenters = [for s in ["dc1", "dc2"] : upper(s)]
|
||
|
|
region = vars.region_var
|
||
|
|
}
|
||
|
|
`
|
||
|
|
|
||
|
|
out, err := ParseWithArgs("input.hcl", strings.NewReader(hcl), map[string]string{"region_var": "aug"}, true)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
require.Equal(t, []string{"DC1", "DC2"}, out.Datacenters)
|
||
|
|
require.NotNil(t, out.Region)
|
||
|
|
require.Equal(t, "aug", *out.Region)
|
||
|
|
}
|
||
|
|
|
||
|
|
// TestParse_UnknownVariables asserts that unknown variables are left intact for further processing
|
||
|
|
func TestParse_UnknownVariables(t *testing.T) {
|
||
|
|
hcl := `
|
||
|
|
job "example" {
|
||
|
|
datacenters = [for s in ["dc1", "dc2"] : upper(s)]
|
||
|
|
region = vars.region_var
|
||
|
|
meta {
|
||
|
|
known_var = "${vars.region_var}"
|
||
|
|
unknown_var = "${UNKNOWN}"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
`
|
||
|
|
|
||
|
|
out, err := ParseWithArgs("input.hcl", strings.NewReader(hcl), map[string]string{"region_var": "aug"}, true)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
meta := map[string]string{
|
||
|
|
"known_var": "aug",
|
||
|
|
"unknown_var": "${UNKNOWN}",
|
||
|
|
}
|
||
|
|
|
||
|
|
require.Equal(t, meta, out.Meta)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParse_FileOperators(t *testing.T) {
|
||
|
|
hcl := `
|
||
|
|
job "example" {
|
||
|
|
region = file("parse_test.go")
|
||
|
|
}
|
||
|
|
`
|
||
|
|
|
||
|
|
t.Run("enabled", func(t *testing.T) {
|
||
|
|
out, err := ParseWithArgs("input.hcl", strings.NewReader(hcl), nil, true)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
expected, err := ioutil.ReadFile("parse_test.go")
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
require.NotNil(t, out.Region)
|
||
|
|
require.Equal(t, string(expected), *out.Region)
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("disabled", func(t *testing.T) {
|
||
|
|
_, err := ParseWithArgs("input.hcl", strings.NewReader(hcl), nil, false)
|
||
|
|
require.Error(t, err)
|
||
|
|
require.Contains(t, err.Error(), "filesystem function disabled")
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParseDynamic(t *testing.T) {
|
||
|
|
hcl := `
|
||
|
|
job "example" {
|
||
|
|
|
||
|
|
dynamic "group" {
|
||
|
|
for_each = ["groupA", "groupB", "groupC"]
|
||
|
|
labels = [group.value]
|
||
|
|
|
||
|
|
content {
|
||
|
|
task "simple" {
|
||
|
|
driver = "raw_exec"
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
`
|
||
|
|
out, err := ParseWithArgs("input.hcl", strings.NewReader(hcl), nil, true)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
require.Len(t, out.TaskGroups, 3)
|
||
|
|
require.Equal(t, "groupA", *out.TaskGroups[0].Name)
|
||
|
|
require.Equal(t, "groupB", *out.TaskGroups[1].Name)
|
||
|
|
require.Equal(t, "groupC", *out.TaskGroups[2].Name)
|
||
|
|
}
|