github.com/vishnusomank/figtree@v0.1.0/overwrite_test.go

package figtree

import (
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"gopkg.in/yaml.v3"
)

func init() {
	StringifyValue = false
}

func TestOptionsOverwriteConfigD3(t *testing.T) {
	opts := TestOptions{}
	require.NoError(t, os.Chdir("d1/d2/d3"))
	t.Cleanup(func() {
		_ = os.Chdir("../../..")
	})

	arr1 := []StringOption{}
	arr1 = append(arr1, StringOption{tSrc("../overwrite.yml", 8, 5), true, "d2arr1val1"})
	arr1 = append(arr1, StringOption{tSrc("../overwrite.yml", 9, 5), true, "d2arr1val2"})

	expected := TestOptions{
		String1:    StringOption{tSrc("../overwrite.yml", 6, 7), true, "d2str1val1"},
		LeaveEmpty: StringOption{},
		Array1:     arr1,
		Map1: map[string]StringOption{
			"key0": {tSrc("../../overwrite.yml", 11, 9), true, "d1map1val0"},
			"key1": {tSrc("../../overwrite.yml", 12, 9), true, "d1map1val1"},
		},
		Int1:   IntOption{tSrc("../../overwrite.yml", 13, 7), true, 111},
		Float1: Float32Option{tSrc("../../overwrite.yml", 14, 9), true, 1.11},
		Bool1:  BoolOption{tSrc("../overwrite.yml", 15, 8), true, false},
	}

	fig := newFigTreeFromEnv()
	err := fig.LoadAllConfigs("overwrite.yml", &opts)
	require.NoError(t, err)
	require.Exactly(t, expected, opts)
}

func TestOptionsOverwriteConfigD2(t *testing.T) {
	opts := TestOptions{}
	require.NoError(t, os.Chdir("d1/d2"))
	t.Cleanup(func() {
		_ = os.Chdir("../..")
	})

	arr1 := []StringOption{}
	arr1 = append(arr1, StringOption{tSrc("overwrite.yml", 8, 5), true, "d2arr1val1"})
	arr1 = append(arr1, StringOption{tSrc("overwrite.yml", 9, 5), true, "d2arr1val2"})

	expected := TestOptions{
		String1:    StringOption{tSrc("overwrite.yml", 6, 7), true, "d2str1val1"},
		LeaveEmpty: StringOption{},
		Array1:     arr1,
		Map1: map[string]StringOption{
			"key0": {tSrc("../overwrite.yml", 11, 9), true, "d1map1val0"},
			"key1": {tSrc("../overwrite.yml", 12, 9), true, "d1map1val1"},
		},
		Int1:   IntOption{tSrc("../overwrite.yml", 13, 7), true, 111},
		Float1: Float32Option{tSrc("../overwrite.yml", 14, 9), true, 1.11},
		Bool1:  BoolOption{tSrc("overwrite.yml", 15, 8), true, false},
	}

	fig := newFigTreeFromEnv()
	err := fig.LoadAllConfigs("overwrite.yml", &opts)
	assert.NoError(t, err)
	assert.Exactly(t, expected, opts)
}

func TestBuiltinOverwriteConfigD3(t *testing.T) {
	opts := TestBuiltin{}
	require.NoError(t, os.Chdir("d1/d2/d3"))
	t.Cleanup(func() {
		_ = os.Chdir("../../..")
	})

	arr1 := []string{}
	arr1 = append(arr1, "d2arr1val1")
	arr1 = append(arr1, "d2arr1val2")

	expected := TestBuiltin{
		String1:    "d2str1val1",
		LeaveEmpty: "",
		Array1:     arr1,
		Map1: map[string]string{
			"key0": "d1map1val0",
			"key1": "d1map1val1",
		},
		Int1:   111,
		Float1: 1.11,
		Bool1:  false,
	}

	fig := newFigTreeFromEnv()
	err := fig.LoadAllConfigs("overwrite.yml", &opts)
	assert.NoError(t, err)
	assert.Exactly(t, expected, opts)
}

func TestBuiltinOverwriteConfigD2(t *testing.T) {
	opts := TestBuiltin{}
	require.NoError(t, os.Chdir("d1/d2"))
	t.Cleanup(func() {
		_ = os.Chdir("../..")
	})

	arr1 := []string{}
	arr1 = append(arr1, "d2arr1val1")
	arr1 = append(arr1, "d2arr1val2")

	expected := TestBuiltin{
		String1:    "d2str1val1",
		LeaveEmpty: "",
		Array1:     arr1,
		Map1: map[string]string{
			"key0": "d1map1val0",
			"key1": "d1map1val1",
		},
		Int1:   111,
		Float1: 1.11,
		Bool1:  false,
	}

	fig := newFigTreeFromEnv()
	err := fig.LoadAllConfigs("overwrite.yml", &opts)
	assert.NoError(t, err)
	assert.Exactly(t, expected, opts)
}

