Browse Source

pkg/flags: fix "TestSetFlagsFromEnvParsingError"

Signed-off-by: Gyuho Lee <leegyuho@amazon.com>
Gyuho Lee 7 years ago
parent
commit
7b32c07899
1 changed files with 10 additions and 2 deletions
  1. 10 2
      pkg/flags/flag_test.go

+ 10 - 2
pkg/flags/flag_test.go

@@ -17,6 +17,7 @@ package flags
 import (
 import (
 	"flag"
 	"flag"
 	"os"
 	"os"
+	"strings"
 	"testing"
 	"testing"
 )
 )
 
 
@@ -80,7 +81,14 @@ func TestSetFlagsFromEnvParsingError(t *testing.T) {
 	}
 	}
 	defer os.Unsetenv("ETCD_HEARTBEAT_INTERVAL")
 	defer os.Unsetenv("ETCD_HEARTBEAT_INTERVAL")
 
 
-	if serr := SetFlagsFromEnv("ETCD", fs); serr.Error() != `invalid value "100 # ms" for ETCD_HEARTBEAT_INTERVAL: strconv.ParseUint: parsing "100 # ms": invalid syntax` {
-		t.Fatalf("expected parsing error, got %v", serr)
+	err := SetFlagsFromEnv("ETCD", fs)
+	for _, v := range []string{"invalid syntax", "parse error"} {
+		if strings.Contains(err.Error(), v) {
+			err = nil
+			break
+		}
+	}
+	if err != nil {
+		t.Fatalf("unexpected error %v", err)
 	}
 	}
 }
 }