tabwright 1.0.0 → 1.0.1

package/src/executor.unit.test.ts

@@ -1,168 +0,0 @@
-import { describe, it, expect } from 'vitest'
-import { shouldAutoReturn, wrapCode, isPlaywrightChannelOwner } from './executor.js'
-
-describe('shouldAutoReturn', () => {
-  it('returns true for simple expressions', () => {
-    expect(shouldAutoReturn('1 + 2')).toBe(true)
-    expect(shouldAutoReturn('page.title()')).toBe(true)
-    expect(shouldAutoReturn('await page.title()')).toBe(true)
-    expect(shouldAutoReturn('snapshot({ page })')).toBe(true)
-    expect(shouldAutoReturn('accessibilitySnapshot({ page })')).toBe(true)
-    expect(shouldAutoReturn('context.pages().map(p => p.url())')).toBe(true)
-  })
-
-  it('returns true for awaited expressions', () => {
-    expect(shouldAutoReturn('await Promise.resolve(42)')).toBe(true)
-    expect(shouldAutoReturn('await page.goto("https://example.com")')).toBe(true)
-  })
-
-  it('returns false for assignment expressions', () => {
-    expect(shouldAutoReturn('state.page = await context.newPage()')).toBe(false)
-    expect(shouldAutoReturn('const x = 1')).toBe(false)
-    expect(shouldAutoReturn('let y = 2')).toBe(false)
-    expect(shouldAutoReturn('x = 5')).toBe(false)
-  })
-
-  it('returns false for update expressions', () => {
-    expect(shouldAutoReturn('x++')).toBe(false)
-    expect(shouldAutoReturn('++x')).toBe(false)
-    expect(shouldAutoReturn('x--')).toBe(false)
-    expect(shouldAutoReturn('--x')).toBe(false)
-  })
-
-  it('returns false for delete expressions', () => {
-    expect(shouldAutoReturn('delete obj.prop')).toBe(false)
-  })
-
-  it('returns false for multiple statements', () => {
-    expect(shouldAutoReturn('const x = 1; x + 1')).toBe(false)
-    expect(shouldAutoReturn('await page.click("button"); await page.title()')).toBe(false)
-    expect(shouldAutoReturn('const a = 1\nconst b = 2')).toBe(false)
-  })
-
-  it('returns false for return statements', () => {
-    expect(shouldAutoReturn('return 5')).toBe(false)
-    expect(shouldAutoReturn('return await page.title()')).toBe(false)
-  })
-
-  it('returns false for non-expression statements', () => {
-    expect(shouldAutoReturn('if (true) { x }')).toBe(false)
-    expect(shouldAutoReturn('for (let i = 0; i < 10; i++) {}')).toBe(false)
-    expect(shouldAutoReturn('function foo() {}')).toBe(false)
-    expect(shouldAutoReturn('class Foo {}')).toBe(false)
-  })
-
-  it('returns false for invalid syntax', () => {
-    expect(shouldAutoReturn('const')).toBe(false)
-    expect(shouldAutoReturn('{')).toBe(false)
-    expect(shouldAutoReturn('await')).toBe(false)
-  })
-
-  it('handles edge cases', () => {
-    // Empty string
-    expect(shouldAutoReturn('')).toBe(false)
-    // Just whitespace
-    expect(shouldAutoReturn('   ')).toBe(false)
-    // Object literal (parsed as block)
-    expect(shouldAutoReturn('{ foo: 1 }')).toBe(false)
-    // Parenthesized object literal (expression)
-    expect(shouldAutoReturn('({ foo: 1 })')).toBe(true)
-    // Array literal
-    expect(shouldAutoReturn('[1, 2, 3]')).toBe(true)
-  })
-
-  it('handles sequence expressions with assignments', () => {
-    // Sequence with assignment should not auto-return
-    expect(shouldAutoReturn('(x = 1, x + 1)')).toBe(false)
-    // Sequence without assignment should auto-return
-    expect(shouldAutoReturn('(1, 2, 3)')).toBe(true)
-  })
-
-  it('handles comments in code', () => {
-    expect(shouldAutoReturn('// comment\npage.title()')).toBe(true)
-    expect(shouldAutoReturn('page.title() // comment')).toBe(true)
-    expect(shouldAutoReturn('/* block */ 42')).toBe(true)
-  })
-
-  it('handles template literals', () => {
-    expect(shouldAutoReturn('`hello`')).toBe(true)
-    expect(shouldAutoReturn('`hello ${name}`')).toBe(true)
-  })
-
-  it('returns true for expressions with trailing semicolons', () => {
-    expect(shouldAutoReturn('await page.title();')).toBe(true)
-    expect(shouldAutoReturn('page.title();')).toBe(true)
-    expect(shouldAutoReturn('snapshot({ page });')).toBe(true)
-    expect(shouldAutoReturn('await screenshotWithAccessibilityLabels({ page });')).toBe(true)
-    // trailing whitespace after semicolon
-    expect(shouldAutoReturn('await foo();   ')).toBe(true)
-    // comment after semicolon
-    expect(shouldAutoReturn('await foo(); // comment')).toBe(true)
-    // double semicolons are two statements, not auto-return
-    expect(shouldAutoReturn('await foo();;')).toBe(false)
-  })
-})
-
-describe('wrapCode', () => {
-  it('wraps single expressions with return await', () => {
-    expect(wrapCode('await page.title()')).toBe(
-      '(async () => { return await (await page.title()) })()',
-    )
-  })
-
-  it('strips trailing semicolons to avoid SyntaxError (#58)', () => {
-    // The bug: trailing semicolons inside return await (...;) produce invalid JS
-    const wrapped = wrapCode('await screenshotWithAccessibilityLabels({ page });')
-    expect(wrapped).toBe(
-      '(async () => { return await (await screenshotWithAccessibilityLabels({ page })) })()',
-    )
-    // Verify it's valid JS by parsing it
-    expect(() => new Function(wrapped)).not.toThrow()
-  })
-
-  it('produces valid JS for trailing semicolons on simple calls', () => {
-    const wrapped = wrapCode('page.title();')
-    expect(wrapped).toBe('(async () => { return await (page.title()) })()')
-    expect(() => new Function(wrapped)).not.toThrow()
-  })
-
-  it('wraps multi-statement code without return', () => {
-    const wrapped = wrapCode('const x = 1; console.log(x);')
-    expect(wrapped).toBe('(async () => { const x = 1; console.log(x); })()')
-  })
-
-  it('wraps assignment expressions without return', () => {
-    expect(wrapCode('x = 5')).toBe('(async () => { x = 5 })()')
-  })
-})
-
-describe('isPlaywrightChannelOwner', () => {
-  it('detects ChannelOwner-shaped objects', () => {
-    const fakeResponse = {
-      _type: 'Response',
-      _guid: 'response@abc123',
-      _connection: { _platform: { env: {} } },
-    }
-    expect(isPlaywrightChannelOwner(fakeResponse)).toBe(true)
-  })
-
-  it('rejects plain objects and primitives', () => {
-    expect(isPlaywrightChannelOwner(null)).toBe(false)
-    expect(isPlaywrightChannelOwner(undefined)).toBe(false)
-    expect(isPlaywrightChannelOwner('string')).toBe(false)
-    expect(isPlaywrightChannelOwner(42)).toBe(false)
-    expect(isPlaywrightChannelOwner({})).toBe(false)
-    expect(isPlaywrightChannelOwner([])).toBe(false)
-    expect(isPlaywrightChannelOwner({ url: 'https://example.com', status: 200 })).toBe(false)
-  })
-
-  it('rejects partial matches missing required fields', () => {
-    expect(isPlaywrightChannelOwner({ _type: 'Response' })).toBe(false)
-    expect(isPlaywrightChannelOwner({ _guid: 'abc' })).toBe(false)
-    expect(isPlaywrightChannelOwner({ _type: 'x', _guid: 'y' })).toBe(false)
-    // Missing _connection
-    expect(isPlaywrightChannelOwner({ _type: 'x', _guid: 'y' })).toBe(false)
-    // _type must be a string
-    expect(isPlaywrightChannelOwner({ _type: 123, _guid: 'y', _connection: {} })).toBe(false)
-  })
-})