/*
 * 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(() => {
    // 在每个测试用例之前设置 DOM 结构
    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); // 设置范围开始于 root 的第一个子节点
    range.setEnd(root, 3); // 设置范围结束于 root 的最后一个子节点

    const nodes = getAllNodesInRange(range);
    // 预期包含：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); // 设置范围的开始和结束都在同一位置
    range.setEnd(root, 1); // 这将创建一个折叠的范围，即没有包含任何节点

    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
  });
});