add test

falcon/app_test.go

@@ -0,0 +1,50 @@
+package falcon_test
+
+import (
+	"os"
+	"path"
+	"testing"
+
+	"github.com/pelletier/go-toml/v2"
+	"github.com/stretchr/testify/require"
+
+	"github.com/bandprotocol/falcon/falcon"
+)
+
+func TestInitConfig(t *testing.T) {
+	tmpDir := t.TempDir()
+
+	// Test InitConfig
+	app := falcon.NewApp(nil, nil, tmpDir, false, nil)
+
+	err := app.InitConfigFile(tmpDir)
+	require.NoError(t, err)
+
+	require.FileExists(t, path.Join(tmpDir, "config", "config.toml"))
+
+	// read the file
+	b, err := os.ReadFile(path.Join(tmpDir, "config", "config.toml"))
+	require.NoError(t, err)
+
+	// unmarshal data
+	actual := falcon.Config{}
+	err = toml.Unmarshal(b, &actual)
+	require.NoError(t, err)
+
+	expect := falcon.DefaultConfig()
+
+	require.Equal(t, expect, actual)
+}
+
+func TestInitExistingConfig(t *testing.T) {
+	tmpDir := t.TempDir()
+
+	// Test InitConfig
+	app := falcon.NewApp(nil, nil, tmpDir, false, nil)
+
+	err := app.InitConfigFile(tmpDir)
+	require.NoError(t, err)
+
+	err = app.InitConfigFile(tmpDir)
+	require.ErrorContains(t, err, "config already exists:")
+}

go.mod

@@ -8,15 +8,18 @@ require (
 	github.com/pelletier/go-toml/v2 v2.2.2
 	github.com/spf13/cobra v1.8.1
 	github.com/spf13/viper v1.19.0
+	github.com/stretchr/testify v1.9.0
 	go.uber.org/zap v1.27.0
 )
 
 require (
+	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
 	github.com/fsnotify/fsnotify v1.7.0 // indirect
 	github.com/hashicorp/hcl v1.0.0 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect
 	github.com/magiconair/properties v1.8.7 // indirect
 	github.com/mitchellh/mapstructure v1.5.0 // indirect
+	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
 	github.com/sagikazarmark/locafero v0.4.0 // indirect
 	github.com/sagikazarmark/slog-shim v0.1.0 // indirect
 	github.com/sourcegraph/conc v0.3.0 // indirect