snapshot of https://github.com/dream-num/univer.git (dev, 2026-09-16; blobs > 131072B omitted)

packages/drawing/src/commands/operations/__tests__/set-drawing-selected.operation.spec.ts

@@ -0,0 +1,89 @@
+/**
+ * Copyright 2023-present DreamNum Co., Ltd.
+ *
+ * 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 type { IDrawingParam, IDrawingSearch } from '@univerjs/core';
+import { DrawingTypeEnum, ICommandService, Univer } from '@univerjs/core';
+import { afterEach, beforeEach, describe, expect, it } from 'vitest';
+import { DrawingManagerService } from '../../../services/drawing-manager-impl.service';
+import { IDrawingManagerService } from '../../../services/drawing-manager.service';
+import { SetDrawingSelectedOperation } from '../set-drawing-selected.operation';
+
+const unitId = 'drawing-unit';
+const subUnitId = 'drawing-subunit';
+
+function createDrawing(drawingId: string): IDrawingParam {
+    return {
+        unitId,
+        subUnitId,
+        drawingId,
+        drawingType: DrawingTypeEnum.DRAWING_IMAGE,
+    };
+}
+
+function createSearch(drawingId: string): IDrawingSearch {
+    return {
+        unitId,
+        subUnitId,
+        drawingId,
+    };
+}
+
+describe('SetDrawingSelectedOperation', () => {
+    let univer: Univer;
+    let commandService: ICommandService;
+    let drawingManagerService: IDrawingManagerService;
+
+    beforeEach(() => {
+        univer = new Univer();
+        const injector = univer.__getInjector();
+        injector.add([IDrawingManagerService, { useClass: DrawingManagerService }]);
+
+        commandService = injector.get(ICommandService);
+        commandService.registerCommand(SetDrawingSelectedOperation);
+        drawingManagerService = injector.get(IDrawingManagerService);
+        drawingManagerService.registerDrawingData(unitId, {
+            [subUnitId]: {
+                data: {
+                    image1: createDrawing('image1'),
+                    image2: createDrawing('image2'),
+                },
+                order: ['image1', 'image2'],
+            },
+        });
+    });
+
+    afterEach(() => {
+        univer.dispose();
+    });
+
+    it('selects only drawings that exist in the drawing manager', async () => {
+        const result = await commandService.executeCommand(SetDrawingSelectedOperation.id, [
+            createSearch('image2'),
+            createSearch('missing'),
+        ]);
+
+        expect(result).toBe(true);
+        expect(drawingManagerService.getFocusDrawings()).toEqual([createDrawing('image2')]);
+    });
+
+    it('keeps the current drawing focus when selection params are missing', async () => {
+        await commandService.executeCommand(SetDrawingSelectedOperation.id, [createSearch('image1')]);
+        const result = await commandService.executeCommand(SetDrawingSelectedOperation.id);
+
+        expect(result).toBe(false);
+        expect(drawingManagerService.getFocusDrawings()).toEqual([createDrawing('image1')]);
+    });
+});