snapshot of https://github.com/toeverything/AFFiNE.git (canary, 2026-09-16; blobs > 131072B omitted)

blocksuite/integration-test/src/__tests__/edgeless/tools.spec.ts

@@ -0,0 +1,96 @@
+import type { EdgelessRootBlockComponent } from '@blocksuite/affine/blocks/root';
+import {
+  DefaultTool,
+  type SurfaceBlockComponent,
+} from '@blocksuite/affine/blocks/surface';
+import { beforeEach, describe, expect, test } from 'vitest';
+
+import { click, drag, wait } from '../utils/common.js';
+import { addNote, getDocRootBlock, getSurface } from '../utils/edgeless.js';
+import { setupEditor } from '../utils/setup.js';
+
+describe('default tool', () => {
+  let surface!: SurfaceBlockComponent;
+  let edgeless!: EdgelessRootBlockComponent;
+  let service!: EdgelessRootBlockComponent['service'];
+
+  beforeEach(async () => {
+    const cleanup = await setupEditor('edgeless');
+
+    edgeless = getDocRootBlock(doc, editor, 'edgeless');
+    surface = getSurface(window.doc, window.editor);
+    service = edgeless.service;
+
+    edgeless.gfx.tool.setTool(DefaultTool);
+
+    return cleanup;
+  });
+
+  test('element click selection', async () => {
+    const id = service.crud.addElement('shape', {
+      shapeType: 'rect',
+      xywh: '[0,0,100,100]',
+      fillColor: 'red',
+    });
+
+    await wait();
+
+    service.viewport.setViewport(1, [
+      service.viewport.width / 2,
+      service.viewport.height / 2,
+    ]);
+
+    click(edgeless.host, { x: 0, y: 50 });
+
+    expect(edgeless.service.selection.surfaceSelections[0].elements).toEqual([
+      id,
+    ]);
+  });
+
+  test('element drag moving', async () => {
+    const id = edgeless.service.crud.addElement('shape', {
+      shapeType: 'rect',
+      xywh: '[0,0,100,100]',
+      fillColor: 'red',
+    });
+    await wait();
+
+    edgeless.service.viewport.setViewport(1, [
+      edgeless.service.viewport.width / 2,
+      edgeless.service.viewport.height / 2,
+    ]);
+    await wait();
+
+    click(edgeless.host, { x: 0, y: 50 });
+    drag(edgeless.host, { x: 0, y: 50 }, { x: 0, y: 150 });
+    await wait();
+
+    const element = service.crud.getElementById(id!)!;
+    expect(element.xywh).toEqual(`[0,100,100,100]`);
+  });
+
+  test('block drag moving', async () => {
+    const noteId = addNote(doc);
+
+    await wait();
+
+    edgeless.service.viewport.setViewport(1, [
+      surface.renderer.viewport.width / 2,
+      surface.renderer.viewport.height / 2,
+    ]);
+    await wait();
+
+    click(edgeless.host, { x: 50, y: 50 });
+    expect(edgeless.service.selection.surfaceSelections[0].elements).toEqual([
+      noteId,
+    ]);
+    drag(edgeless.host, { x: 50, y: 50 }, { x: 150, y: 150 });
+    await wait();
+
+    const element = service.crud.getElementById(noteId)!;
+    const [x, y] = JSON.parse(element.xywh);
+
+    expect(x).toEqual(100);
+    expect(y).toEqual(100);
+  });
+});