autoroll/go/config_vars/config_vars_test.go

package config_vars

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"strings"
	"testing"

	"github.com/stretchr/testify/require"
	"go.skia.org/infra/go/chrome_branch/mocks"
	"go.skia.org/infra/go/deepequal/assertdeep"
	"go.skia.org/infra/go/testutils/unittest"
)

func TestVarsValidate(t *testing.T) {
	unittest.SmallTest(t)

	test := func(fn func(*Vars), expectErr string) {
		v := FakeVars()
		fn(v)
		err := v.Validate()
		if expectErr == "" {
			require.NoError(t, err)
		} else {
			require.NotNil(t, err)
			require.True(t, strings.Contains(err.Error(), expectErr), err)
		}
	}

	// OK.
	test(func(v *Vars) {}, "")

	// Missing Branches.
	test(func(v *Vars) {
		v.Branches = nil
	}, "Branches is required")

	// Should call Branches.Validate().
	test(func(v *Vars) {
		v.Branches.Chromium = nil
	}, "Chromium is required.")
}

func TestBranchesValidate(t *testing.T) {
	unittest.SmallTest(t)

	test := func(fn func(*Branches), expectErr string) {
		b := FakeVars().Branches
		fn(b)
		err := b.Validate()
		if expectErr == "" {
			require.NoError(t, err)
		} else {
			require.NotNil(t, err)
			require.True(t, strings.Contains(err.Error(), expectErr), err)
		}
	}

	// OK.
	test(func(b *Branches) {}, "")

	// Missing Chromium.
	test(func(b *Branches) {
		b.Chromium = nil
	}, "Chromium is required")

	// Should call chrome_branch.Branches.Validate().
	test(func(b *Branches) {
		b.Chromium.Beta = nil
	}, "Beta branch is missing")
}

func TestTemplate(t *testing.T) {
	unittest.SmallTest(t)

	raw := "refs/branch-heads/{{.Branches.Chromium.Beta.Number}}"
	tmpl, err := NewTemplate(raw)
	require.NoError(t, err)
	require.NotNil(t, tmpl)
	require.NoError(t, tmpl.Validate())

	// JSON encode/decode.
	cfgJson := fmt.Sprintf(`{"tmpl":"%s"}`, raw)
	var cfg1, cfg2 struct {
		T *Template `json:"tmpl"`
	}
	require.NoError(t, json.NewDecoder(bytes.NewReader([]byte(cfgJson))).Decode(&cfg1))
	require.NoError(t, cfg1.T.Validate())
	b, err := json.Marshal(cfg1)
	require.NoError(t, err)
	require.Equal(t, cfgJson, string(b))
	require.NoError(t, json.NewDecoder(bytes.NewReader(b)).Decode(&cfg2))
	assertdeep.Equal(t, cfg1, cfg2)

	// Update() and String().
	require.Equal(t, "", tmpl.String())
	v := FakeVars()
	require.NoError(t, tmpl.Update(v))
	require.Equal(t, "refs/branch-heads/4044", tmpl.String())
	v.Branches.Chromium.Beta.Number = 5000
	require.Equal(t, "refs/branch-heads/4044", tmpl.String())
	require.NoError(t, tmpl.Update(v))
	require.Equal(t, "refs/branch-heads/5000", tmpl.String())

	// Invalid template string.
	invalid := func(raw, expect string) {
		_, err := NewTemplate(raw)
		require.NotNil(t, err)
		require.True(t, strings.Contains(err.Error(), expect), err)
	}
	invalid("", "Template is missing")
	invalid("{{.Bogus}}", "can't evaluate field Bogus")

	// Banned template string.
	invalid("refs/branch-heads/{{.Branches.Chromium.Master.Number}}", "Templates should not use \"Chromium.Master.Number\"")
}

func TestRegistry(t *testing.T) {
	unittest.SmallTest(t)

	ctx := context.Background()
	v := FakeVars()
	cbc := &mocks.Client{}
	cbc.On("Get", ctx).Return(v.Branches.Chromium, v.Branches.ActiveMilestones, nil)
	r, err := NewRegistry(ctx, cbc)
	require.NoError(t, err)

	// Register().
	t1, err := NewTemplate("refs/branch-heads/{{.Branches.Chromium.Beta.Number}}")
	require.NoError(t, err)
	require.NoError(t, r.Register(t1))
	require.Equal(t, t1.String(), "refs/branch-heads/4044")

	t2, err := NewTemplate("refs/branch-heads/{{.Branches.Chromium.Stable.Number}}")
	require.NoError(t, err)
	require.NoError(t, r.Register(t2))
	require.Equal(t, t2.String(), "refs/branch-heads/3987")

	// Update().
	v.Branches.Chromium.Beta.Number = 5000
	v.Branches.Chromium.Stable.Number = 4044
	require.NoError(t, r.Update(ctx))
	require.Equal(t, t1.String(), "refs/branch-heads/5000")
	require.Equal(t, t2.String(), "refs/branch-heads/4044")
}