wehub-resource-sync
89 lines
2.6 KiB
TypeScript
89 lines
2.6 KiB
TypeScript
/**
|
|
* @license
|
|
* Copyright 2025 AionUi (aionui.com)
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*
|
|
* Unit tests for the upload state store, focused on the abort wiring added
|
|
* for ELECTRON-1K2 (uploads cannot be cancelled / are not bound to the
|
|
* conversation lifecycle).
|
|
*/
|
|
|
|
import { describe, it, expect, vi } from 'vitest';
|
|
|
|
import { abortUpload, abortUploads, trackUpload } from '@/renderer/hooks/file/useUploadState';
|
|
|
|
describe('useUploadState abort wiring', () => {
|
|
it('invokes onAbort exactly once when abortUpload is called', () => {
|
|
const onAbort = vi.fn();
|
|
const tracker = trackUpload(100, { source: 'sendbox', name: 'a.txt', onAbort });
|
|
|
|
abortUpload(tracker.id);
|
|
abortUpload(tracker.id); // second call must be a no-op
|
|
|
|
expect(onAbort).toHaveBeenCalledTimes(1);
|
|
|
|
// Cleanup so the store is empty between tests.
|
|
tracker.finish();
|
|
});
|
|
|
|
it('tracker.abort() reaches the registered onAbort handler', () => {
|
|
const onAbort = vi.fn();
|
|
const tracker = trackUpload(100, { source: 'sendbox', name: 'b.txt', onAbort });
|
|
|
|
tracker.abort();
|
|
|
|
expect(onAbort).toHaveBeenCalledTimes(1);
|
|
tracker.finish();
|
|
});
|
|
|
|
it('abortUploads with exceptConversationId only aborts uploads bound to other conversations', () => {
|
|
const onAbortKept = vi.fn();
|
|
const onAbortDropped = vi.fn();
|
|
const onAbortUnbound = vi.fn();
|
|
|
|
const kept = trackUpload(10, {
|
|
source: 'sendbox',
|
|
name: 'keep.txt',
|
|
conversationId: 'cur',
|
|
onAbort: onAbortKept,
|
|
});
|
|
const dropped = trackUpload(10, {
|
|
source: 'sendbox',
|
|
name: 'drop.txt',
|
|
conversationId: 'old',
|
|
onAbort: onAbortDropped,
|
|
});
|
|
const unbound = trackUpload(10, {
|
|
source: 'sendbox',
|
|
name: 'unbound.txt',
|
|
onAbort: onAbortUnbound,
|
|
});
|
|
|
|
abortUploads({ source: 'sendbox', exceptConversationId: 'cur' });
|
|
|
|
expect(onAbortKept).not.toHaveBeenCalled();
|
|
expect(onAbortDropped).toHaveBeenCalledTimes(1);
|
|
expect(onAbortUnbound).toHaveBeenCalledTimes(1); // unbound != 'cur', so it is aborted
|
|
|
|
kept.finish();
|
|
dropped.finish();
|
|
unbound.finish();
|
|
});
|
|
|
|
it('abortUploads scoped by source only affects matching uploads', () => {
|
|
const onAbortSendbox = vi.fn();
|
|
const onAbortWorkspace = vi.fn();
|
|
|
|
const sendbox = trackUpload(10, { source: 'sendbox', onAbort: onAbortSendbox });
|
|
const workspace = trackUpload(10, { source: 'workspace', onAbort: onAbortWorkspace });
|
|
|
|
abortUploads({ source: 'sendbox' });
|
|
|
|
expect(onAbortSendbox).toHaveBeenCalledTimes(1);
|
|
expect(onAbortWorkspace).not.toHaveBeenCalled();
|
|
|
|
sendbox.finish();
|
|
workspace.finish();
|
|
});
|
|
});
|