mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-30 03:32:59 +03:00
84 lines
1.9 KiB
Go
84 lines
1.9 KiB
Go
package config_test
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/neilotoole/sq/cli/config"
|
|
"github.com/neilotoole/sq/testh/proj"
|
|
)
|
|
|
|
func TestFileStore_Nil_Save(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var f *config.YAMLFileStore
|
|
|
|
// noinspection GoNilness
|
|
err := f.Save(config.New())
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestFileStore_LoadSaveLoad(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
// good.01.sq.yml has a bunch of fixtures in it
|
|
fs := &config.YAMLFileStore{Path: "testdata/good.01.sq.yml", HookLoad: hookExpand}
|
|
const expectGood01SrcCount = 34
|
|
|
|
cfg, err := fs.Load()
|
|
require.NoError(t, err)
|
|
require.NotNil(t, cfg)
|
|
require.NotNil(t, cfg.Sources)
|
|
require.Equal(t, expectGood01SrcCount, len(cfg.Sources.Items()))
|
|
|
|
f, err := ioutil.TempFile("", "*.sq.yml")
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { assert.NoError(t, f.Close()) })
|
|
|
|
fs.Path = f.Name()
|
|
t.Logf("writing to tmp file: %s", fs.Path)
|
|
|
|
err = fs.Save(cfg)
|
|
require.NoError(t, err)
|
|
|
|
cfg2, err := fs.Load()
|
|
require.NoError(t, err)
|
|
require.NotNil(t, cfg2)
|
|
require.Equal(t, expectGood01SrcCount, len(cfg2.Sources.Items()))
|
|
require.EqualValues(t, cfg, cfg2)
|
|
}
|
|
|
|
// hookExpand expands variables in data, e.g. ${SQ_ROOT}.
|
|
var hookExpand = func(data []byte) ([]byte, error) {
|
|
return []byte(proj.Expand(string(data))), nil
|
|
}
|
|
|
|
func TestFileStore_Load(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
good, err := filepath.Glob("testdata/good.*")
|
|
require.NoError(t, err)
|
|
bad, err := filepath.Glob("testdata/bad.*")
|
|
require.NoError(t, err)
|
|
|
|
t.Logf("%d good fixtures, %d bad fixtures", len(good), len(bad))
|
|
|
|
fs := &config.YAMLFileStore{HookLoad: hookExpand}
|
|
|
|
for _, match := range good {
|
|
fs.Path = match
|
|
_, err = fs.Load()
|
|
require.NoError(t, err, match)
|
|
}
|
|
|
|
for _, match := range bad {
|
|
fs.Path = match
|
|
_, err = fs.Load()
|
|
require.Error(t, err, match)
|
|
}
|
|
}
|