coze-studio/frontend/packages/common/chat-area/text-grab/__tests__/get-all-node-in-range.test.ts

72 lines
2.6 KiB
TypeScript

/*
* Copyright 2025 coze-dev Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import { getAllNodesInRange } from '../src/utils/helper/get-all-nodes-in-range';
describe('getAllNodesInRange', () => {
let root: HTMLElement;
beforeEach(() => {
// Setting up the DOM structure before each test case
document.body.innerHTML = `
<div id="root">
<span>Text 1</span>
<div>
<p>Text 2</p>
</div>
Text 3
</div>
`;
root = document.getElementById('root') as HTMLElement;
});
it('should return all nodes within a range, including text and element nodes', () => {
const range = new Range();
range.setStart(root, 0); // Setup scope starts at the first sub-node of root
range.setEnd(root, 3); // The setting range ends at the last sub-node of root.
const nodes = getAllNodesInRange(range);
// Expected to include: span, div, Text 3
expect(nodes.length).toBe(5);
expect((nodes[0] as Node).nodeType).toBe(Node.ELEMENT_NODE); // span
expect((nodes[1] as Node).nodeType).toBe(Node.TEXT_NODE); // text
expect((nodes[2] as Node).nodeType).toBe(Node.ELEMENT_NODE); // span
expect((nodes[3] as Node).nodeType).toBe(Node.TEXT_NODE); // text
expect((nodes[4] as Node).nodeType).toBe(Node.TEXT_NODE); // Text 3
});
it('should return an empty array if the range is collapsed', () => {
const range = document.createRange();
range.setStart(root, 1); // Set the start and end of the range at the same location
range.setEnd(root, 1); // This will create a collapsed scope that contains no nodes
const nodes = getAllNodesInRange(range);
expect(nodes).toEqual([root]);
});
it('should correctly handle a range that only includes a text node', () => {
const textNode = root.childNodes[2] as ChildNode; // Text 3
const range = document.createRange();
range.setStartBefore(textNode);
range.setEndAfter(textNode);
const nodes = getAllNodesInRange(range);
expect(nodes.length).toBe(2);
expect((nodes[0] as Node).nodeType).toBe(Node.ELEMENT_NODE); // Text 3
expect((nodes[1] as Node).nodeType).toBe(Node.TEXT_NODE); // Text 3
});
});