update test

mafredri · mafredri · commit 96f10e1275a2 · 2025-06-24T17:04:57.000Z
diff --git a/agent/agentcontainers/api_test.go b/agent/agentcontainers/api_test.go
@@ -2040,7 +2040,7 @@ func TestAPI(t *testing.T) {
 		// Verify commands were executed through the custom shell and environment.
 		require.NotEmpty(t, fakeExec.commands, "commands should be executed")
 
-		// Want: /bin/custom-shell -c "docker ps --all --quiet --no-trunc"
+		// Want: /bin/custom-shell -c '"docker" "ps" "--all" "--quiet" "--no-trunc"'
 		require.Equal(t, testShell, fakeExec.commands[0][0], "custom shell should be used")
 		if runtime.GOOS == "windows" {
 			require.Equal(t, "/c", fakeExec.commands[0][1], "shell should be called with /c on Windows")
@@ -2049,6 +2049,7 @@ func TestAPI(t *testing.T) {
 		}
 		require.Len(t, fakeExec.commands[0], 3, "command should have 3 arguments")
 		require.GreaterOrEqual(t, strings.Count(fakeExec.commands[0][2], " "), 2, "command/script should have multiple arguments")
+		require.True(t, strings.HasPrefix(fakeExec.commands[0][2], `"docker" "ps" "--all" "--quiet" "--no-trunc"`), "command should start with docker ps --all --quiet --no-trunc")
 
 		// Verify the environment was set on the command.
 		lastCmd := fakeExec.getLastCommand()
