webui/src/utils/workers/scriptVerification.spec.ts - master - githubmirror/traefik

webui/src/utils/workers/scriptVerification.spec.ts
117 строк3 KB
Gina A.
Restore remote Upgrade to Hub button web component
12 ноя 2025, 14:16
12 ноя 2025, 14:16a01c73d
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'

import verifySignature from './scriptVerification'

const SCRIPT_PATH = 'https://example.com/script.js'
const MOCK_PUBLIC_KEY = 'MCowBQYDK2VwAyEAWH71OHphISjNK3mizCR/BawiDxc6IXT1vFHpBcxSIA0='
class MockWorker {
  onmessage: ((event: MessageEvent) => void) | null = null
  onerror: ((error: ErrorEvent) => void) | null = null
  postMessage = vi.fn()
  terminate = vi.fn()

  simulateMessage(data: unknown) {
    if (this.onmessage) {
      this.onmessage(new MessageEvent('message', { data }))
    }
  }

  simulateError(error: Error) {
    if (this.onerror) {
      this.onerror(new ErrorEvent('error', { error, message: error.message }))
    }
  }
}

describe('verifySignature', () => {
  let mockWorkerInstance: MockWorker
  let originalWorker: typeof Worker

  beforeEach(() => {
    vi.clearAllMocks()

    originalWorker = globalThis.Worker

    mockWorkerInstance = new MockWorker()

    globalThis.Worker = class extends EventTarget {
      constructor() {
        super()
        return mockWorkerInstance as any
      }
    } as any
  })

  afterEach(() => {
    globalThis.Worker = originalWorker
    vi.restoreAllMocks()
  })

  it('should return true when verification succeeds', async () => {
    const promise = verifySignature(SCRIPT_PATH, `${SCRIPT_PATH}.sig`, MOCK_PUBLIC_KEY)

    await new Promise((resolve) => setTimeout(resolve, 0))

    expect(mockWorkerInstance.postMessage).toHaveBeenCalledWith(
      expect.objectContaining({
        scriptUrl: SCRIPT_PATH,
        signatureUrl: `${SCRIPT_PATH}.sig`,
        requestId: expect.any(String),
      }),
    )

    const mockScriptContent = new ArrayBuffer(100)
    mockWorkerInstance.simulateMessage({
      success: true,
      verified: true,
      error: null,
      scriptContent: mockScriptContent,
    })

    const result = await promise

    expect(result).toEqual({ verified: true, scriptContent: mockScriptContent })
    expect(mockWorkerInstance.terminate).toHaveBeenCalled()
  })

  it('should return false when verification fails', async () => {
    const consoleErrorSpy = vi.spyOn(console, 'error').mockImplementation(() => {})

    const promise = verifySignature(SCRIPT_PATH, `${SCRIPT_PATH}.sig`, MOCK_PUBLIC_KEY)

    await new Promise((resolve) => setTimeout(resolve, 0))

    mockWorkerInstance.simulateMessage({
      success: false,
      verified: false,
      error: 'Signature verification failed',
    })

    const result = await promise

    expect(result).toEqual({ verified: false })
    expect(mockWorkerInstance.terminate).toHaveBeenCalled()
    expect(consoleErrorSpy).toHaveBeenCalledWith('Worker verification failed:', 'Signature verification failed')

    consoleErrorSpy.mockRestore()
  })

  it('should return false when worker throws an error', async () => {
    const consoleErrorSpy = vi.spyOn(console, 'error').mockImplementation(() => {})

    const promise = verifySignature(SCRIPT_PATH, `${SCRIPT_PATH}.sig`, MOCK_PUBLIC_KEY)

    await new Promise((resolve) => setTimeout(resolve, 0))

    const error = new Error('Worker crashed')
    mockWorkerInstance.simulateError(error)

    const result = await promise

    expect(result).toEqual({ verified: false })
    expect(mockWorkerInstance.terminate).toHaveBeenCalled()
    expect(consoleErrorSpy).toHaveBeenCalledWith('Worker error:', expect.any(ErrorEvent))

    consoleErrorSpy.mockRestore()
  })
})