100 lines
2.8 KiB
TypeScript
100 lines
2.8 KiB
TypeScript
/*
|
|
* 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 { uploadFileV2 } from '../src/upload-file-v2';
|
|
vi.mock('@coze-arch/bot-api', () => ({
|
|
DeveloperApi: {
|
|
GetUploadAuthToken: vi.fn(() =>
|
|
Promise.resolve({ data: { service_id: '', upload_host: '' } }),
|
|
),
|
|
},
|
|
}));
|
|
vi.mock('@coze-studio/uploader-adapter', () => {
|
|
class MockUploader {
|
|
start = () => 0;
|
|
addFile = () => '12312341';
|
|
test: 'test';
|
|
on(event: string, cb: (data: any) => void) {
|
|
if (event === 'complete') {
|
|
cb({ uploadResult: { Uri: 'test_url' } });
|
|
} else if (event === 'error') {
|
|
cb({ extra: 'error' });
|
|
} else if (event === 'progress') {
|
|
cb(50);
|
|
}
|
|
}
|
|
}
|
|
return {
|
|
getUploader: vi.fn((props: any, isOverSea?: boolean) => new MockUploader()),
|
|
};
|
|
});
|
|
|
|
describe('upload-file', () => {
|
|
afterEach(() => {
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
test('upLoadFile should resolve Url of result if upload success', () =>
|
|
new Promise((resolve, reject) => {
|
|
global.IS_OVERSEA = true;
|
|
uploadFileV2({
|
|
fileItemList: [{ file: new File([], 'test_file'), fileType: 'image' }],
|
|
userId: '123',
|
|
timeout: undefined,
|
|
signal: new AbortSignal(),
|
|
onSuccess: event => {
|
|
try {
|
|
expect(event.uploadResult.Uri).equal('test_url');
|
|
resolve('ok');
|
|
} catch (e) {
|
|
reject(e);
|
|
}
|
|
},
|
|
onUploadAllSuccess(event) {
|
|
try {
|
|
expect(event[0].uploadResult.Uri).equal('test_url');
|
|
resolve('ok');
|
|
} catch (e) {
|
|
reject(e);
|
|
}
|
|
},
|
|
});
|
|
global.IS_OVERSEA = false;
|
|
uploadFileV2({
|
|
fileItemList: [{ file: new File([], 'test_file'), fileType: 'image' }],
|
|
userId: '123',
|
|
timeout: undefined,
|
|
signal: new AbortSignal(),
|
|
onSuccess: event => {
|
|
try {
|
|
expect(event.uploadResult.Uri).equal('test_url');
|
|
resolve('ok');
|
|
} catch (e) {
|
|
reject(e);
|
|
}
|
|
},
|
|
onUploadAllSuccess(event) {
|
|
try {
|
|
expect(event[0].uploadResult.Uri).equal('test_url');
|
|
resolve('ok');
|
|
} catch (e) {
|
|
reject(e);
|
|
}
|
|
},
|
|
});
|
|
}));
|
|
});
|