package configx

import (
	"context"
	"encoding/json"
	"io/ioutil"
	"path/filepath"
	"testing"

	"github.com/ghodss/yaml"
	"github.com/pelletier/go-toml"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestKoanfFile(t *testing.T) {
	setupFile := func(t *testing.T, fn, fc, subKey string) (*KoanfFile, context.CancelFunc) {
		dir := t.TempDir()
		fn = filepath.Join(dir, fn)
		require.NoError(t, ioutil.WriteFile(fn, []byte(fc), 0600))

		ctx, cancel := context.WithCancel(context.Background())
		kf, err := NewKoanfFileSubKey(ctx, fn, subKey)
		require.NoError(t, err)
		return kf, cancel
	}

	t.Run("case=reads json root file", func(t *testing.T) {
		v := map[string]interface{}{
			"foo": "bar",
		}
		encV, err := json.Marshal(v)
		require.NoError(t, err)

		kf, cancel := setupFile(t, "config.json", string(encV), "")
		defer cancel()

		actual, err := kf.Read()
		require.NoError(t, err)
		assert.Equal(t, v, actual)
	})

	t.Run("case=reads yaml root file", func(t *testing.T) {
		v := map[string]interface{}{
			"foo": "yaml string",
		}
		encV, err := yaml.Marshal(v)
		require.NoError(t, err)

		kf, cancel := setupFile(t, "config.yml", string(encV), "")
		defer cancel()

		actual, err := kf.Read()
		require.NoError(t, err)
		assert.Equal(t, v, actual)
	})

	t.Run("case=reads toml root file", func(t *testing.T) {
		v := map[string]interface{}{
			"foo": "toml string",
		}
		encV, err := toml.Marshal(v)
		require.NoError(t, err)

		kf, cancel := setupFile(t, "config.toml", string(encV), "")
		defer cancel()

		actual, err := kf.Read()
		require.NoError(t, err)
		assert.Equal(t, v, actual)
	})

	t.Run("case=reads json file as subkey", func(t *testing.T) {
		v := map[string]interface{}{
			"bar": "asdf",
		}
		encV, err := json.Marshal(v)
		require.NoError(t, err)

		kf, cancel := setupFile(t, "config.json", string(encV), "parent.of.config")
		defer cancel()

		actual, err := kf.Read()
		require.NoError(t, err)
		assert.Equal(t, map[string]interface{}{
			"parent": map[string]interface{}{
				"of": map[string]interface{}{
					"config": v,
				},
			},
		}, actual)
	})
}