small fixes

Author: dobrac

packages/cli/src/commands/template/migrate.ts

@@ -100,12 +100,11 @@ export const migrateCommand = new commander.Command('migrate')
     `specify target language: ${Object.values(Language).join(', ')}`,
     (value) => {
       if (!Object.values(Language).includes(value as Language)) {
-        console.error(
+        throw new commander.InvalidArgumentError(
           `Invalid language. Must be one of: ${Object.values(Language).join(
             ', '
           )}`
         )
-        process.exit(1)
       }
       return value as Language
     }

packages/cli/src/templates/readme.hbs

@@ -47,7 +47,7 @@ cd {{templateDir}} && npx tsx {{buildProdFile}}
 # For development
 cd {{templateDir}} && python {{buildDevFile}}
 
-# For production  
+# For production
 cd {{templateDir}} && python {{buildProdFile}}
 ```
 {{/if}}

packages/cli/tests/commands/template/init.test.ts

@@ -3,6 +3,7 @@ import { existsSync } from 'fs'
 import * as fs from 'fs/promises'
 import * as path from 'path'
 import { afterEach, beforeEach, describe, expect, test } from 'vitest'
+import { Language } from '../../../src/commands/template/generators'
 
 describe('Template Init', () => {
   let testDir: string
@@ -20,10 +21,8 @@ describe('Template Init', () => {
     }
   })
 
-  const languages = ['typescript', 'python-sync', 'python-async'] as const
-
   describe('CLI Options', () => {
-    languages.forEach((language) => {
+    Object.values(Language).forEach((language) => {
       test(`should generate files with --name and --language ${language}`, async () => {
         const templateName = 'my-test-template'
 
@@ -199,7 +198,7 @@ describe('Template Init', () => {
       expect(templateContent).toContain("import { Template } from 'e2b'")
       expect(templateContent).toContain('export const template = Template()')
       expect(templateContent).toContain('fromImage')
-      expect(templateContent).toContain('ubuntu:22.04')
+      expect(templateContent).toContain('e2bdev/base')
     })
 
     test('should generate correct Python template content', async () => {
@@ -219,7 +218,7 @@ describe('Template Init', () => {
       expect(templateContent).toContain('template = (')
       expect(templateContent).toContain('Template()')
       expect(templateContent).toContain('from_image')
-      expect(templateContent).toContain('ubuntu:22.04')
+      expect(templateContent).toContain('e2bdev/base')
     })
 
     test('should generate correct async Python template content', async () => {