Fix test

AccumulateNetwork · Jul 5, 2024 · a810028 · a810028
1 parent f631039
commit a810028
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/cmd/accumulated/run/config.go b/cmd/accumulated/run/config.go
@@ -137,7 +137,7 @@ func (c *Config) Save() error {
 	if c.file == "" {
 		return errors.BadRequest.With("not loaded from a file")
 	}
-	if c.fs != os.DirFS(".") {
+	if c.fs != nil && c.fs != os.DirFS(".") {
 		return errors.BadRequest.With("loaded from an immutable filesystem")
 	}
 	return c.SaveTo(c.file)

diff --git a/cmd/accumulated/run/dotenv_test.go b/cmd/accumulated/run/dotenv_test.go
@@ -77,6 +77,18 @@ func TestDotenv(t *testing.T) {
 	})
 }
 
+func TestSaveToFS(t *testing.T) {
+	fs := mkfs(map[string]string{
+		"accumulate.toml": `network = "foo"`,
+	})
+
+	cfg := new(Config)
+	require.NoError(t, cfg.LoadFromFS(fs, "accumulate.toml"))
+
+	// Ensure Save doesn't panic when loaded from [fs.FS]
+	require.Error(t, cfg.Save())
+}
+
 func mkfs(files map[string]string) fs.FS {
 	root := new(testutil.Dir)
 	for name, data := range files {