From cb3e2f336d7bcebb62c129e7d6a962a863593c76 Mon Sep 17 00:00:00 2001
From: aaron <462826@qq.com>
Date: Thu, 21 Jul 2022 13:22:25 +0800
Subject: [PATCH] fix load config and zip results issues for unittest

---
 pkg/lib/file/file.go | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/pkg/lib/file/file.go b/pkg/lib/file/file.go
index 32cb401b..b3e787d5 100644
--- a/pkg/lib/file/file.go
+++ b/pkg/lib/file/file.go
@@ -245,11 +245,16 @@ func GetWorkDir() string { // where we run file in
 }
 
 func GetZTFDir() (dir string) { // where ztf exe file in
+	exeDir, _ := os.Executable()
+
 	if commonUtils.IsRelease() { // release
-		dir, _ = os.Executable()
-		dir = filepath.Dir(dir)
-	} else { // debug
-		dir = GetWorkDir()
+		dir = filepath.Dir(exeDir)
+	} else { // debug mode
+		if strings.Index(strings.ToLower(exeDir), "goland") > -1 { // run with ide
+			dir = os.Getenv("ZTF_CODE_DIR")
+		} else {
+			dir = GetWorkDir()
+		}
 	}
 
 	dir, _ = filepath.Abs(dir)
-- 
GitLab