fix account testnet environment load (#560)

* fix account testnet environment load * wip --------- Co-authored-by: rian <[email protected]>
NethermindEth · Apr 12, 2024 · ec1e9e8 · ec1e9e8
1 parent 1889b90
commit ec1e9e8
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/account/account_test.go b/account/account_test.go
@@ -45,14 +45,14 @@ var (
 func TestMain(m *testing.M) {
 	flag.StringVar(&testEnv, "env", "mock", "set the test environment")
 	flag.Parse()
-	if testEnv != "mock" {
-		if err := godotenv.Load(fmt.Sprintf(".env.%s", testEnv), ".env"); err != nil {
-			panic(fmt.Sprint("Failed to load env for ", testEnv))
-		}
+	if testEnv == "mock" {
+		return
 	}
 	base = os.Getenv("INTEGRATION_BASE")
-	if base == "" && testEnv != "mock" {
-		panic(fmt.Sprint("Failed to set INTEGRATION_BASE for ", testEnv))
+	if base == "" {
+		if err := godotenv.Load(fmt.Sprintf(".env.%s", testEnv)); err != nil {
+			panic(fmt.Sprintf("Failed to load .env.%s, err: %s", testEnv, err))
+		}
 	}
 	os.Exit(m.Run())
 }