zrepl/config/config_test.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)
	}

	paths = append(paths, "../packaging/systemd-default-zrepl.yml")

	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)
		})
	}

}

func TestEmptyConfig(t *testing.T) {
	cases := []string{
		"",
		"\n",
		"---",
		"---\n",
	}
	for _, input := range cases {
		config := testValidConfig(t, input)
		require.NotNil(t, config)
		require.NotNil(t, config.Global)
		require.Empty(t, config.Jobs)
	}
}