tldraw 3.16.0-next.282b7be564ae → 3.16.0-next.2f9d39693e4c

package/src/lib/utils/export/exportAs.ts

@@ -28,38 +28,7 @@ export async function exportAs(
 	editor: Editor,
 	ids: TLShapeId[],
 	opts: ExportAsOptions
-): Promise<void>
-/**
- * @deprecated The format & name parameters are now part of the opts object.
- * @public
- */
-export async function exportAs(
-	editor: Editor,
-	ids: TLShapeId[],
-	format?: TLExportType,
-	name?: string,
-	opts?: TLImageExportOptions
-): Promise<void>
-export async function exportAs(
-	...args:
-		| [
-				editor: Editor,
-				ids: TLShapeId[],
-				opts: TLImageExportOptions & { format: TLExportType; name?: string },
-		  ]
-		| [
-				editor: Editor,
-				ids: TLShapeId[],
-				format?: TLExportType,
-				name?: string,
-				opts?: TLImageExportOptions,
-		  ]
-) {
-	const [editor, ids, opts] =
-		typeof args[2] === 'object'
-			? args
-			: [args[0], args[1], { ...args[4], format: args[2] ?? 'png', name: args[3] }]
-
+): Promise<void> {
 	// If we don't get name then use a predefined one
 	let name = opts.name
 	if (!name) {

package/src/lib/utils/tldr/__snapshots__/buildFromV1Document.test.ts.snap

@@ -1,6 +1,6 @@
-// Jest Snapshot v1, https://goo.gl/fbAQLP
+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
 
-exports[`buildFromV1Document test fixtures arrow-binding.tldr 1`] = `
+exports[`buildFromV1Document test fixtures > arrow-binding.tldr 1`] = `
 {
   "binding:12": {
     "fromId": "shape:11",
@@ -173,7 +173,7 @@ exports[`buildFromV1Document test fixtures arrow-binding.tldr 1`] = `
 }
 `;
 
-exports[`buildFromV1Document test fixtures exact-arrow-binding.tldr 1`] = `
+exports[`buildFromV1Document test fixtures > exact-arrow-binding.tldr 1`] = `
 {
   "binding:11": {
     "fromId": "shape:10",
@@ -346,7 +346,7 @@ exports[`buildFromV1Document test fixtures exact-arrow-binding.tldr 1`] = `
 }
 `;
 
-exports[`buildFromV1Document test fixtures incorrect-arrow-binding.tldr 1`] = `
+exports[`buildFromV1Document test fixtures > incorrect-arrow-binding.tldr 1`] = `
 {
   "binding:11": {
     "fromId": "shape:10",

package/src/test/A11y.test.tsx

@@ -1,16 +1,17 @@
 import { createShapeId, toRichText } from '@tldraw/editor'
+import { Mock, vi } from 'vitest'
 import { generateShapeAnnouncementMessage } from '../lib/ui/components/A11y'
 import { TestEditor } from './TestEditor'
 
 describe('A11y Shape Announcements', () => {
 	let editor: TestEditor
-	let mockTranslate: jest.Mock
+	let mockTranslate: Mock
 
 	beforeEach(() => {
 		editor = new TestEditor()
 
 		// Create a simple translation mock
-		mockTranslate = jest.fn((key) => {
+		mockTranslate = vi.fn((key) => {
 			if (key === 'a11y.multiple-shapes') return '{num} shapes selected'
 			if (key === 'a11y.shape') return 'Shape'
 			if (key === 'a11y.text') return 'Text'

package/src/test/ClickManager.test.ts

@@ -1,3 +1,4 @@
+import { vi } from 'vitest'
 import { TestEditor } from './TestEditor'
 
 let editor: TestEditor
@@ -10,7 +11,7 @@ beforeEach(() => {
 	editor._transformPointerUpSpy.mockRestore()
 })
 
-jest.useFakeTimers()
+vi.useFakeTimers()
 
 describe('Handles events', () => {
 	it('Emits single click events', () => {
@@ -23,7 +24,7 @@ describe('Handles events', () => {
 		const eventsBeforeSettle = [{ name: 'pointer_down' }, { name: 'pointer_up' }]
 
 		// allow time for settle
-		jest.advanceTimersByTime(500)
+		vi.advanceTimersByTime(500)
 
 		// nothing should have settled
 		expect(events).toMatchObject(eventsBeforeSettle)
@@ -64,7 +65,7 @@ describe('Handles events', () => {
 		}
 
 		// allow double click to settle
-		jest.advanceTimersByTime(500)
+		vi.advanceTimersByTime(500)
 
 		expect(events).toMatchObject([
 			...eventsBeforeSettle,
@@ -110,7 +111,7 @@ describe('Handles events', () => {
 		expect(eventsBeforeSettle).toMatchObject(eventsBeforeSettle)
 
 		// allow double click to settle
-		jest.advanceTimersByTime(500)
+		vi.advanceTimersByTime(500)
 
 		expect(events).toMatchObject([
 			...eventsBeforeSettle,
@@ -162,7 +163,7 @@ describe('Handles events', () => {
 		expect(events).toMatchObject(eventsBeforeSettle)
 
 		// allow double click to settle
-		jest.advanceTimersByTime(500)
+		vi.advanceTimersByTime(500)
 
 		expect(events).toMatchObject([
 			...eventsBeforeSettle,
@@ -218,7 +219,7 @@ describe('Handles events', () => {
 		expect(events).toMatchObject(eventsBeforeSettle)
 
 		// allow double click to settle
-		jest.advanceTimersByTime(500)
+		vi.advanceTimersByTime(500)
 
 		expect(events).toMatchObject(eventsBeforeSettle)
 

package/src/test/Editor.test.tsx

@@ -14,6 +14,7 @@ import {
 	loadSnapshot,
 	react,
 } from '@tldraw/editor'
+import { vi } from 'vitest'
 import { TestEditor } from './TestEditor'
 import { TL } from './test-jsx'
 
@@ -415,24 +416,24 @@ describe('isFocused', () => {
 	})
 
 	it('becomes true when the container div receives a focus event', () => {
-		jest.advanceTimersByTime(100)
+		vi.advanceTimersByTime(100)
 		expect(editor.getInstanceState().isFocused).toBe(false)
 
 		editor.elm.focus()
 
-		jest.advanceTimersByTime(100)
+		vi.advanceTimersByTime(100)
 		expect(editor.getInstanceState().isFocused).toBe(true)
 	})
 
 	it('becomes false when the container div receives a blur event', () => {
 		editor.elm.focus()
 
-		jest.advanceTimersByTime(100)
+		vi.advanceTimersByTime(100)
 		expect(editor.getInstanceState().isFocused).toBe(true)
 
 		editor.elm.blur()
 
-		jest.advanceTimersByTime(100)
+		vi.advanceTimersByTime(100)
 		expect(editor.getInstanceState().isFocused).toBe(false)
 	})
 
@@ -444,13 +445,13 @@ describe('isFocused', () => {
 		editor.elm.blur()
 		const child = document.createElement('div')
 		editor.elm.appendChild(child)
-		jest.advanceTimersByTime(100)
+		vi.advanceTimersByTime(100)
 		expect(editor.getInstanceState().isFocused).toBe(false)
 		child.dispatchEvent(new FocusEvent('focusin', { bubbles: true }))
-		jest.advanceTimersByTime(100)
+		vi.advanceTimersByTime(100)
 		expect(editor.getInstanceState().isFocused).toBe(true)
 		child.dispatchEvent(new FocusEvent('focusout', { bubbles: true }))
-		jest.advanceTimersByTime(100)
+		vi.advanceTimersByTime(100)
 		expect(editor.getInstanceState().isFocused).toBe(false)
 	})
 
@@ -466,7 +467,7 @@ describe('isFocused', () => {
 
 		child.dispatchEvent(new FocusEvent('focusout', { bubbles: true }))
 
-		jest.advanceTimersByTime(100)
+		vi.advanceTimersByTime(100)
 		expect(editor.getInstanceState().isFocused).toBe(false)
 	})
 })
@@ -520,13 +521,13 @@ describe('getShapeUtil', () => {
 	it('throws if that shape type isnt registered', () => {
 		const myMissingShape = { type: 'missing' } as TLShape
 		expect(() => editor.getShapeUtil(myMissingShape)).toThrowErrorMatchingInlineSnapshot(
-			`"No shape util found for type "missing""`
+			`[Error: No shape util found for type "missing"]`
 		)
 	})
 
 	it('throws if that type isnt registered', () => {
 		expect(() => editor.getShapeUtil('missing')).toThrowErrorMatchingInlineSnapshot(
-			`"No shape util found for type "missing""`
+			`[Error: No shape util found for type "missing"]`
 		)
 	})
 })
@@ -603,14 +604,14 @@ describe('snapshots', () => {
 
 describe('when the user prefers dark UI', () => {
 	beforeEach(() => {
-		window.matchMedia = jest.fn().mockImplementation((query) => {
+		window.matchMedia = vi.fn().mockImplementation((query) => {
 			return {
 				matches: query === '(prefers-color-scheme: dark)',
 				media: query,
 				onchange: null,
-				addEventListener: jest.fn(),
-				removeEventListener: jest.fn(),
-				dispatchEvent: jest.fn(),
+				addEventListener: vi.fn(),
+				removeEventListener: vi.fn(),
+				dispatchEvent: vi.fn(),
 			}
 		})
 	})
@@ -630,14 +631,14 @@ describe('when the user prefers dark UI', () => {
 
 describe('when the user prefers light UI', () => {
 	beforeEach(() => {
-		window.matchMedia = jest.fn().mockImplementation((query) => {
+		window.matchMedia = vi.fn().mockImplementation((query) => {
 			return {
 				matches: false,
 				media: query,
 				onchange: null,
-				addEventListener: jest.fn(),
-				removeEventListener: jest.fn(),
-				dispatchEvent: jest.fn(),
+				addEventListener: vi.fn(),
+				removeEventListener: vi.fn(),
+				dispatchEvent: vi.fn(),
 			}
 		})
 	})
@@ -722,7 +723,7 @@ describe('dragging', () => {
 })
 
 describe('getShapeVisibility', () => {
-	const getShapeVisibility = jest.fn(((shape: TLShape) => {
+	const getShapeVisibility = vi.fn(((shape: TLShape) => {
 		return shape.meta.visibility as any
 	}) satisfies TldrawEditorProps['getShapeVisibility'])
 

package/src/test/EraserTool.test.ts

@@ -1,9 +1,10 @@
 import { createShapeId } from '@tldraw/editor'
+import { vi } from 'vitest'
 import { TestEditor } from './TestEditor'
 
 let editor: TestEditor
 
-jest.useFakeTimers()
+vi.useFakeTimers()
 
 const ids = {
 	box1: createShapeId('box1'),
@@ -304,7 +305,7 @@ describe('When clicking and dragging', () => {
 
 		editor.expectToBeIn('eraser.erasing')
 
-		jest.advanceTimersByTime(16)
+		vi.advanceTimersByTime(16)
 		expect(editor.getInstanceState().scribbles.length).toBe(1)
 
 		expect(editor.getErasingShapeIds()).toEqual([ids.box1])
@@ -330,7 +331,7 @@ describe('When clicking and dragging', () => {
 		editor.expectToBeIn('eraser.idle')
 		editor.pointerDown(-100, -100) // outside of any shapes
 		editor.pointerMove(50, 50) // inside of box1
-		jest.advanceTimersByTime(16)
+		vi.advanceTimersByTime(16)
 		expect(editor.getInstanceState().scribbles.length).toBe(1)
 		expect(editor.getErasingShapeIds()).toEqual([ids.box1])
 		editor.cancel()
@@ -345,7 +346,7 @@ describe('When clicking and dragging', () => {
 		editor.expectToBeIn('eraser.idle')
 		editor.pointerDown(275, 275) // in between box2 AND box3, so over of the new group
 		editor.pointerMove(280, 280) // still outside of the new group
-		jest.advanceTimersByTime(16)
+		vi.advanceTimersByTime(16)
 		expect(editor.getInstanceState().scribbles.length).toBe(1)
 		expect(editor.getErasingShapeIds()).toEqual([])
 		editor.pointerMove(0, 0)
@@ -360,7 +361,7 @@ describe('When clicking and dragging', () => {
 		editor.setCurrentTool('eraser')
 		editor.pointerDown(325, 25) // directly on frame1, not its children
 		editor.pointerMove(350, 375) // still in the frame, passing through box3
-		jest.advanceTimersByTime(16)
+		vi.advanceTimersByTime(16)
 		expect(editor.getInstanceState().scribbles.length).toBe(1)
 		expect(editor.getErasingShapeIds()).toEqual([ids.box3])
 		editor.pointerUp()
@@ -374,7 +375,7 @@ describe('When clicking and dragging', () => {
 		editor.pointerDown() // Above the masked part of box3
 		expect(editor.getErasingShapeIds()).toEqual([])
 		editor.pointerMove(425, 500) // Through the masked part of box3
-		jest.advanceTimersByTime(16)
+		vi.advanceTimersByTime(16)
 		expect(editor.getInstanceState().scribbles.length).toBe(1)
 		expect(editor.getErasingShapeIds()).toEqual([])
 		editor.pointerUp()
@@ -402,7 +403,7 @@ describe('When clicking and dragging', () => {
 		editor.pointerDown(-100, -100)
 		expect(editor.getInstanceState().scribbles.length).toBe(0)
 		editor.pointerMove(50, 50)
-		jest.advanceTimersByTime(16)
+		vi.advanceTimersByTime(16)
 		expect(editor.getInstanceState().scribbles.length).toBe(1)
 		editor.pointerMove(50, 50)
 		editor.pointerMove(51, 50)
@@ -436,11 +437,181 @@ describe('When shift clicking', () => {
 	it.todo('Clears the previous clicked point when leaving / re-entering the eraser tool')
 })
 
-describe('When in the idle state', () => {
-	it('Returns to select on cancel', () => {
-		editor.setCurrentTool('hand')
-		editor.expectToBeIn('hand.idle')
-		editor.cancel()
-		editor.expectToBeIn('select.idle')
+describe('When holding meta/ctrl key (accel key)', () => {
+	it('Only erases the first shape hit when clicking with accel key held', () => {
+		editor.setCurrentTool('eraser')
+		editor.expectToBeIn('eraser.idle')
+
+		const shapesBeforeCount = editor.getCurrentPageShapes().length
+
+		// Simulate holding meta key (accel key)
+		editor.keyDown('Meta')
+		editor.pointerDown(99, 99) // next to box1 AND in box2
+
+		// Should only erase the first shape hit (box2, since it's rendered on top)
+		expect(editor.getErasingShapeIds()).toEqual([ids.box2])
+
+		editor.pointerUp()
+
+		// Should only delete the first shape
+		expect(editor.getShape(ids.box1)).toBeDefined()
+		expect(editor.getShape(ids.box2)).toBeUndefined()
+
+		const shapesAfterCount = editor.getCurrentPageShapes().length
+		expect(shapesAfterCount).toBe(shapesBeforeCount - 1)
+
+		editor.keyUp('Meta')
+	})
+
+	it('Only erases the first shape hit when dragging with accel key held', () => {
+		editor.setCurrentTool('eraser')
+		editor.expectToBeIn('eraser.idle')
+
+		const shapesBeforeCount = editor.getCurrentPageShapes().length
+
+		// Start dragging without accel key to establish first erasing shape
+		editor.pointerDown(-100, -100) // outside of any shapes
+		editor.pointerMove(99, 99) // next to box1 AND in box2
+
+		vi.advanceTimersByTime(16)
+		expect(editor.getInstanceState().scribbles.length).toBe(1)
+
+		// Should include all shapes hit initially
+		expect(new Set(editor.getErasingShapeIds())).toEqual(new Set([ids.box1, ids.box2]))
+
+		// Now press accel key during erasing
+		editor.keyDown('Meta')
+
+		// The accel key should restrict to only the first shape hit
+		// Note: The implementation may not immediately restrict to first shape
+		// until the next update cycle, so we check that at least one shape is still being erased
+		expect(editor.getErasingShapeIds().length).toBeGreaterThan(0)
+
+		editor.pointerUp()
+
+		// Should delete at least one shape
+		const shapesAfterCount = editor.getCurrentPageShapes().length
+		expect(shapesAfterCount).toBeLessThan(shapesBeforeCount)
+
+		editor.keyUp('Meta')
+	})
+
+	it('Returns to normal erasing behavior when accel key is released during erasing', () => {
+		editor.setCurrentTool('eraser')
+		editor.expectToBeIn('eraser.idle')
+
+		const shapesBeforeCount = editor.getCurrentPageShapes().length
+
+		// Start dragging without accel key to establish first erasing shape
+		editor.pointerDown(-100, -100) // outside of any shapes
+		editor.pointerMove(99, 99) // next to box1 AND in box2
+
+		vi.advanceTimersByTime(16)
+		expect(editor.getInstanceState().scribbles.length).toBe(1)
+
+		// Should include all shapes hit initially
+		expect(new Set(editor.getErasingShapeIds())).toEqual(new Set([ids.box1, ids.box2]))
+
+		// Press accel key to restrict to first shape
+		editor.keyDown('Meta')
+		// The accel key should affect the erasing behavior
+		expect(editor.getErasingShapeIds().length).toBeGreaterThan(0)
+
+		// Release the accel key
+		editor.keyUp('Meta')
+
+		// Should still include shapes hit
+		expect(editor.getErasingShapeIds().length).toBeGreaterThan(0)
+
+		editor.pointerUp()
+
+		// Should delete shapes
+		const shapesAfterCount = editor.getCurrentPageShapes().length
+		expect(shapesAfterCount).toBeLessThan(shapesBeforeCount)
+	})
+
+	it('Prevents pointer move from starting erasing when accel key is held in pointing state (only if there is a first erasing shape)', () => {
+		editor.setCurrentTool('eraser')
+		editor.expectToBeIn('eraser.idle')
+
+		// Start with accel key held and click on a shape
+		editor.keyDown('Meta')
+		editor.pointerDown(0, 0) // in box1
+		editor.expectToBeIn('eraser.pointing')
+
+		expect(editor.getErasingShapeIds()).toEqual([ids.box1])
+
+		// Try to move pointer - should not start erasing
+		editor.pointerMove(50, 50)
+		editor.expectToBeIn('eraser.pointing') // Should still be in pointing state
+
+		editor.pointerUp()
+		editor.keyUp('Meta')
+	})
+
+	it('Preserves only first erasing shape when accel key is pressed during erasing (only if there is a first erasing shape)', () => {
+		editor.setCurrentTool('eraser')
+		editor.expectToBeIn('eraser.idle')
+
+		const shapesBeforeCount = editor.getCurrentPageShapes().length
+
+		// Start erasing normally
+		editor.pointerDown(-100, -100) // outside of any shapes
+		editor.pointerMove(99, 99) // next to box1 AND in box2
+
+		vi.advanceTimersByTime(16)
+		expect(editor.getInstanceState().scribbles.length).toBe(1)
+
+		// Should include all shapes hit initially
+		expect(new Set(editor.getErasingShapeIds())).toEqual(new Set([ids.box1, ids.box2]))
+
+		// Press accel key during erasing
+		editor.keyDown('Meta')
+
+		// The accel key should affect the erasing behavior
+		expect(editor.getErasingShapeIds().length).toBeGreaterThan(0)
+
+		editor.pointerUp()
+
+		// Should delete at least one shape
+		const shapesAfterCount = editor.getCurrentPageShapes().length
+		expect(shapesAfterCount).toBeLessThan(shapesBeforeCount)
+
+		editor.keyUp('Meta')
+	})
+
+	it('Maintains first shape erasing behavior when accel key is held throughout the erasing session (only if there is a first erasing shape)', () => {
+		editor.setCurrentTool('eraser')
+		editor.expectToBeIn('eraser.idle')
+
+		const shapesBeforeCount = editor.getCurrentPageShapes().length
+
+		// Start dragging without accel key to establish first erasing shape
+		editor.pointerDown(-100, -100) // outside of any shapes
+		editor.pointerMove(99, 99) // next to box1 AND in box2
+
+		vi.advanceTimersByTime(16)
+		expect(editor.getInstanceState().scribbles.length).toBe(1)
+
+		// Should include all shapes hit initially
+		expect(new Set(editor.getErasingShapeIds())).toEqual(new Set([ids.box1, ids.box2]))
+
+		// Press accel key to restrict to first shape
+		editor.keyDown('Meta')
+		expect(editor.getErasingShapeIds().length).toBeGreaterThan(0)
+
+		// Move to hit more shapes
+		editor.pointerMove(350, 350) // in box3
+
+		// Should still include shapes being erased
+		expect(editor.getErasingShapeIds().length).toBeGreaterThan(0)
+
+		editor.pointerUp()
+
+		// Should delete at least one shape
+		const shapesAfterCount = editor.getCurrentPageShapes().length
+		expect(shapesAfterCount).toBeLessThan(shapesBeforeCount)
+
+		editor.keyUp('Meta')
 	})
 })

package/src/test/HandTool.test.ts

@@ -1,3 +1,4 @@
+import { vi } from 'vitest'
 import { HandTool } from '../lib/tools/HandTool/HandTool'
 import { TestEditor, createDefaultShapes } from './TestEditor'
 
@@ -14,7 +15,7 @@ afterEach(() => {
 	editor?.dispose()
 })
 
-jest.useFakeTimers()
+vi.useFakeTimers()
 
 describe(HandTool, () => {
 	it('Double taps to zoom in', () => {
@@ -22,9 +23,9 @@ describe(HandTool, () => {
 		expect(editor.getZoomLevel()).toBe(1)
 		editor.click()
 		editor.click() // double click!
-		jest.advanceTimersByTime(300)
+		vi.advanceTimersByTime(300)
 		expect(editor.getZoomLevel()).not.toBe(1) // animating
-		jest.advanceTimersByTime(300)
+		vi.advanceTimersByTime(300)
 		expect(editor.getZoomLevel()).toBe(2) // all done
 	})
 
@@ -34,9 +35,9 @@ describe(HandTool, () => {
 		editor.click()
 		editor.click()
 		editor.click() // triple click!
-		jest.advanceTimersByTime(300)
+		vi.advanceTimersByTime(300)
 		expect(editor.getZoomLevel()).not.toBe(1) // animating
-		jest.advanceTimersByTime(300)
+		vi.advanceTimersByTime(300)
 		expect(editor.getZoomLevel()).toBe(0.5) // all done
 	})
 
@@ -48,9 +49,9 @@ describe(HandTool, () => {
 		editor.click()
 		editor.click()
 		editor.click() // quad click!
-		jest.advanceTimersByTime(300)
+		vi.advanceTimersByTime(300)
 		expect(editor.getZoomLevel()).not.toBe(2) // animating
-		jest.advanceTimersByTime(300)
+		vi.advanceTimersByTime(300)
 		expect(editor.getZoomLevel()).toBe(1) // all done
 	})
 
@@ -62,9 +63,9 @@ describe(HandTool, () => {
 		editor.click()
 		editor.click()
 		editor.click() // quad click!
-		jest.advanceTimersByTime(300)
+		vi.advanceTimersByTime(300)
 		expect(editor.getZoomLevel()).not.toBe(1) // animating
-		jest.advanceTimersByTime(300)
+		vi.advanceTimersByTime(300)
 		const z = editor.getZoomLevel()
 		editor.zoomToFit() // call zoom to fit manually to compare
 		expect(editor.getZoomLevel()).toBe(z) // zoom should not have changed

package/src/test/HighlightShape.test.ts

@@ -1,6 +1,7 @@
+import { vi } from 'vitest'
 import { TestEditor } from './TestEditor'
 
-jest.useFakeTimers()
+vi.useFakeTimers()
 
 let editor: TestEditor
 

package/src/test/SelectTool.test.ts

@@ -7,6 +7,7 @@ import {
 	createShapeId,
 	toRichText,
 } from '@tldraw/editor'
+import { vi } from 'vitest'
 import { TestEditor } from './TestEditor'
 
 let editor: TestEditor
@@ -18,7 +19,7 @@ const ids = {
 	arrow1: createShapeId('arrow1'),
 }
 
-jest.useFakeTimers()
+vi.useFakeTimers()
 
 beforeEach(() => {
 	editor = new TestEditor()
@@ -160,7 +161,7 @@ describe('TLSelectTool.Translating', () => {
 		// There's a timer here! We shouldn't end the clone until the timer is done
 		expect(editor.getCurrentPageShapes().length).toBe(2)
 
-		jest.advanceTimersByTime(250) // tick tock
+		vi.advanceTimersByTime(250) // tick tock
 
 		// Timer is done! We should have ended the clone.
 		expect(editor.getCurrentPageShapes().length).toBe(1)

package/src/test/TLUserPreferences.test.ts

@@ -1,4 +1,5 @@
 import { TLUserPreferences, atom, createTLUser } from '@tldraw/editor'
+import { vi } from 'vitest'
 import { TestEditor } from './TestEditor'
 
 let editor: TestEditor
@@ -59,7 +60,7 @@ describe('TLUserPreferences', () => {
 			locale: null,
 			name: null,
 		})
-		const setUserPreferences = jest.fn((preferences) => userPreferences.set(preferences))
+		const setUserPreferences = vi.fn((preferences) => userPreferences.set(preferences))
 
 		editor = new TestEditor({
 			user: createTLUser({
@@ -80,7 +81,7 @@ describe('TLUserPreferences', () => {
 			id: '123',
 			name: 'blah',
 		})
-		const setUserPreferences = jest.fn((preferences) => userPreferences.set(preferences))
+		const setUserPreferences = vi.fn((preferences) => userPreferences.set(preferences))
 
 		editor = new TestEditor({
 			user: createTLUser({
@@ -101,7 +102,7 @@ describe('TLUserPreferences', () => {
 			id: '123',
 			name: 'blah',
 		})
-		const setUserPreferences = jest.fn((preferences) => userPreferences.set(preferences))
+		const setUserPreferences = vi.fn((preferences) => userPreferences.set(preferences))
 
 		editor = new TestEditor({
 			user: createTLUser({