databricks-cli/libs/dyn/yamlsaver/utils_test.go

package yamlsaver

import (
	"testing"

	"github.com/databricks/cli/libs/dyn"
	assert "github.com/databricks/cli/libs/dyn/dynassert"
)

func TestConvertToMapValueWithOrder(t *testing.T) {
	type test struct {
		Name            string            `json:"name"`
		Map             map[string]string `json:"map"`
		List            []string          `json:"list"`
		LongNameField   string            `json:"long_name_field"`
		ForceSendFields []string          `json:"-"`
		Format          string            `json:"format"`
	}

	v := &test{
		Name: "test",
		Map: map[string]string{
			"key1": "value1",
			"key2": "value2",
		},
		List: []string{"a", "b", "c"},
		ForceSendFields: []string{
			"Name",
		},
		LongNameField: "long name goes here",
	}
	result, err := ConvertToMapValue(v, NewOrder([]string{"list", "name", "map"}), []string{"format"}, map[string]dyn.Value{})
	assert.NoError(t, err)

	assert.Equal(t, dyn.V(map[string]dyn.Value{
		"list": dyn.NewValue([]dyn.Value{
			dyn.V("a"),
			dyn.V("b"),
			dyn.V("c"),
		}, dyn.Location{Line: -3}),
		"name": dyn.NewValue("test", dyn.Location{Line: -2}),
		"map": dyn.NewValue(map[string]dyn.Value{
			"key1": dyn.V("value1"),
			"key2": dyn.V("value2"),
		}, dyn.Location{Line: -1}),
		"long_name_field": dyn.NewValue("long name goes here", dyn.Location{Line: 1}),
	}), result)
}