1 files changed, 259 insertions, 0 deletions
diff --git a/test/document-util.test.js b/test/document-util.test.js
new file mode 100644
index 00000000..f2552f78
--- /dev/null
+++ b/test/document-util.test.js
@@ -0,0 +1,259 @@
+/*
+ * Copyright (C) 2023  Yomitan Authors
+ * Copyright (C) 2020-2022  Yomichan Authors
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+import fs from 'fs';
+import {JSDOM} from 'jsdom';
+import {fileURLToPath} from 'node:url';
+import path from 'path';
+import {expect, test} from 'vitest';
+import {DocumentUtil} from '../ext/js/dom/document-util.js';
+import {DOMTextScanner} from '../ext/js/dom/dom-text-scanner.js';
+import {TextSourceElement} from '../ext/js/dom/text-source-element.js';
+import {TextSourceRange} from '../ext/js/dom/text-source-range.js';
+
+const dirname = path.dirname(fileURLToPath(import.meta.url));
+
+// DOMRect class definition
+class DOMRect {
+    constructor(x, y, width, height) {
+        this._x = x;
+        this._y = y;
+        this._width = width;
+        this._height = height;
+    }
+
+    get x() { return this._x; }
+    get y() { return this._y; }
+    get width() { return this._width; }
+    get height() { return this._height; }
+    get left() { return this._x + Math.min(0, this._width); }
+    get right() { return this._x + Math.max(0, this._width); }
+    get top() { return this._y + Math.min(0, this._height); }
+    get bottom() { return this._y + Math.max(0, this._height); }
+}
+
+
+function createJSDOM(fileName) {
+    const domSource = fs.readFileSync(fileName, {encoding: 'utf8'});
+    const dom = new JSDOM(domSource);
+    const document = dom.window.document;
+    const window = dom.window;
+
+    // Define innerText setter as an alias for textContent setter
+    Object.defineProperty(window.HTMLDivElement.prototype, 'innerText', {
+        set(value) { this.textContent = value; }
+    });
+
+    // Placeholder for feature detection
+    document.caretRangeFromPoint = () => null;
+
+    return dom;
+}
+
+function querySelectorChildOrSelf(element, selector) {
+    return selector ? element.querySelector(selector) : element;
+}
+
+function getChildTextNodeOrSelf(dom, node) {
+    if (node === null) { return null; }
+    const Node = dom.window.Node;
+    const childNode = node.firstChild;
+    return (childNode !== null && childNode.nodeType === Node.TEXT_NODE ? childNode : node);
+}
+
+function getPrototypeOfOrNull(value) {
+    try {
+        return Object.getPrototypeOf(value);
+    } catch (e) {
+        return null;
+    }
+}
+
+function findImposterElement(document) {
+    // Finds the imposter element based on it's z-index style
+    return document.querySelector('div[style*="2147483646"]>*');
+}
+
+
+async function testDocument1() {
+    const dom = createJSDOM(path.join(dirname, 'data', 'html', 'test-document1.html'));
+    const window = dom.window;
+
+    try {
+        await testDocumentTextScanningFunctions(dom);
+        await testTextSourceRangeSeekFunctions(dom);
+    } finally {
+        window.close();
+    }
+}
+
+async function testDocumentTextScanningFunctions(dom) {
+    const document = dom.window.document;
+
+    test('DocumentTextScanningFunctions', () => {
+        for (const testElement of document.querySelectorAll('.test[data-test-type=scan]')) {
+        // Get test parameters
+            let {
+                elementFromPointSelector,
+                caretRangeFromPointSelector,
+                startNodeSelector,
+                startOffset,
+                endNodeSelector,
+                endOffset,
+                resultType,
+                sentenceScanExtent,
+                sentence,
+                hasImposter,
+                terminateAtNewlines
+            } = testElement.dataset;
+
+            const elementFromPointValue = querySelectorChildOrSelf(testElement, elementFromPointSelector);
+            const caretRangeFromPointValue = querySelectorChildOrSelf(testElement, caretRangeFromPointSelector);
+            const startNode = getChildTextNodeOrSelf(dom, querySelectorChildOrSelf(testElement, startNodeSelector));
+            const endNode = getChildTextNodeOrSelf(dom, querySelectorChildOrSelf(testElement, endNodeSelector));
+
+            startOffset = parseInt(startOffset, 10);
+            endOffset = parseInt(endOffset, 10);
+            sentenceScanExtent = parseInt(sentenceScanExtent, 10);
+            terminateAtNewlines = (terminateAtNewlines !== 'false');
+
+            expect(elementFromPointValue).not.toStrictEqual(null);
+            expect(caretRangeFromPointValue).not.toStrictEqual(null);
+            expect(startNode).not.toStrictEqual(null);
+            expect(endNode).not.toStrictEqual(null);
+
+            // Setup functions
+            document.elementFromPoint = () => elementFromPointValue;
+
+            document.caretRangeFromPoint = (x, y) => {
+                const imposter = getChildTextNodeOrSelf(dom, findImposterElement(document));
+                expect(!!imposter).toStrictEqual(hasImposter === 'true');
+
+                const range = document.createRange();
+                range.setStart(imposter ? imposter : startNode, startOffset);
+                range.setEnd(imposter ? imposter : startNode, endOffset);
+
+                // Override getClientRects to return a rect guaranteed to contain (x, y)
+                range.getClientRects = () => [new DOMRect(x - 1, y - 1, 2, 2)];
+                return range;
+            };
+
+            // Test docRangeFromPoint
+            const source = DocumentUtil.getRangeFromPoint(0, 0, {
+                deepContentScan: false,
+                normalizeCssZoom: true
+            });
+            switch (resultType) {
+                case 'TextSourceRange':
+                    expect(getPrototypeOfOrNull(source)).toStrictEqual(TextSourceRange.prototype);
+                    break;
+                case 'TextSourceElement':
+                    expect(getPrototypeOfOrNull(source)).toStrictEqual(TextSourceElement.prototype);
+                    break;
+                case 'null':
+                    expect(source).toStrictEqual(null);
+                    break;
+                default:
+                    expect.unreachable();
+                    break;
+            }
+            if (source === null) { continue; }
+
+            // Sentence info
+            const terminatorString = '…。．.？?！!';
+            const terminatorMap = new Map();
+            for (const char of terminatorString) {
+                terminatorMap.set(char, [false, true]);
+            }
+            const quoteArray = [['「', '」'], ['『', '』'], ['\'', '\''], ['"', '"']];
+            const forwardQuoteMap = new Map();
+            const backwardQuoteMap = new Map();
+            for (const [char1, char2] of quoteArray) {
+                forwardQuoteMap.set(char1, [char2, false]);
+                backwardQuoteMap.set(char2, [char1, false]);
+            }
+
+            // Test docSentenceExtract
+            const sentenceActual = DocumentUtil.extractSentence(
+                source,
+                false,
+                sentenceScanExtent,
+                terminateAtNewlines,
+                terminatorMap,
+                forwardQuoteMap,
+                backwardQuoteMap
+            ).text;
+            expect(sentenceActual).toStrictEqual(sentence);
+
+            // Clean
+            source.cleanup();
+        }
+    });
+}
+
+async function testTextSourceRangeSeekFunctions(dom) {
+    const document = dom.window.document;
+
+    test('TextSourceRangeSeekFunctions', async () => {
+        for (const testElement of document.querySelectorAll('.test[data-test-type=text-source-range-seek]')) {
+        // Get test parameters
+            let {
+                seekNodeSelector,
+                seekNodeIsText,
+                seekOffset,
+                seekLength,
+                seekDirection,
+                expectedResultNodeSelector,
+                expectedResultNodeIsText,
+                expectedResultOffset,
+                expectedResultContent
+            } = testElement.dataset;
+
+            seekOffset = parseInt(seekOffset, 10);
+            seekLength = parseInt(seekLength, 10);
+            expectedResultOffset = parseInt(expectedResultOffset, 10);
+
+            let seekNode = testElement.querySelector(seekNodeSelector);
+            if (seekNodeIsText === 'true') {
+                seekNode = seekNode.firstChild;
+            }
+
+            let expectedResultNode = testElement.querySelector(expectedResultNodeSelector);
+            if (expectedResultNodeIsText === 'true') {
+                expectedResultNode = expectedResultNode.firstChild;
+            }
+
+            const {node, offset, content} = (
+            seekDirection === 'forward' ?
+            new DOMTextScanner(seekNode, seekOffset, true, false).seek(seekLength) :
+            new DOMTextScanner(seekNode, seekOffset, true, false).seek(-seekLength)
+            );
+
+            expect(node).toStrictEqual(expectedResultNode);
+            expect(offset).toStrictEqual(expectedResultOffset);
+            expect(content).toStrictEqual(expectedResultContent);
+        }
+    });
+}
+
+
+async function main() {
+    await testDocument1();
+}
+
+await main();