Add test

Author: jakubno

tests/integration/internal/tests/api/sandboxes/sandbox_set_timeout_test.go

@@ -0,0 +1,87 @@
+package sandboxes
+
+import (
+	"fmt"
+	"net/http"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+	"golang.org/x/sync/errgroup"
+
+	"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"
+)
+
+func TestSandboxSetTimeoutPausingSandbox(t *testing.T) {
+	c := setup.GetAPIClient()
+
+	t.Run("test concurrent pauses", func(t *testing.T) {
+		sbx := utils.SetupSandboxWithCleanup(t, c, utils.WithAutoPause(true))
+		sbxId := sbx.SandboxID
+
+		// Pause the sandbox
+		wg := errgroup.Group{}
+		wg.Go(func() error {
+			pauseResp, err := c.PostSandboxesSandboxIDPauseWithResponse(t.Context(), sbxId, setup.WithAPIKey())
+			require.NoError(t, err)
+			require.Equal(t, http.StatusNoContent, pauseResp.StatusCode())
+
+			return nil
+		})
+
+		for range 5 {
+			time.Sleep(200 * time.Millisecond)
+			wg.Go(func() error {
+				setTimeoutResp, err := c.PostSandboxesSandboxIDTimeout(t.Context(), sbxId, api.PostSandboxesSandboxIDTimeoutJSONRequestBody{
+					Timeout: 15,
+				},
+					setup.WithAPIKey())
+
+				if err != nil {
+					return err
+				}
+
+				if setTimeoutResp.StatusCode != http.StatusNotFound {
+					return fmt.Errorf("unexpected status code: %d", setTimeoutResp.StatusCode)
+				}
+
+				return nil
+			})
+		}
+
+		err := wg.Wait()
+		require.NoError(t, err)
+	})
+
+	t.Run("pause killed sandbox", func(t *testing.T) {
+		// Create a sandbox with auto-pause disabled
+		sbx := utils.SetupSandboxWithCleanup(t, c, utils.WithAutoPause(true))
+		sbxId := sbx.SandboxID
+
+		// Kill the sandbox
+		killResp, err := c.DeleteSandboxesSandboxIDWithResponse(t.Context(), sbxId, setup.WithAPIKey())
+		require.NoError(t, err)
+		require.Equal(t, http.StatusNoContent, killResp.StatusCode())
+
+		pauseResp, err := c.PostSandboxesSandboxIDPauseWithResponse(t.Context(), sbxId, setup.WithAPIKey())
+		require.NoError(t, err)
+		require.Equal(t, http.StatusNotFound, pauseResp.StatusCode())
+	})
+
+	t.Run("pause already paused sandbox", func(t *testing.T) {
+		sbx := utils.SetupSandboxWithCleanup(t, c)
+		sbxId := sbx.SandboxID
+
+		// Pause the sandbox
+		resp, err := c.PostSandboxesSandboxIDPauseWithResponse(t.Context(), sbxId, setup.WithAPIKey())
+		require.NoError(t, err)
+		require.Equal(t, http.StatusNoContent, resp.StatusCode())
+
+		// Try to pause the sandbox again
+		resp, err = c.PostSandboxesSandboxIDPauseWithResponse(t.Context(), sbxId, setup.WithAPIKey())
+		require.NoError(t, err)
+		require.Equal(t, http.StatusConflict, resp.StatusCode())
+	})
+}