F

stateful · Nov 8, 2024 · 0d7e3c5 · 0d7e3c5
1 parent ab1fb3c
commit 0d7e3c5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/internal/command/command_terminal_test.go b/internal/command/command_terminal_test.go
@@ -19,6 +19,7 @@ import (
 
 	"github.com/stateful/runme/v3/internal/sbuffer"
 	"github.com/stateful/runme/v3/internal/session"
+	"github.com/stateful/runme/v3/internal/testutils"
 	runnerv2 "github.com/stateful/runme/v3/pkg/api/gen/proto/go/runme/runner/v2"
 )
 
@@ -63,7 +64,7 @@ func TestTerminalCommand_EnvPropagation(t *testing.T) {
 
 	// Wait for the prompt before sending the next command.
 	prompt := "$"
-	if os.Getenv("RUNME_TEST_ENV") == "docker" {
+	if testutils.IsDockerTestEnv() {
 		prompt = "#"
 	}
 	expectContainLines(ctx, t, stdout, []string{prompt})