@@ -18,13 +18,9 @@ func TestLoadEnv_FindsEnvAboveFlow(t *testing.T) {
1818 require .NoError (t , os .MkdirAll (e2eDir , 0o755 ))
1919
2020 envFile := filepath .Join (projectDir , ".env" )
21- require .NoError (t , os .WriteFile (envFile , []byte ("TEST_LOAD_ENV=found_above_flow\n " ), 0o644 ))
21+ require .NoError (t , os .WriteFile (envFile , []byte ("TEST_LOAD_ENV=found_above_flow\n " ), 0o600 ))
2222
23- originalDir , err := os .Getwd ()
24- require .NoError (t , err )
25- t .Cleanup (func () { os .Chdir (originalDir ) })
26-
27- require .NoError (t , os .Chdir (e2eDir ))
23+ t .Chdir (e2eDir )
2824
2925 os .Unsetenv ("TEST_LOAD_ENV" )
3026 LoadEnv ()
@@ -41,13 +37,9 @@ func TestLoadEnv_IgnoresEnvNotAboveFlow(t *testing.T) {
4137 require .NoError (t , os .MkdirAll (subDir , 0o755 ))
4238
4339 envFile := filepath .Join (tmp , ".env" )
44- require .NoError (t , os .WriteFile (envFile , []byte ("TEST_LOAD_ENV_IGNORE=should_not_load\n " ), 0o644 ))
45-
46- originalDir , err := os .Getwd ()
47- require .NoError (t , err )
48- t .Cleanup (func () { os .Chdir (originalDir ) })
40+ require .NoError (t , os .WriteFile (envFile , []byte ("TEST_LOAD_ENV_IGNORE=should_not_load\n " ), 0o600 ))
4941
50- require . NoError ( t , os .Chdir (subDir ) )
42+ t .Chdir (subDir )
5143
5244 os .Unsetenv ("TEST_LOAD_ENV_IGNORE" )
5345 LoadEnv ()
@@ -62,11 +54,7 @@ func TestLoadEnv_NoEnvFile(t *testing.T) {
6254 e2eDir := filepath .Join (tmp , "project" , "flow" , "e2e" )
6355 require .NoError (t , os .MkdirAll (e2eDir , 0o755 ))
6456
65- originalDir , err := os .Getwd ()
66- require .NoError (t , err )
67- t .Cleanup (func () { os .Chdir (originalDir ) })
68-
69- require .NoError (t , os .Chdir (e2eDir ))
57+ t .Chdir (e2eDir )
7058
7159 os .Unsetenv ("TEST_LOAD_ENV_NONE" )
7260 LoadEnv ()
@@ -84,13 +72,9 @@ func TestLoadEnv_EnvInsideFlowIgnored(t *testing.T) {
8472 require .NoError (t , os .MkdirAll (e2eDir , 0o755 ))
8573
8674 envFile := filepath .Join (flowDir , ".env" )
87- require .NoError (t , os .WriteFile (envFile , []byte ("TEST_LOAD_ENV_INSIDE=inside_flow\n " ), 0o644 ))
75+ require .NoError (t , os .WriteFile (envFile , []byte ("TEST_LOAD_ENV_INSIDE=inside_flow\n " ), 0o600 ))
8876
89- originalDir , err := os .Getwd ()
90- require .NoError (t , err )
91- t .Cleanup (func () { os .Chdir (originalDir ) })
92-
93- require .NoError (t , os .Chdir (e2eDir ))
77+ t .Chdir (e2eDir )
9478
9579 os .Unsetenv ("TEST_LOAD_ENV_INSIDE" )
9680 LoadEnv ()
@@ -107,13 +91,9 @@ func TestLoadEnv_FromConnectorsSubdir(t *testing.T) {
10791 require .NoError (t , os .MkdirAll (connectorsDir , 0o755 ))
10892
10993 envFile := filepath .Join (projectDir , ".env" )
110- require .NoError (t , os .WriteFile (envFile , []byte ("TEST_LOAD_ENV_CONNECTORS=from_connectors\n " ), 0o644 ))
111-
112- originalDir , err := os .Getwd ()
113- require .NoError (t , err )
114- t .Cleanup (func () { os .Chdir (originalDir ) })
94+ require .NoError (t , os .WriteFile (envFile , []byte ("TEST_LOAD_ENV_CONNECTORS=from_connectors\n " ), 0o600 ))
11595
116- require . NoError ( t , os .Chdir (connectorsDir ) )
96+ t .Chdir (connectorsDir )
11797
11898 os .Unsetenv ("TEST_LOAD_ENV_CONNECTORS" )
11999 LoadEnv ()
@@ -130,13 +110,9 @@ func TestLoadEnv_RunFromFlowDirectly(t *testing.T) {
130110 require .NoError (t , os .MkdirAll (flowDir , 0o755 ))
131111
132112 envFile := filepath .Join (projectDir , ".env" )
133- require .NoError (t , os .WriteFile (envFile , []byte ("TEST_LOAD_ENV_FROM_FLOW=from_flow\n " ), 0o644 ))
134-
135- originalDir , err := os .Getwd ()
136- require .NoError (t , err )
137- t .Cleanup (func () { os .Chdir (originalDir ) })
113+ require .NoError (t , os .WriteFile (envFile , []byte ("TEST_LOAD_ENV_FROM_FLOW=from_flow\n " ), 0o600 ))
138114
139- require . NoError ( t , os .Chdir (flowDir ) )
115+ t .Chdir (flowDir )
140116
141117 os .Unsetenv ("TEST_LOAD_ENV_FROM_FLOW" )
142118 LoadEnv ()
0 commit comments