View Javadoc
1   /*
2    * Copyright (c) 2002-2025 Gargoyle Software Inc.
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    * https://www.apache.org/licenses/LICENSE-2.0
8    *
9    * Unless required by applicable law or agreed to in writing, software
10   * distributed under the License is distributed on an "AS IS" BASIS,
11   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12   * See the License for the specific language governing permissions and
13   * limitations under the License.
14   */
15  package org.htmlunit.html;
16  
17  import org.htmlunit.SimpleWebTestCase;
18  import org.junit.jupiter.api.Assertions;
19  import org.junit.jupiter.api.Test;
20  import org.w3c.dom.traversal.NodeFilter;
21  import org.w3c.dom.traversal.NodeIterator;
22  
23  /**
24   * Tests for {@link DomNodeIterator}.
25   *
26   * @author Ahmed Ashour
27   * @author Ronald Brill
28   */
29  public final class DomNodeIterator2Test extends SimpleWebTestCase {
30  
31      /**
32       * Test case for issue 1982.
33       * @throws Exception if the test fails
34       */
35      @Test
36      public void subroot() throws Exception {
37          final String html = DOCTYPE_HTML
38              + "<html>\n"
39              + "<head>\n"
40              + "</head>\n"
41              + "<body>\n"
42              + "<table>\n"
43              + "  <tr id='1'>\n"
44              + "    <td id='11'>11</td>\n"
45              + "    <td id='12'>12</td>\n"
46              + "  </tr>\n"
47              + "  <tr id='2'>\n"
48              + "    <td id='21'>21</td>\n"
49              + "    <td id='22'>22</td>\n"
50              + "  </tr>\n"
51              + "</table>\n"
52              + "</body></html>";
53  
54          final HtmlPage page = loadPage(html);
55  
56          final HtmlElement subroot = (HtmlElement) page.getElementById("1");
57          final NodeIterator iterator = page.createNodeIterator(subroot, NodeFilter.SHOW_ELEMENT, null, true);
58  
59          HtmlElement element = (HtmlElement) iterator.nextNode();
60          Assertions.assertEquals("1", element.getId());
61  
62          element = (HtmlElement) iterator.nextNode();
63          Assertions.assertEquals("11", element.getId());
64  
65          element = (HtmlElement) iterator.nextNode();
66          Assertions.assertEquals("12", element.getId());
67  
68          Assertions.assertNull(iterator.nextNode());
69      }
70  }