tldraw 3.16.0-internal.51e99e128bd4 → 3.16.0-internal.a478398270c6

package/src/test/selection-omnibus.test.ts

@@ -7,7 +7,6 @@ import {
 	tlenv,
 	toRichText,
 } from '@tldraw/editor'
-import { vi } from 'vitest'
 import { TestEditor } from './TestEditor'
 
 let editor: TestEditor
@@ -1653,7 +1652,7 @@ describe('shift brushes to add to the selection', () => {
 		editor.keyUp('Shift')
 		// there's a timer here—we should keep the shift mode until the timer expires
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box2, ids.box1])
-		vi.advanceTimersByTime(500)
+		jest.advanceTimersByTime(500)
 		// once the timer expires, we should be back in regular mode
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box1])
 		editor.keyDown('Shift')
@@ -1723,7 +1722,7 @@ describe('scribble brushes to add to the selection', () => {
 		editor.pointerMove(50, 50)
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box1, ids.box2])
 		editor.keyUp('Shift')
-		vi.advanceTimersByTime(500)
+		jest.advanceTimersByTime(500)
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box1])
 		editor.keyDown('Shift')
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box1, ids.box2])
@@ -1740,7 +1739,7 @@ describe('scribble brushes to add to the selection', () => {
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box4])
 		editor.keyUp('Alt') // scribble
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box4]) // still in timer
-		vi.advanceTimersByTime(1000) // let timer expire
+		jest.advanceTimersByTime(1000) // let timer expire
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box3, ids.box4]) // brushed!
 		editor.keyDown('Alt') // scribble
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box4]) // back to brushed only
@@ -1825,7 +1824,7 @@ describe('When brushing close to the edges of the screen', () => {
 		editor.pointerMove(300, 300)
 		editor.pointerDown()
 		editor.pointerMove(0, 0)
-		vi.advanceTimersByTime(100)
+		jest.advanceTimersByTime(100)
 		editor.pointerUp()
 		const camera2 = editor.getCamera()
 		expect(camera2.x).toBeGreaterThan(camera1.x) // for some reason > is left
@@ -1838,7 +1837,7 @@ describe('When brushing close to the edges of the screen', () => {
 		editor.pointerMove(300, 300)
 		editor.pointerDown()
 		editor.pointerMove(100, 100)
-		vi.advanceTimersByTime(100)
+		jest.advanceTimersByTime(100)
 		editor.pointerUp()
 		const camera2 = editor.getCamera()
 		// should NOT have moved the camera by edge scrolling
@@ -1852,7 +1851,7 @@ describe('When brushing close to the edges of the screen', () => {
 		editor.pointerMove(300, 300)
 		editor.pointerDown()
 		editor.pointerMove(100, 100)
-		vi.advanceTimersByTime(100)
+		jest.advanceTimersByTime(100)
 		editor.pointerUp()
 		const camera4 = editor.getCamera()
 		// should NOT have moved the camera by edge scrolling because the edge is now "inset"
@@ -1861,7 +1860,7 @@ describe('When brushing close to the edges of the screen', () => {
 
 		editor.pointerDown()
 		editor.pointerMove(90, 90) // off the edge of the component
-		vi.advanceTimersByTime(100)
+		jest.advanceTimersByTime(100)
 		const camera5 = editor.getCamera()
 		// should have moved the camera by edge scrolling off the component edge
 		expect(camera5.x).toBeGreaterThan(camera4.x)
@@ -1883,7 +1882,7 @@ describe('When brushing close to the edges of the screen', () => {
 		editor.pointerMove(0, 0)
 		// still only box 1...
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box1])
-		vi.advanceTimersByTime(100)
+		jest.advanceTimersByTime(100)
 		// ...but now viewport will have moved to select box2 as well
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box1, ids.box2])
 		editor.pointerUp()
@@ -1903,7 +1902,7 @@ describe('When brushing close to the edges of the screen', () => {
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box1])
 		editor.pointerMove(0, 0)
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box1])
-		vi.advanceTimersByTime(100)
+		jest.advanceTimersByTime(100)
 		expect(editor.getSelectedShapeIds()).toEqual([ids.box1])
 		editor.pointerUp()
 	})
@@ -2200,7 +2199,7 @@ describe('long press', () => {
 	it('works correctly with screenbounds offset', () => {
 		editor.updateViewportScreenBounds(new Box(100, 100, 800, 600))
 		editor.pointerDown(201, 202)
-		vi.advanceTimersByTime(1000)
+		jest.advanceTimersByTime(1000)
 		// without the fix added in this PR, it would have been 1, 2
 		expect(editor.inputs.currentScreenPoint).toMatchObject({ x: 101, y: 102 })
 	})

package/src/test/shapeutils.test.ts

@@ -1,5 +1,4 @@
-import { createShapeId, TLFrameShape, TLGeoShape, TLLineShape } from '@tldraw/editor'
-import { vi } from 'vitest'
+import { createShapeId, TLFrameShape, TLGeoShape } from '@tldraw/editor'
 import { TestEditor } from './TestEditor'
 
 let editor: TestEditor
@@ -12,8 +11,6 @@ const ids = {
 	frame1: createShapeId('frame1'),
 	box1: createShapeId('box1'),
 	box2: createShapeId('box2'),
-	line1: createShapeId('line1'),
-	page1: createShapeId('page1'),
 }
 
 beforeEach(() => {
@@ -59,93 +56,34 @@ describe('When interacting with a shape...', () => {
 		// Set start / change / end events on only the geo shape
 		const util = editor.getShapeUtil<TLFrameShape>('frame')
 
-		const calls: string[] = []
+		const fnStart = jest.fn()
+		util.onRotateStart = fnStart
 
-		util.onRotateStart = () => {
-			calls.push('start')
-		}
+		const fnChange = jest.fn()
+		util.onRotate = fnChange
 
-		util.onRotate = () => {
-			calls.push('change')
-		}
-
-		util.onRotateEnd = () => {
-			calls.push('end')
-		}
+		const fnEnd = jest.fn()
+		util.onRotateEnd = fnEnd
 
 		editor.selectAll()
 		expect(editor.getSelectedShapeIds()).toMatchObject([ids.frame1, ids.box1])
 
-		editor.pointerDown(300, 300, {
-			target: 'selection',
-			handle: 'bottom_right_rotate',
-		})
-
-		// Should not have called any callbacks yet
-		expect(calls).toEqual([])
-
-		editor.pointerMove(200, 200)
-
-		// Should have called start once and change at least once now
-		expect(calls).toEqual(['start', 'change'])
-
-		editor.pointerMove(200, 210)
-
-		// Should have called start once and change multiple times
-		expect(calls).toEqual(['start', 'change', 'change'])
-
-		editor.pointerUp(200, 210)
-
-		// Should have called end once now
-		expect(calls).toEqual(['start', 'change', 'change', 'change', 'end'])
-	})
-
-	it('fires rotate cancel events', () => {
-		const util = editor.getShapeUtil<TLFrameShape>('frame')
-
-		const calls: string[] = []
-
-		util.onRotateStart = () => {
-			calls.push('start')
-		}
-
-		util.onRotate = () => {
-			calls.push('change')
-		}
-
-		util.onRotateEnd = () => {
-			calls.push('end')
-		}
-
-		util.onRotateCancel = () => {
-			calls.push('cancel')
-		}
-
-		editor.selectAll()
-		expect(editor.getSelectedShapeIds()).toMatchObject([ids.frame1, ids.box1])
-
-		editor.pointerDown(300, 300, {
-			target: 'selection',
-			handle: 'bottom_right_rotate',
-		})
-
-		// Should not have called any callbacks yet
-		expect(calls).toEqual([])
-
-		editor.pointerMove(200, 200)
-
-		// Should have called start once and change at least once now
-		expect(calls).toEqual(['start', 'change'])
+		editor
+			.pointerDown(300, 300, {
+				target: 'selection',
+				handle: 'bottom_right_rotate',
+			})
+			.pointerMove(200, 200)
+			.pointerUp(200, 200)
 
-		editor.pointerMove(200, 210)
-
-		// Should have called start once and change multiple times
-		expect(calls).toEqual(['start', 'change', 'change'])
+		// Once on start (for frame only)
+		expect(fnStart).toHaveBeenCalledTimes(1)
 
-		editor.cancel()
+		// Once on start, once during the move
+		expect(fnChange).toHaveBeenCalledTimes(2)
 
-		// Should have called cancel instead of end
-		expect(calls).toEqual(['start', 'change', 'change', 'cancel'])
+		// Once on end
+		expect(fnEnd).toHaveBeenCalledTimes(1)
 	})
 
 	it('cleans up events', () => {
@@ -156,7 +94,7 @@ describe('When interacting with a shape...', () => {
 	it('fires double click handler event', () => {
 		const util = editor.getShapeUtil<TLGeoShape>('geo')
 
-		const fnStart = vi.fn()
+		const fnStart = jest.fn()
 		util.onDoubleClick = fnStart
 
 		editor.doubleClick(50, 50, ids.box2)
@@ -167,19 +105,14 @@ describe('When interacting with a shape...', () => {
 	it('Fires resisizing events', () => {
 		const util = editor.getShapeUtil<TLFrameShape>('frame')
 
-		const calls: string[] = []
-
-		util.onResizeStart = () => {
-			calls.push('start')
-		}
+		const fnStart = jest.fn()
+		util.onResizeStart = fnStart
 
-		util.onResize = () => {
-			calls.push('change')
-		}
+		const fnChange = jest.fn()
+		util.onResize = fnChange
 
-		util.onResizeEnd = () => {
-			calls.push('end')
-		}
+		const fnEnd = jest.fn()
+		util.onResizeEnd = fnEnd
 
 		editor.selectAll()
 		expect(editor.getSelectedShapeIds()).toMatchObject([ids.frame1, ids.box1])
@@ -190,171 +123,54 @@ describe('When interacting with a shape...', () => {
 		})
 
 		editor.expectToBeIn('select.pointing_resize_handle')
-
-		// Should not have called any callbacks yet
-		expect(calls).toEqual([])
-
 		editor.pointerMove(200, 200)
 		editor.expectToBeIn('select.resizing')
-
-		// Should have called start once and change at least once now
-		expect(calls).toEqual(['start', 'change'])
-
 		editor.pointerMove(200, 210)
-
-		// Should have called start once and change multiple times
-		expect(calls).toEqual(['start', 'change', 'change'])
-
 		editor.pointerUp(200, 210)
 		editor.expectToBeIn('select.idle')
 
-		// Should have called end once now
-		expect(calls).toEqual(['start', 'change', 'change', 'end'])
-	})
-
-	it('Fires resizing cancel events', () => {
-		const util = editor.getShapeUtil<TLFrameShape>('frame')
-
-		const calls: string[] = []
-
-		util.onResizeStart = () => {
-			calls.push('start')
-		}
-
-		util.onResize = () => {
-			calls.push('change')
-		}
-
-		util.onResizeEnd = () => {
-			calls.push('end')
-		}
-
-		util.onResizeCancel = () => {
-			calls.push('cancel')
-		}
-
-		editor.selectAll()
-		expect(editor.getSelectedShapeIds()).toMatchObject([ids.frame1, ids.box1])
-
-		editor.pointerDown(300, 300, {
-			target: 'selection',
-			handle: 'bottom_right',
-		})
-
-		editor.expectToBeIn('select.pointing_resize_handle')
-
-		// Should not have called any callbacks yet
-		expect(calls).toEqual([])
-
-		editor.pointerMove(200, 200)
-		editor.expectToBeIn('select.resizing')
-
-		// Should have called start once and change at least once now
-		expect(calls).toEqual(['start', 'change'])
-
-		editor.pointerMove(200, 210)
-
-		// Should have called start once and change multiple times
-		expect(calls).toEqual(['start', 'change', 'change'])
+		// Once on start (for frame only)
+		expect(fnStart).toHaveBeenCalledTimes(1)
 
-		editor.cancel()
+		// Once on start, once during the resize
+		expect(fnChange).toHaveBeenCalledTimes(2)
 
-		// Should have called cancel instead of end
-		expect(calls).toEqual(['start', 'change', 'change', 'cancel'])
+		// Once on end
+		expect(fnEnd).toHaveBeenCalledTimes(1)
 	})
 
 	it('Fires translating events', () => {
 		const util = editor.getShapeUtil<TLFrameShape>('frame')
 
-		const calls: string[] = []
+		const fnStart = jest.fn()
+		util.onTranslateStart = fnStart
 
-		util.onTranslateStart = () => {
-			calls.push('start')
-		}
+		const fnChange = jest.fn()
+		util.onTranslate = fnChange
 
-		util.onTranslate = () => {
-			calls.push('change')
-		}
-
-		util.onTranslateEnd = () => {
-			calls.push('end')
-		}
+		const fnEnd = jest.fn()
+		util.onTranslateEnd = fnEnd
 
 		editor.selectAll()
 		expect(editor.getSelectedShapeIds()).toMatchObject([ids.frame1, ids.box1])
 
 		// Translate the shapes...
-		editor.pointerDown(50, 50, ids.box1)
-
-		// Should not have called any callbacks yet
-		expect(calls).toEqual([])
-
-		editor.pointerMove(50, 40)
-
-		// Should have called start once and change at least once now
-		expect(calls).toEqual(['start', 'change'])
-
-		editor.pointerMove(50, 35)
-
-		// Should have called start once and change multiple times
-		expect(calls).toEqual(['start', 'change', 'change'])
-
-		editor.pointerUp(50, 35)
-
-		// Should have called end once now
-		expect(calls).toEqual(['start', 'change', 'change', 'change', 'end'])
-	})
-
-	it('Fires translating cancel events', () => {
-		const util = editor.getShapeUtil<TLFrameShape>('frame')
-
-		const calls: string[] = []
+		editor.pointerDown(50, 50, ids.box1).pointerMove(50, 40).pointerUp(50, 40)
 
-		util.onTranslateStart = () => {
-			calls.push('start')
-		}
-
-		util.onTranslate = () => {
-			calls.push('change')
-		}
-
-		util.onTranslateEnd = () => {
-			calls.push('end')
-		}
-
-		util.onTranslateCancel = () => {
-			calls.push('cancel')
-		}
-
-		editor.selectAll()
-		expect(editor.getSelectedShapeIds()).toMatchObject([ids.frame1, ids.box1])
-
-		// Translate the shapes...
-		editor.pointerDown(50, 50, ids.box1)
-
-		// Should not have called any callbacks yet
-		expect(calls).toEqual([])
-
-		editor.pointerMove(50, 40)
-
-		// Should have called start once and change at least once now
-		expect(calls).toEqual(['start', 'change'])
-
-		editor.pointerMove(50, 35)
-
-		// Should have called start once and change multiple times
-		expect(calls).toEqual(['start', 'change', 'change'])
+		// Once on start for frame
+		expect(fnStart).toHaveBeenCalledTimes(1)
 
-		editor.cancel()
+		// Once on start, once during the move
+		expect(fnChange).toHaveBeenCalledTimes(2)
 
-		// Should have called cancel instead of end
-		expect(calls).toEqual(['start', 'change', 'change', 'cancel'])
+		// Once on end
+		expect(fnEnd).toHaveBeenCalledTimes(1)
 	})
 
 	it('Uses the shape utils onClick handler', () => {
 		const util = editor.getShapeUtil<TLFrameShape>('frame')
 
-		const fnClick = vi.fn()
+		const fnClick = jest.fn()
 		util.onClick = fnClick
 
 		editor.pointerDown(50, 50, ids.frame1)
@@ -368,7 +184,7 @@ describe('When interacting with a shape...', () => {
 	it('Uses the shape utils onClick handler', () => {
 		const util = editor.getShapeUtil<TLFrameShape>('frame')
 
-		const fnClick = vi.fn((shape: any) => {
+		const fnClick = jest.fn((shape: any) => {
 			return {
 				...shape,
 				x: 100,
@@ -385,170 +201,4 @@ describe('When interacting with a shape...', () => {
 		// it should not be selected.
 		expect(editor.getSelectedShapeIds().length).toBe(0)
 	})
-
-	it('Fires handle dragging events', () => {
-		const util = editor.getShapeUtil<TLLineShape>('line')
-
-		const calls: string[] = []
-
-		util.onHandleDragStart = () => {
-			calls.push('start')
-		}
-
-		util.onHandleDrag = () => {
-			calls.push('change')
-		}
-
-		util.onHandleDragEnd = () => {
-			calls.push('end')
-		}
-
-		util.onHandleDragCancel = () => {
-			calls.push('cancel')
-		}
-
-		// Create a line shape with handles
-		const lineShape: TLLineShape = {
-			id: ids.line1,
-			type: 'line',
-			typeName: 'shape',
-			parentId: ids.page1,
-			index: 'a1' as any,
-			x: 100,
-			y: 100,
-			rotation: 0,
-			isLocked: false,
-			opacity: 1,
-			meta: {},
-			props: {
-				dash: 'draw',
-				size: 'm',
-				color: 'black',
-				spline: 'line',
-				scale: 1,
-				points: {
-					a1: { id: 'a1', index: 'a1' as any, x: 0, y: 0 },
-					a2: { id: 'a2', index: 'a2' as any, x: 100, y: 100 },
-				},
-			},
-		}
-
-		editor.createShapes([lineShape])
-
-		// Get the handle point
-		const handlePagePoint = editor
-			.getShapePageTransform(lineShape.id)!
-			.applyToPoint(lineShape.props.points['a2'])
-
-		editor.pointerDown(handlePagePoint.x, handlePagePoint.y, {
-			target: 'handle',
-			shape: editor.getShape(lineShape.id)!,
-			handle: { id: 'a2', type: 'vertex', index: 'a2' as any, x: 100, y: 100 },
-		})
-
-		editor.expectToBeIn('select.pointing_handle')
-
-		// Should not have called any callbacks yet
-		expect(calls).toEqual([])
-
-		editor.pointerMove(handlePagePoint.x + 20, handlePagePoint.y + 20) // Larger move to trigger drag
-		editor.expectToBeIn('select.dragging_handle')
-
-		// Should have called start once and change at least once now
-		expect(calls).toEqual(['start', 'change'])
-
-		editor.pointerMove(150, 150)
-
-		// Should have called start once and change multiple times
-		expect(calls).toEqual(['start', 'change', 'change'])
-
-		editor.pointerUp(150, 150)
-		editor.expectToBeIn('select.idle')
-
-		// Should have called end once now
-		expect(calls).toEqual(['start', 'change', 'change', 'end'])
-	})
-
-	it('Fires handle dragging cancel events', () => {
-		const util = editor.getShapeUtil<TLLineShape>('line')
-
-		const calls: string[] = []
-
-		util.onHandleDragStart = () => {
-			calls.push('start')
-		}
-
-		util.onHandleDrag = () => {
-			calls.push('change')
-		}
-
-		util.onHandleDragEnd = () => {
-			calls.push('end')
-		}
-
-		util.onHandleDragCancel = () => {
-			calls.push('cancel')
-		}
-
-		// Create a line shape with handles
-		const lineShape: TLLineShape = {
-			id: ids.line1,
-			type: 'line',
-			typeName: 'shape',
-			parentId: ids.page1,
-			index: 'a1' as any,
-			x: 100,
-			y: 100,
-			rotation: 0,
-			isLocked: false,
-			opacity: 1,
-			meta: {},
-			props: {
-				dash: 'draw',
-				size: 'm',
-				color: 'black',
-				spline: 'line',
-				scale: 1,
-				points: {
-					a1: { id: 'a1', index: 'a1' as any, x: 0, y: 0 },
-					a2: { id: 'a2', index: 'a2' as any, x: 100, y: 100 },
-				},
-			},
-		}
-
-		editor.createShapes([lineShape])
-
-		// Get the handle point
-		const handlePagePoint = editor
-			.getShapePageTransform(lineShape.id)!
-			.applyToPoint(lineShape.props.points['a2'])
-
-		editor.pointerDown(handlePagePoint.x, handlePagePoint.y, {
-			target: 'handle',
-			shape: editor.getShape(lineShape.id)!,
-			handle: { id: 'a2', type: 'vertex', index: 'a2' as any, x: 100, y: 100 },
-		})
-
-		editor.expectToBeIn('select.pointing_handle')
-
-		// Should not have called any callbacks yet
-		expect(calls).toEqual([])
-
-		editor.pointerMove(handlePagePoint.x + 20, handlePagePoint.y + 20) // Larger move to trigger drag
-		editor.expectToBeIn('select.dragging_handle')
-
-		// Should have called start once and change at least once now
-		expect(calls).toEqual(['start', 'change'])
-
-		editor.pointerMove(150, 150)
-
-		// Should have called start once and change multiple times
-		expect(calls).toEqual(['start', 'change', 'change'])
-
-		editor.cancel()
-		editor.expectToBeIn('select.idle')
-
-		// Should have called cancel instead of end
-		expect(calls).toEqual(['start', 'change', 'change', 'cancel'])
-	})
 })