root_test: Fix test errors

Copying the config file sometimes fails with "rename /builds/go/src/github.com/zaquestion/lab/testdata--8119903081016337361/lab.toml /home/root /.config/lab/lab.toml: no such file or directory" because there is a new-line character in the output of 'whoami'. After applying a fix to remove the new-line character I also occasionally see issues with the patch for /home/root/.config/lab not existing. Remove new line characters in the ConfigFile() return string, and make sure the path for the file exists. Signed-off-by: Prarit Bhargava <[email protected]>
zaquestion · May 4, 2021 · d6c53cf · d6c53cf
1 parent 4a94ee9
commit d6c53cf
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cmd/root_test.go b/cmd/root_test.go
@@ -175,7 +175,13 @@ func whoAmI() string {
 }
 
 func configFile() string {
-	return "/home/" + whoAmI() + "/.config/lab/lab.toml"
+	str := "/home/" + whoAmI() + "/.config/lab"
+	str = strings.Replace(str, "\n", "", -1)
+	if _, err := os.Stat(str); os.IsNotExist(err) {
+		os.MkdirAll(str, os.ModePerm)
+	}
+	str = str + "/lab.toml"
+	return str
 }
 
 // getAppOutput splits and truncates the list of strings returned from the "lab"