Fix root dir

AanZee · Aug 26, 2020 · f44dc33 · f44dc33
1 parent 9205d38
commit f44dc33
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 20 deletions.
diff --git a/goimportssort.go b/goimportssort.go
@@ -331,19 +331,9 @@ func isStandardPackage(pkg string) bool {
 	return ok
 }
 
-func getRootPath() (string, error) {
-	_, b, _, ok := runtime.Caller(0)
-	if !ok {
-		return "", errors.New("could not get root directory")
-	}
-	basepath := filepath.Dir(b)
-
-	return basepath, nil
-}
-
 // getModuleName parses the GOMOD name
 func getModuleName() string {
-	root, err := getRootPath()
+	root, err := os.Getwd()
 	if err != nil {
 		log.Println("error when getting root path: ", err)
 		return ""

diff --git a/goimportssort_test.go b/goimportssort_test.go
@@ -189,15 +189,6 @@ func main() {
 `, string(output))
 }
 
-func TestGetRootPath(t *testing.T) {
-	asserts := assert.New(t)
-
-	path, err := getRootPath()
-
-	asserts.Nil(err)
-	asserts.Contains(path, "goimportssort")
-}
-
 func TestGetModuleName(t *testing.T) {
 	asserts := assert.New(t)