fix '-' character trimming and expand the test

joho · Sep 2, 2024 · 2f570db · 2f570db
1 parent de05e2d
commit 2f570db
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/godotenv.go b/godotenv.go
@@ -160,10 +160,10 @@ func Write(envMap map[string]string, filename string) error {
 }
 
 func isInt(s string) bool {
+    s = strings.TrimPrefix(s, "-")
+
     if len(s) == 0 {
 		return false
-	} else if s[0] == '-' {
-		s = s[1:]
 	}
 
 	for _, r := range s {

diff --git a/godotenv_test.go b/godotenv_test.go
@@ -493,6 +493,8 @@ func TestIsInt(t *testing.T) {
     checkAndCompare("12a3", false)
     checkAndCompare("abc", false)
     checkAndCompare("12 3", false)
+    checkAndCompare("-", false)
+    checkAndCompare(" ", false)
 
     // valid values
     checkAndCompare("-123", true)