9 "github.com/stretchr/testify/assert"
10 "github.com/stretchr/testify/require"
13 func ensureFiles(t *testing.T, rootDir string, files ...string) {
14 for _, f := range files {
15 p := rootify(rootDir, f)
21 func TestEnsureRoot(t *testing.T) {
22 assert, require := assert.New(t), require.New(t)
24 // setup temp dir for test
25 tmpDir, err := ioutil.TempDir("", "config-test")
27 defer os.RemoveAll(tmpDir)
30 EnsureRoot(tmpDir, "mainnet")
32 // make sure config is set properly
33 data, err := ioutil.ReadFile(filepath.Join(tmpDir, "config.toml"))
35 assert.Equal([]byte(selectNetwork("mainnet")), data)
37 ensureFiles(t, tmpDir, "data")