type TestArray struct {
	IntArr      [2]int    `yaml:"intArr"`
	PartialInt  [2]int    `yaml:"partialInt"`
	TooManyInt  [2]int    `yaml:"tooManyInt"`
	StrArr      [2]string `yaml:"strArr"`
	ToOverwrite [2]string `yaml:"toOverwrite"`
}

func TestBuiltinOverwriteArrayD2(t *testing.T) {
	opts := TestArray{}
	require.NoError(t, os.Chdir("d1/d2"))
	t.Cleanup(func() {
		_ = os.Chdir("../..")
	})

	expected := TestArray{
		IntArr:      [2]int{1, 2},
		PartialInt:  [2]int{1, 0},
		TooManyInt:  [2]int{1, 2},
		StrArr:      [2]string{"abc", "def"},
		ToOverwrite: [2]string{"c", "d"},
	}

	fig := newFigTreeFromEnv()
	err := fig.LoadAllConfigs("array.yml", &opts)
	assert.NoError(t, err)
	assert.Exactly(t, expected, opts)
}

type TestOptionsArray struct {
	IntArr      [2]IntOption    `yaml:"intArr"`
	PartialInt  [2]IntOption    `yaml:"partialInt"`
	TooManyInt  [2]IntOption    `yaml:"tooManyInt"`
	StrArr      [2]StringOption `yaml:"strArr"`
	ToOverwrite [2]StringOption `yaml:"toOverwrite"`
}

func TestOptionsOverwriteArrayD2(t *testing.T) {
	opts := TestOptionsArray{}
	require.NoError(t, os.Chdir("d1/d2"))
	t.Cleanup(func() {
		_ = os.Chdir("../..")
	})

	expected := TestOptionsArray{
		IntArr: [2]IntOption{
			{tSrc("array.yml", 1, 10), true, 1},
			{tSrc("array.yml", 1, 12), true, 2},
		},
		PartialInt: [2]IntOption{
			{tSrc("array.yml", 2, 14), true, 1},
			{NewSource(""), false, 0},
		},
		TooManyInt: [2]IntOption{
			{tSrc("array.yml", 3, 14), true, 1},
			{tSrc("array.yml", 3, 16), true, 2},
		},
		StrArr: [2]StringOption{
			{tSrc("array.yml", 4, 10), true, "abc"},
			{tSrc("array.yml", 4, 15), true, "def"},
		},
		ToOverwrite: [2]StringOption{
			{tSrc("../array.yml", 5, 15), true, "c"},
			{tSrc("../array.yml", 5, 17), true, "d"},
		},
	}

	fig := newFigTreeFromEnv()
	err := fig.LoadAllConfigs("array.yml", &opts)
	assert.NoError(t, err)
	assert.Exactly(t, expected, opts)
}

// auto-upgrade:
//   enabled: true

func TestOverwritePartialStruct(t *testing.T) {
	type MyStruct struct {
		A StringOption `yaml:"a"`
		B BoolOption   `yaml:"b"`
	}
	type data struct {
		MyStruct MyStruct `yaml:"my-struct"`
	}
	configs := []struct {
		Name string
		Body string
	}{{
		Name: "test",
		Body: `
my-struct:
  b: true
`,
	}, {
		Name: "../test",
		Body: `
config: {overwrite: [my-struct]}
my-struct:
  b: false
  a: foo
`,
	}}
	expected := data{
		MyStruct: MyStruct{
			A: StringOption{tSrc("../test", 5, 6), true, "foo"},
			B: BoolOption{tSrc("../test", 4, 6), true, false},
		},
	}
	sources := []ConfigSource{}
	for _, c := range configs {
		var node yaml.Node
		err := yaml.Unmarshal([]byte(c.Body), &node)
		require.NoError(t, err)
		sources = append(sources, ConfigSource{
			Config:   &node,
			Filename: c.Name,
		})
	}
	fig := newFigTreeFromEnv()
	got := data{}
	err := fig.LoadAllConfigSources(sources, &got)
	require.NoError(t, err)
	require.Equal(t, expected, got)
}

func TestOverwriteNil(t *testing.T) {
	type MyStruct struct {
		A StringOption `yaml:"a"`
		B BoolOption   `yaml:"b"`
	}
	type data struct {
		MyStruct MyStruct `yaml:"my-struct"`
	}
	config := `
config: {overwrite: [my-struct]}
my-struct:
`
	expected := data{
		MyStruct: MyStruct{
			A: StringOption{},
			B: BoolOption{},
		},
	}
	var node yaml.Node
	err := yaml.Unmarshal([]byte(config), &node)
	require.NoError(t, err)
	fig := newFigTreeFromEnv()
	got := data{}
	err = fig.LoadConfigSource(&node, "test", &got)
	require.NoError(t, err)
	require.Equal(t, expected, got)
}