fix: [#1952] Accept Document nodes as valid boundary points in Selection API (#1954)

Author: skoch13

packages/happy-dom/src/selection/Selection.ts

@@ -266,7 +266,10 @@ export default class Selection {
 			);
 		}
 
-		if (node[PropertySymbol.ownerDocument] !== this.#ownerDocument) {
+		if (
+			node !== this.#ownerDocument &&
+			node[PropertySymbol.ownerDocument] !== this.#ownerDocument
+		) {
 			return;
 		}
 
@@ -388,7 +391,10 @@ export default class Selection {
 	 * @param offset Offset.
 	 */
 	public extend(node: Node, offset: number): void {
-		if (node[PropertySymbol.ownerDocument] !== this.#ownerDocument) {
+		if (
+			node !== this.#ownerDocument &&
+			node[PropertySymbol.ownerDocument] !== this.#ownerDocument
+		) {
 			return;
 		}
 
@@ -451,7 +457,10 @@ export default class Selection {
 			);
 		}
 
-		if (node[PropertySymbol.ownerDocument] !== this.#ownerDocument) {
+		if (
+			node !== this.#ownerDocument &&
+			node[PropertySymbol.ownerDocument] !== this.#ownerDocument
+		) {
 			return;
 		}
 
@@ -492,8 +501,10 @@ export default class Selection {
 		}
 
 		if (
-			anchorNode[PropertySymbol.ownerDocument] !== this.#ownerDocument ||
-			focusNode[PropertySymbol.ownerDocument] !== this.#ownerDocument
+			(anchorNode !== this.#ownerDocument &&
+				anchorNode[PropertySymbol.ownerDocument] !== this.#ownerDocument) ||
+			(focusNode !== this.#ownerDocument &&
+				focusNode[PropertySymbol.ownerDocument] !== this.#ownerDocument)
 		) {
 			return;
 		}

packages/happy-dom/test/selection/Selection.test.ts

@@ -419,6 +419,19 @@ describe('Selection', () => {
 				expect((<Event>(<unknown>triggeredEvent)).bubbles).toBe(false);
 				expect((<Event>(<unknown>triggeredEvent)).cancelable).toBe(false);
 			});
+
+			it('Accepts Document node.', () => {
+				selection[method](document, 0);
+
+				expect(selection.rangeCount).toBe(1);
+
+				const newRange = selection.getRangeAt(0);
+
+				expect(newRange.startContainer).toBe(document);
+				expect(newRange.startOffset).toBe(0);
+				expect(newRange.endContainer).toBe(document);
+				expect(newRange.endOffset).toBe(0);
+			});
 		});
 	}
 
@@ -659,6 +672,19 @@ describe('Selection', () => {
 			expect((<Event>(<unknown>triggeredEvent)).bubbles).toBe(false);
 			expect((<Event>(<unknown>triggeredEvent)).cancelable).toBe(false);
 		});
+
+		it('Accepts Document node.', () => {
+			const text = document.createTextNode('text');
+
+			document.body.appendChild(text);
+
+			selection.collapse(text, 0);
+			selection.extend(document, 0);
+
+			expect(selection.rangeCount).toBe(1);
+			expect(selection.focusNode).toBe(document);
+			expect(selection.focusOffset).toBe(0);
+		});
 	});
 
 	describe('selectAllChildren()', () => {
@@ -682,6 +708,19 @@ describe('Selection', () => {
 			expect(newRange.endOffset).toBe(3);
 		});
 
+		it('Accepts Document node.', () => {
+			selection.selectAllChildren(document);
+
+			expect(selection.rangeCount).toBe(1);
+
+			const newRange = selection.getRangeAt(0);
+
+			expect(newRange.startContainer).toBe(document);
+			expect(newRange.startOffset).toBe(0);
+			expect(newRange.endContainer).toBe(document);
+			expect(newRange.endOffset).toBe(document.childNodes.length);
+		});
+
 		it(`Throws error if node type is ${NodeTypeEnum.documentTypeNode}.`, () => {
 			const documentType = document.implementation.createDocumentType('', '', '');
 
@@ -748,6 +787,21 @@ describe('Selection', () => {
 			expect(selection.anchorNode).toBe(newRange.endContainer);
 		});
 
+		it('Accepts Document node as boundary point.', () => {
+			const text = document.createTextNode('text');
+
+			document.body.appendChild(text);
+
+			selection.setBaseAndExtent(document, 0, text, 2);
+
+			expect(selection.rangeCount).toBe(1);
+
+			const newRange = selection.getRangeAt(0);
+
+			expect(newRange.startContainer).toBe(document);
+			expect(newRange.endContainer).toBe(text);
+		});
+
 		it('Throws error if wrong offset.', () => {
 			const start = document.createTextNode('start');
 			const end = document.createTextNode('end');