Set test environment with t.Setenv

replicate · Jan 23, 2025 · d9562ec · d9562ec
1 parent 78e6236
commit d9562ec
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 20 deletions.
diff --git a/pkg/docker/fast_push_test.go b/pkg/docker/fast_push_test.go
@@ -21,10 +21,8 @@ func TestFastPush(t *testing.T) {
 	defer server.Close()
 	url, err := url.Parse(server.URL)
 	require.NoError(t, err)
-	err = os.Setenv(schemeEnv, url.Scheme)
-	require.NoError(t, err)
-	err = os.Setenv(hostEnv, url.Host)
-	require.NoError(t, err)
+	t.Setenv(schemeEnv, url.Scheme)
+	t.Setenv(hostEnv, url.Host)
 
 	// Create directories
 	dir := t.TempDir()
@@ -41,10 +39,4 @@ func TestFastPush(t *testing.T) {
 	// Run fast push
 	err = FastPush("test", dir, command, context.Background())
 	require.NoError(t, err)
-
-	// Cleanup env
-	err = os.Unsetenv(schemeEnv)
-	require.NoError(t, err)
-	err = os.Unsetenv(hostEnv)
-	require.NoError(t, err)
 }
diff --git a/pkg/docker/push_test.go b/pkg/docker/push_test.go
@@ -20,10 +20,8 @@ func TestPush(t *testing.T) {
 	defer server.Close()
 	url, err := url.Parse(server.URL)
 	require.NoError(t, err)
-	err = os.Setenv(schemeEnv, url.Scheme)
-	require.NoError(t, err)
-	err = os.Setenv(hostEnv, url.Host)
-	require.NoError(t, err)
+	t.Setenv(schemeEnv, url.Scheme)
+	t.Setenv(hostEnv, url.Host)
 
 	// Create directories
 	dir := t.TempDir()
@@ -40,10 +38,4 @@ func TestPush(t *testing.T) {
 	// Run fast push
 	err = Push("test", true, dir, command)
 	require.NoError(t, err)
-
-	// Cleanup env
-	err = os.Unsetenv(schemeEnv)
-	require.NoError(t, err)
-	err = os.Unsetenv(hostEnv)
-	require.NoError(t, err)
 }