package main

import (
	"os"
	"path"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestReadsVersionFromYaml(t *testing.T) {
	tmp := t.TempDir()
	path := path.Join(tmp, "test.yml")
	f, err := os.Create(path)
	assert.NoError(t, err)
	f.WriteString("yaml_test_ver: v2137")
	f.Close()

	p := YamlVersionProvider{
		File: path,
		Key:  "yaml_test_ver",
	}
	v, err := p.GetVersion()
	assert.NoError(t, err)
	assert.Equal(t, "v2137", v.String())
}