mirror of
https://github.com/zrepl/zrepl.git
synced 2024-12-18 05:00:50 +01:00
141 lines
2.7 KiB
Go
141 lines
2.7 KiB
Go
package config
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"fmt"
|
|
"path"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"text/template"
|
|
|
|
"github.com/kr/pretty"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/zrepl/yaml-config"
|
|
)
|
|
|
|
func TestSampleConfigsAreParsedWithoutErrors(t *testing.T) {
|
|
paths, err := filepath.Glob("./samples/*")
|
|
if err != nil {
|
|
t.Errorf("glob failed: %+v", err)
|
|
}
|
|
|
|
for _, p := range paths {
|
|
|
|
if path.Ext(p) != ".yml" {
|
|
t.Logf("skipping file %s", p)
|
|
continue
|
|
}
|
|
|
|
t.Run(p, func(t *testing.T) {
|
|
c, err := ParseConfig(p)
|
|
if err != nil {
|
|
t.Errorf("error parsing %s:\n%+v", p, err)
|
|
}
|
|
|
|
t.Logf("file: %s", p)
|
|
t.Log(pretty.Sprint(c))
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// template must be a template/text template with a single '{{ . }}' as placeholder for val
|
|
//
|
|
//nolint:deadcode,unused
|
|
func testValidConfigTemplate(t *testing.T, tmpl string, val string) *Config {
|
|
tmp, err := template.New("master").Parse(tmpl)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
var buf bytes.Buffer
|
|
err = tmp.Execute(&buf, val)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return testValidConfig(t, buf.String())
|
|
}
|
|
|
|
func testValidConfig(t *testing.T, input string) *Config {
|
|
t.Helper()
|
|
conf, err := testConfig(t, input)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, conf)
|
|
return conf
|
|
}
|
|
|
|
func testConfig(t *testing.T, input string) (*Config, error) {
|
|
t.Helper()
|
|
return ParseConfigBytes([]byte(input))
|
|
}
|
|
|
|
func trimSpaceEachLineAndPad(s, pad string) string {
|
|
var out strings.Builder
|
|
scan := bufio.NewScanner(strings.NewReader(s))
|
|
for scan.Scan() {
|
|
fmt.Fprintf(&out, "%s%s\n", pad, bytes.TrimSpace(scan.Bytes()))
|
|
}
|
|
return out.String()
|
|
}
|
|
|
|
func TestTrimSpaceEachLineAndPad(t *testing.T) {
|
|
foo := `
|
|
foo
|
|
bar baz
|
|
`
|
|
assert.Equal(t, " \n foo\n bar baz\n \n", trimSpaceEachLineAndPad(foo, " "))
|
|
}
|
|
|
|
func TestCronSpec(t *testing.T) {
|
|
|
|
expectAccept := []string{
|
|
`"* * * * *"`,
|
|
`"0-10 * * * *"`,
|
|
`"* 0-5,8,12 * * *"`,
|
|
}
|
|
|
|
expectFail := []string{
|
|
`* * * *`,
|
|
``,
|
|
`23`,
|
|
`"@reboot"`,
|
|
`"@every 1h30m"`,
|
|
`"@daily"`,
|
|
`* * * * * *`,
|
|
}
|
|
|
|
for _, input := range expectAccept {
|
|
t.Run(input, func(t *testing.T) {
|
|
s := fmt.Sprintf("spec: %s\n", input)
|
|
var v struct {
|
|
Spec CronSpec
|
|
}
|
|
v.Spec.Schedule = nil
|
|
t.Logf("input:\n%s", s)
|
|
err := yaml.UnmarshalStrict([]byte(s), &v)
|
|
t.Logf("error: %T %s", err, err)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, v.Spec.Schedule)
|
|
})
|
|
}
|
|
|
|
for _, input := range expectFail {
|
|
t.Run(input, func(t *testing.T) {
|
|
s := fmt.Sprintf("spec: %s\n", input)
|
|
var v struct {
|
|
Spec CronSpec
|
|
}
|
|
v.Spec.Schedule = nil
|
|
t.Logf("input: %q", s)
|
|
err := yaml.UnmarshalStrict([]byte(s), &v)
|
|
t.Logf("error: %T %s", err, err)
|
|
require.Error(t, err)
|
|
require.Nil(t, v.Spec.Schedule)
|
|
})
|
|
}
|
|
|
|
}
|