Files
2026-07-13 12:47:58 +08:00

75 lines
2.8 KiB
TypeScript

import * as fs from 'fs/promises'
import * as path from 'path'
import { execSync } from 'child_process'
import { afterEach, beforeEach, describe, expect, test } from 'vitest'
describe('Template Publish --config', () => {
let testDir: string
beforeEach(async () => {
testDir = await fs.mkdtemp('e2b-publish-config-test-')
const defaultConfig = `template_id = "default-template-id"
dockerfile = "e2b.Dockerfile"`
await fs.writeFile(path.join(testDir, 'e2b.toml'), defaultConfig)
const customConfig = `template_id = "custom-template-id"
dockerfile = "e2b.Dockerfile"`
await fs.writeFile(path.join(testDir, 'custom.toml'), customConfig)
await fs.writeFile(path.join(testDir, 'e2b.Dockerfile'), 'FROM alpine:3.18')
})
afterEach(async () => {
if (testDir) {
await fs.rm(testDir, { recursive: true, force: true })
}
})
test('uses the config file passed via --config even when e2b.toml exists', async () => {
const cliPath = path.join(process.cwd(), 'dist', 'index.js')
let output = ''
try {
// This will likely fail on the network call, but we only need stdout up to that point
output = execSync(
`node "${cliPath}" template publish --yes --path "${testDir}" --config custom.toml 2>&1`,
{ encoding: 'utf-8', stdio: 'pipe', timeout: 10_000 }
)
} catch (err: any) {
// Capture partial output from the failed process (expected due to no network)
output = (err?.stdout ?? '') + (err?.stderr ?? '')
}
// It should log the sandbox template selection with the custom template ID and path
expect(output).toContain('Sandbox templates to publish')
expect(output).toContain('custom-template-id')
expect(output).toContain('custom.toml')
// And should not show the default template id when custom is provided
expect(output).not.toContain('default-template-id')
})
test('uses the default e2b.toml when --config is not provided', async () => {
const cliPath = path.join(process.cwd(), 'dist', 'index.js')
let output = ''
try {
output = execSync(
`node "${cliPath}" template publish --yes --path "${testDir}" 2>&1`,
{ encoding: 'utf-8', stdio: 'pipe', timeout: 10_000 }
)
} catch (err: any) {
output = (err?.stdout ?? '') + (err?.stderr ?? '')
}
// Should reference the default config and template id
expect(output).toContain('Sandbox templates to publish')
expect(output).toContain('default-template-id')
expect(output).toContain('e2b.toml')
// Should not mention the custom config/template id
expect(output).not.toContain('custom-template-id')
})
})