package templates

import (
	"fmt"
	"testing"

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

func TestYamlDecode(t *testing.T) {
	for _, value := range []string{`value`, `!value`, `@value`, `"value"`, `"va"lue"`, `va'lue`, `@va'lue`, `0815`, `"0815"`, `4711`, `#pwd`, ``} {
		t.Run(value, func(t *testing.T) {
			quoted := yamlQuote(value)
			input := fmt.Sprintf("key: %s", quoted)

			var res struct {
				Value string `yaml:"key"`
			}

			err := yaml.Unmarshal([]byte(input), &res)
			require.NoError(t, err)
			assert.Equal(t, value, res.Value)
		})
	}
}

func TestYamlDecodeLeadingZero(t *testing.T) {
	exp := "'0815'"

	if res := yamlQuote("0815"); res != exp {
		t.Fatalf("expected %s, got %s", exp, res)
	}
}