/* * Copyright 2025 coze-dev Authors * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ import { describe, it, expect, vi, beforeEach } from 'vitest'; import { RemoteWebWorker, register } from '../src/index'; // 获取模拟函数 const mockCreateObjectURL = vi.mocked(URL.createObjectURL); describe('RemoteWebWorker', () => { beforeEach(() => { // 清除模拟调用记录 vi.clearAllMocks(); }); describe('constructor', () => { it('应该直接使用本地 URL', () => { const localUrl = 'worker.js'; const options = { type: 'module' }; new RemoteWebWorker(localUrl, options); expect(mockCreateObjectURL).not.toHaveBeenCalled(); }); it('应该直接使用同源的远程 URL', () => { const sameOriginUrl = 'https://example.com/worker.js'; const options = { type: 'module' }; new RemoteWebWorker(sameOriginUrl, options); expect(mockCreateObjectURL).not.toHaveBeenCalled(); }); it('应该直接使用 blob URL', () => { const blobUrl = 'blob:https://example.com/worker.js'; const options = { type: 'module' }; new RemoteWebWorker(blobUrl, options); expect(mockCreateObjectURL).not.toHaveBeenCalled(); }); it('应该为跨域 URL 创建 Blob 并使用 URL.createObjectURL', () => { const crossOriginUrl = 'https://other-domain.com/worker.js'; const options = { type: 'module' }; new RemoteWebWorker(crossOriginUrl, options); expect(mockCreateObjectURL).toHaveBeenCalledTimes(1); // 验证创建的 Blob 内容 const blobArg = mockCreateObjectURL.mock.calls[0][0]; expect(blobArg).toBeInstanceOf(Blob); // 由于 Blob 的内容无法直接访问,我们只能验证它被创建了 }); it('应该处理非字符串 URL', () => { const nonStringUrl = { toString: () => 'https://other-domain.com/worker.js', }; const options = { type: 'module' }; new RemoteWebWorker(nonStringUrl as any, options); expect(mockCreateObjectURL).toHaveBeenCalledTimes(1); }); }); }); describe('register', () => { it('应该将全局 Worker 替换为 RemoteWebWorker', () => { // 创建一个模拟的全局对象 const mockGlobal = { worker() { /* 空函数 */ }, }; // 将 worker 属性重命名为 Worker,以便测试 register 函数 Object.defineProperty(mockGlobal, 'Worker', { get() { return this.worker; }, set(value) { this.worker = value; }, enumerable: true, configurable: true, }); register(mockGlobal as any); expect(mockGlobal.worker).toBe(RemoteWebWorker); }); it('当全局对象未定义时不应该抛出错误', () => { expect(() => register(undefined as any)).not.toThrow(); }); });