allow home astro config file location to be override via env var (#1122)

* allow home astro config file location to be override via env var

* add and fix unit tests & fixed case when setting & unsetting the env var

config/cloud_test.go

@@ -28,8 +28,7 @@ contexts:
     last_used_workspace: ck05r3bor07h40d02y2hw4n4v
     workspace: ck05r3bor07h40d02y2hw4n4v
 `)
-	HomeConfigFile = "./test/config.yaml"
-	_ = afero.WriteFile(fs, "./test/config.yaml", configRaw, 0o777)
+	_ = afero.WriteFile(fs, HomeConfigFile, configRaw, 0o777)
 	InitConfig(fs)
 }
 

config/config.go

@@ -106,6 +106,15 @@ func initHome(fs afero.Fs) {
 	viperHome.SetFs(fs)
 	viperHome.SetConfigName(ConfigFileName)
 	viperHome.SetConfigType(ConfigFileType)
+
+	configPath := os.Getenv("ASTRO_HOME")
+	if configPath != "" {
+		HomeConfigPath = filepath.Join(configPath, ConfigDir)
+		HomeConfigFile = filepath.Join(HomeConfigPath, ConfigFileNameWithExt)
+	} else {
+		HomeConfigPath = filepath.Join(HomePath, ConfigDir)
+		HomeConfigFile = filepath.Join(HomeConfigPath, ConfigFileNameWithExt)
+	}
 	viperHome.SetConfigFile(HomeConfigFile)
 
 	for _, cfg := range CFGStrMap {

config/config_test.go

@@ -30,10 +30,20 @@ func TestIsProjectDir(t *testing.T) {
 	}
 }
 
-func TestInitHome(t *testing.T) {
+func TestInitHomeDefaultCase(t *testing.T) {
 	fs := afero.NewMemMapFs()
 	initHome(fs)
-	assert.Contains(t, viperHome.ConfigFileUsed(), "config.yaml")
+	homeDir, err := fileutil.GetHomeDir()
+	assert.NoError(t, err)
+	assert.Equal(t, filepath.Join(homeDir, ".astro", "config.yaml"), viperHome.ConfigFileUsed())
+}
+
+func TestInitHomeConfigOverride(t *testing.T) {
+	fs := afero.NewMemMapFs()
+	os.Setenv("ASTRO_HOME", "test")
+	initHome(fs)
+	assert.Equal(t, filepath.Join("test", ".astro", "config.yaml"), viperHome.ConfigFileUsed())
+	os.Unsetenv("ASTRO_HOME")
 }
 
 func TestInitProject(t *testing.T) {