chore: import upstream snapshot with attribution

2026-07-13 12:47:58 +08:00
commit b16403ea71
789 changed files with 115226 additions and 0 deletions
@@ -0,0 +1,188 @@
+import { randomBytes } from 'node:crypto'
+import { describe, expect, test } from 'vitest'
+import { Sandbox } from 'e2b'
+import { getUserConfig } from 'src/user'
+import {
+  type CliRunResult,
+  bufferToText,
+  isDebug,
+  parseEnvInt,
+  runCliWithPipedStdin,
+} from '../../setup'
+
+type PipeCase = {
+  name: string
+  data: Buffer
+  expectedBytes: number
+  timeoutMs?: number
+}
+
+type UserConfigWithDomain = NonNullable<ReturnType<typeof getUserConfig>> & {
+  domain?: string
+  E2B_DOMAIN?: string
+}
+
+const userConfig = safeGetUserConfig() as UserConfigWithDomain | null
+const domain =
+  process.env.E2B_DOMAIN ||
+  userConfig?.E2B_DOMAIN ||
+  userConfig?.domain ||
+  'e2b.app'
+const apiKey = process.env.E2B_API_KEY || userConfig?.teamApiKey
+const shouldSkip = !apiKey || isDebug
+const integrationTest = test.skipIf(shouldSkip)
+const templateId =
+  process.env.E2B_PIPE_TEMPLATE_ID ||
+  process.env.E2B_TEMPLATE_ID ||
+  'base'
+const includeLargeBinary =
+  process.env.E2B_PIPE_INTEGRATION_STRICT === '1' ||
+  process.env.E2B_PIPE_INTEGRATION_BINARY === '1' ||
+  process.env.STRICT === '1'
+const sandboxTimeoutMs = parseEnvInt('E2B_PIPE_SANDBOX_TIMEOUT_MS', 10_000)
+const testTimeoutMs = parseEnvInt('E2B_PIPE_TEST_TIMEOUT_MS', 60_000)
+const defaultCmdTimeoutMs = parseEnvInt(
+  'E2B_PIPE_CMD_TIMEOUT_MS',
+  Math.min(8_000, testTimeoutMs)
+)
+const cliEnv: NodeJS.ProcessEnv = {
+  ...process.env,
+  E2B_DOMAIN: domain,
+  E2B_API_KEY: apiKey,
+}
+
+delete cliEnv.E2B_DEBUG
+
+const defaultCases: PipeCase[] = [
+  {
+    name: 'empty_eof',
+    data: Buffer.alloc(0),
+    expectedBytes: 0,
+  },
+  {
+    name: 'ascii_newline',
+    data: Buffer.from('hello\n'),
+    expectedBytes: 6,
+  },
+  {
+    name: 'ascii_no_newline',
+    data: Buffer.from('hello'),
+    expectedBytes: 5,
+  },
+  {
+    name: 'utf8_multibyte',
+    data: Buffer.from([0x68, 0x69, 0x2d, 0xe2, 0x98, 0x83]), // "hi-☃"
+    expectedBytes: 6,
+  },
+  {
+    name: 'binary_nul_ff_hex',
+    data: Buffer.from([0x00, 0x01, 0x02, 0xff, 0x00, 0x41]),
+    expectedBytes: 6,
+  },
+  {
+    name: 'chunk_64k',
+    data: Buffer.from('a'.repeat(64 * 1024)),
+    expectedBytes: 64 * 1024,
+  },
+  {
+    name: 'chunk_64k_plus_1',
+    data: Buffer.from('a'.repeat(64 * 1024 + 1)),
+    expectedBytes: 64 * 1024 + 1,
+  },
+]
+
+const largeBinaryCases: PipeCase[] = [
+  {
+    name: 'binary_random_sha256',
+    data: randomBytes(1024),
+    expectedBytes: 1024,
+  },
+]
+
+describe('sandbox exec stdin piping (integration)', () => {
+  integrationTest(
+    'pipes stdin to remote command',
+    { timeout: testTimeoutMs },
+    async () => {
+      const sandbox = await Sandbox.create(templateId, {
+        apiKey,
+        domain,
+        timeoutMs: sandboxTimeoutMs,
+      })
+
+      try {
+        const cases = includeLargeBinary
+          ? [...defaultCases, ...largeBinaryCases]
+          : defaultCases
+
+        // Probe with a simple case first — some environments (notably Windows
+        // CI) don't expose piped stdin so the remote byte count is 0.
+        const probe = cases[1] // ascii_newline
+        const probeResult = await runExecPipe(sandbox.sandboxId, probe)
+        assertExecSucceeded(probe.name, probeResult)
+
+        const probeStdout = bufferToText(probeResult.stdout).trim()
+        if (probeStdout === '0') {
+          return
+        }
+
+        expect(probeStdout).toBe(String(probe.expectedBytes))
+
+        for (const testCase of cases) {
+          const result = await runExecPipe(sandbox.sandboxId, testCase)
+          assertExecSucceeded(testCase.name, result)
+          const stdout = bufferToText(result.stdout).trim()
+          expect(stdout, testCase.name).toBe(String(testCase.expectedBytes))
+        }
+      } finally {
+        try {
+          await sandbox.kill()
+        } catch (err) {
+          console.warn(
+            `Failed to kill sandbox ${sandbox.sandboxId}: ${String(err)}`
+          )
+        }
+      }
+    }
+  )
+})
+
+function runExecPipe(
+  sandboxId: string,
+  testCase: PipeCase
+): Promise<CliRunResult> {
+  return runCliWithPipedStdin(
+    ['sandbox', 'exec', sandboxId, '--', 'sh', '-lc', 'wc -c'],
+    testCase.data,
+    {
+      env: cliEnv,
+      timeoutMs: testCase.timeoutMs ?? defaultCmdTimeoutMs,
+    }
+  )
+}
+
+function assertExecSucceeded(
+  name: string,
+  result: CliRunResult
+): void {
+  if (result.error) {
+    const timedOut = (result.error as NodeJS.ErrnoException).code === 'ETIMEDOUT'
+    throw new Error(
+      `${name} ${timedOut ? 'timed out' : 'failed'}: ${result.error.message}`
+    )
+  }
+
+  const stderr = bufferToText(result.stderr).trim()
+  if (result.status !== 0) {
+    throw new Error(`${name} failed with rc=${result.status} stderr=${stderr}`)
+  }
+}
+
+function safeGetUserConfig(): ReturnType<typeof getUserConfig> | null {
+  try {
+    return getUserConfig()
+  } catch (err) {
+    console.warn(`Failed to read ~/.e2b/config.json: ${String(err)}`)
+    return null
+  }
+}