import { describe } from 'node:test'
import { createTLStore } from '../config/createTLStore'
import { Editor } from '../editor/Editor'
import { StateNode } from '../editor/tools/StateNode'

let editor: Editor

class A extends StateNode {
	static override id = 'A'
}

class B extends StateNode {
	static override id = 'B'
}

class C extends StateNode {
	static override id = 'C'

	override onEnter() {
		this.setCurrentToolIdMask('A')
	}
}

beforeEach(() => {
	editor = new Editor({
		initialState: 'A',
		shapeUtils: [],
		bindingUtils: [],
		tools: [A, B, C],
		store: createTLStore({ shapeUtils: [] }),
		getContainer: () => document.body,
	})
})

describe('current tool id mask', () => {
	it('starts with the correct tool id', () => {
		expect(editor.getCurrentToolId()).toBe('A')
	})

	it('updates the current tool id', () => {
		editor.setCurrentTool('B')
		expect(editor.getCurrentToolId()).toBe('B')
	})

	it('masks the current tool id', () => {
		editor.setCurrentTool('C')
		expect(editor.getCurrentToolId()).toBe('A')
	})
})