fix issue

tests/modal-helper.test.ts

@@ -0,0 +1,76 @@
+import { expect, test } from 'bun:test';
+import type MediaDbPlugin from 'packages/obsidian/src/main';
+import { ModalHelper, type ModalLifecycle, type ModalSession } from 'packages/obsidian/src/utils/ModalHelper';
+import type { MDBError } from 'packages/obsidian/src/utils/MDBError';
+import { MDBErrorKind } from 'packages/obsidian/src/utils/MDBError';
+import { OutcomeStatus } from 'packages/obsidian/src/utils/result';
+
+class FakeModal implements ModalLifecycle {
+	closeCount = 0;
+
+	open(): void {
+		// Test modal does not need to render.
+	}
+
+	close(): void {
+		this.closeCount += 1;
+	}
+}
+
+function createModalSession(cancelled: () => boolean): ModalSession<undefined, FakeModal> {
+	const modal = new FakeModal();
+
+	return {
+		modal,
+		modalResult: { status: OutcomeStatus.Ok, data: undefined },
+		close: (): void => modal.close(),
+		isCancelled: cancelled,
+	};
+}
+
+function createHelper(): ModalHelper {
+	return new ModalHelper({} as MediaDbPlugin);
+}
+
+test('runModalTask closes the modal after active task success', async () => {
+	const helper = createHelper();
+	const session = createModalSession(() => false);
+
+	const outcome = await helper.runModalTask(session, async () => 'done');
+
+	expect(outcome).toEqual({ status: OutcomeStatus.Ok, data: 'done' });
+	expect(session.modal.closeCount).toBe(1);
+});
+
+test('runModalTask ignores late success after cancellation', async () => {
+	const helper = createHelper();
+	let cancelled = false;
+	const session = createModalSession(() => cancelled);
+
+	const outcome = await helper.runModalTask(session, async () => {
+		cancelled = true;
+		return 'late result';
+	});
+
+	expect(outcome).toEqual({ status: OutcomeStatus.Cancelled });
+	expect(session.modal.closeCount).toBe(0);
+});
+
+test('runModalTask ignores late errors after cancellation', async () => {
+	const helper = createHelper();
+	let cancelled = false;
+	const session = createModalSession(() => cancelled);
+	const fallback: MDBError = { kind: MDBErrorKind.Modal, message: 'Task failed' };
+
+	const outcome = await helper.runModalTask(
+		session,
+		async () => {
+			cancelled = true;
+			throw new Error('Late failure');
+		},
+		fallback,
+	);
+
+	expect(outcome).toEqual({ status: OutcomeStatus.Cancelled });
+	expect(session.modal.closeCount).toBe(0);
+});