fix(tests): deflake remaining integration flakes

Author: ValentaTomas

tests/integration/internal/tests/envd/process_test.go

@@ -10,6 +10,7 @@ import (
 	"github.com/stretchr/testify/require"
 
 	"github.com/e2b-dev/infra/packages/shared/pkg/grpc/envd/process"
+	"github.com/e2b-dev/infra/tests/integration/internal/api"
 	"github.com/e2b-dev/infra/tests/integration/internal/setup"
 	"github.com/e2b-dev/infra/tests/integration/internal/utils"
 )
@@ -20,15 +21,28 @@ func TestCommandKillNextApp(t *testing.T) {
 	defer cancel()
 
 	client := setup.GetAPIClient()
-	sbx := utils.SetupSandboxWithCleanup(t, client, utils.WithTimeout(300))
-
-	envdClient := setup.GetEnvdClient(t, ctx)
 
-	// Run `npx create-next-app`
+	// Pre-bake create-next-app into a template build (fresh-boot VM) instead of
+	// running the page-fault-heavy npm install in the restored sandbox, where
+	// host load has caused guest soft lockups and envd stream EOFs.
 	// Pinned to the latest known-good stable release; `@latest` currently resolves
 	// to a 16.3.0 pre-release that ships a broken SWC binary and 404s on preview.
-	err := utils.ExecCommand(t, ctx, sbx, envdClient, "npx", "create-next-app@16.2.8", "nextapp", "--yes")
-	require.NoError(t, err)
+	tmpl := utils.BuildTemplate(t, utils.TemplateBuildOptions{
+		Name:    "kill-next-app",
+		Timeout: 10 * time.Minute,
+		BuildData: api.TemplateBuildStartV2{
+			FromTemplate: &setup.SandboxTemplateID,
+			Steps: &[]api.TemplateStep{{
+				Type: "RUN",
+				Args: &[]string{"cd /home/user && npx create-next-app@16.2.8 nextapp --yes", "user"},
+			}},
+		},
+		ReqEditors: []api.RequestEditorFn{setup.WithAPIKey()},
+	})
+
+	sbx := utils.SetupSandboxWithCleanup(t, client, utils.WithTimeout(300), utils.WithTemplateID(tmpl.TemplateID))
+
+	envdClient := setup.GetEnvdClient(t, ctx)
 
 	// Run `npm run dev` in background
 	cwd := "~/nextapp"

tests/integration/internal/tests/orchestrator/sandbox_object_not_found_test.go

@@ -22,7 +22,10 @@ func TestSandboxObjectNotFound(t *testing.T) {
 
 	client := setup.GetOrchestratorClient(t, ctx)
 
-	for range 10 {
+	// Under full host load (parallel 100-sandbox tests) the orchestrator can
+	// report ResourceExhausted for minutes before template validation is
+	// reached, so keep the retry window generous.
+	for range 36 {
 		_, err := client.Create(ctx, &orchestrator.SandboxCreateRequest{
 			Sandbox: &orchestrator.SandboxConfig{
 				TemplateId:          "nonexistent-template-id",
@@ -67,6 +70,6 @@ func TestSandboxObjectNotFound(t *testing.T) {
 		return
 	}
 
-	t.Log("failed to create sandbox after 10 retries")
+	t.Log("failed to create sandbox after 36 retries")
 	t.FailNow()
 }

tests/integration/internal/tests/proxies/mask_request_host_test.go

@@ -30,25 +30,21 @@ func TestMaskRequestHostAPIParameter(t *testing.T) {
 	// Create sandbox with maskRequestHost set
 	sbx := utils.SetupSandboxWithCleanup(t, c, utils.WithNetwork(sbxNet), utils.WithTimeout(120))
 
-	// run sudo apt-get update; sudo apt-get install -y netcat
-	// run nc -l -p 8080
 	envdClient := setup.GetEnvdClient(t, ctx)
 
-	// Install netcat for the test
-	err := utils.ExecCommandAsRoot(t, ctx, sbx, envdClient, "apt-get", "update")
-	require.NoError(t, err)
-	err = utils.ExecCommandAsRoot(t, ctx, sbx, envdClient, "apt-get", "install", "-y", "netcat-traditional")
-	require.NoError(t, err)
-
 	port := 8080
-	// Start a netcat listener on port that outputs request headers to a file
+	// Single-shot listener that dumps the raw request to a file. Uses the
+	// preinstalled python instead of installing netcat via apt-get, which is
+	// page-fault-heavy enough to wedge the sandbox under parallel test load.
 	outputFile := "/tmp/nc_output.txt"
+	listener := fmt.Sprintf(
+		`import socket; s=socket.socket(); s.bind(("0.0.0.0",%d)); s.listen(1); c,_=s.accept(); open(%q,"wb").write(c.recv(65536))`,
+		port, outputFile)
 	go func() {
-		_ = utils.ExecCommandAsRoot(t, ctx, sbx, envdClient,
-			"/bin/bash", "-c", fmt.Sprintf("nc -l -p %d > %s", port, outputFile))
+		_ = utils.ExecCommandAsRoot(t, ctx, sbx, envdClient, "python3", "-c", listener)
 	}()
 
-	// Wait for nc to be up
+	// Wait for the listener to be up
 	time.Sleep(2 * time.Second)
 
 	// Prepare the sandbox URL using helpers