/plugin/src/test/java/net/customware/confluence/plugin/toc/StaxDocumentOutlineCreatorTest.java

https://bitbucket.org/atlassian/confluence-toc-plugin · Java · 178 lines · 118 code · 35 blank · 25 comment · 0 complexity · f452f16d6a695bb10d454c21a143e6fa MD5 · raw file 

    

  1. package net.customware.confluence.plugin.toc;
    2. 

  2. 

  3. import com.atlassian.confluence.content.render.xhtml.DefaultXmlEventReaderFactory;
    4. 

  4. import com.atlassian.confluence.content.render.xhtml.DelegateXmlOutputFactory;
    5. 

  5. import com.ctc.wstx.api.WstxOutputProperties;
    6. 

  6. import net.customware.confluence.plugin.toc.DocumentOutline.Heading;
    7. 

  7. import org.junit.Before;
    8. 

  8. import org.junit.Test;
    9. 

  9. 

  10. import javax.xml.stream.XMLOutputFactory;
    11. 

  11. import java.util.Iterator;
    12. 

  12. 

  13. import static org.junit.Assert.assertEquals;
    14. 

  14. 

  15. public class StaxDocumentOutlineCreatorTest {
    16. 

  16.     private StaxDocumentOutlineCreator staxBuilder;
    17. 

  17. 

  18.     @Before
    19. 

  19.     public void setUp() throws Exception {
    20. 

  20.         XMLOutputFactory xmlOutputFactory = XMLOutputFactory.newInstance();
    21. 

  21.         xmlOutputFactory.setProperty(WstxOutputProperties.P_OUTPUT_VALIDATE_STRUCTURE, Boolean.FALSE);
    22. 

  22.         xmlOutputFactory.setProperty(WstxOutputProperties.P_AUTOMATIC_END_ELEMENTS, Boolean.FALSE);
    23. 

  23. 

  24.         staxBuilder = new StaxDocumentOutlineCreator(new DefaultXmlEventReaderFactory(), new DelegateXmlOutputFactory(xmlOutputFactory));
    25. 

  25.     }
    26. 

  26. 

  27.     @Test
    28. 

  28.     public void testMissingTopHeadings() throws Exception {
    29. 

  29.         Iterator<Heading> iterator = staxBuilder.getOutline("<body><p>nothing</p><h3 id=\"anchor1\">Heading 3</h3><h2 id=\"anchor2\">Heading 2</h2><p>nothing</p><h3 id=\"anchor3\">SubHeading 3</h3></body>").iterator();
    30. 

  30. 

  31.         Heading heading = iterator.next();
    32. 

  32.         assertEquals("Heading 3", heading.getName());
    33. 

  33.         assertEquals("anchor1", heading.getAnchor());
    34. 

  34.         assertEquals(2, heading.getEffectiveLevel());
    35. 

  35.         assertEquals(0, heading.getChildCount());
    36. 

  36. 

  37.         heading = iterator.next();
    38. 

  38.         assertEquals("Heading 2", heading.getName());
    39. 

  39.         assertEquals("anchor2", heading.getAnchor());
    40. 

  40.         assertEquals(1, heading.getEffectiveLevel());
    41. 

  41.         assertEquals(1, heading.getChildCount());
    42. 

  42. 

  43.         heading = iterator.next();
    44. 

  44.         assertEquals("SubHeading 3", heading.getName());
    45. 

  45.         assertEquals("anchor3", heading.getAnchor());
    46. 

  46.         assertEquals(2, heading.getEffectiveLevel());
    47. 

  47.         assertEquals(0, heading.getChildCount());
    48. 

  48.     }
    49. 

  49. 

  50.     @Test
    51. 

  51.     public void testOnlyOneLevelOfHeadings() throws Exception {
    52. 

  52.         Iterator<Heading> iterator = staxBuilder.getOutline("<body><p>nothing</p><h1>a</h1><p>stuff</p><H1>b</H1><H1>c</H1></body>").iterator();
    53. 

  53. 

  54.         Heading heading = iterator.next();
    55. 

  55.         assertEquals("a", heading.getName());
    56. 

  56.         assertEquals(1, heading.getEffectiveLevel());
    57. 

  57.         assertEquals(0, heading.getChildCount());
    58. 

  58. 

  59.         heading = iterator.next();
    60. 

  60.         assertEquals("b", heading.getName());
    61. 

  61.         assertEquals(1, heading.getEffectiveLevel());
    62. 

  62.         assertEquals(0, heading.getChildCount());
    63. 

  63. 

  64.         heading = iterator.next();
    65. 

  65.         assertEquals("c", heading.getName());
    66. 

  66.         assertEquals(1, heading.getEffectiveLevel());
    67. 

  67.         assertEquals(0, heading.getChildCount());
    68. 

  68.     }
    69. 

  69. 

  70.     @Test
    71. 

  71.     public void testABigJumbleOfHeadings() throws Exception {
    72. 

  72.         Iterator<Heading> iterator = staxBuilder.getOutline("<body><h1>Level-1-Heading-1</h1><h2>Level-2-Heading-1</h2><h3>Level-3-Heading-<b>1</b></h3><h2>Level-2-Heading-2</h2><h5>Level-5-Heading-1</h5><h6 id='TOCtest-Level6Heading1'>Level-6-Heading-1</h6><h4>Level-4-Heading-1</h4><h2>Level-2-Heading-3</h2></body>").iterator();
    73. 

  73. 

  74.         Heading heading = iterator.next();
    75. 

  75.         assertEquals("Level-1-Heading-1", heading.getName());
    76. 

  76.         assertEquals(1, heading.getEffectiveLevel());
    77. 

  77. 

  78.         heading = iterator.next();
    79. 

  79.         assertEquals("Level-2-Heading-1", heading.getName());
    80. 

  80.         assertEquals(2, heading.getEffectiveLevel());
    81. 

  81. 

  82.         heading = iterator.next();
    83. 

  83.         assertEquals("Level-3-Heading-1", heading.getName());
    84. 

  84.         assertEquals(3, heading.getEffectiveLevel());
    85. 

  85. 

  86.         heading = iterator.next();
    87. 

  87.         assertEquals("Level-2-Heading-2", heading.getName());
    88. 

  88.         assertEquals(2, heading.getEffectiveLevel());
    89. 

  89. 

  90.         heading = iterator.next();
    91. 

  91.         assertEquals("Level-5-Heading-1", heading.getName());
    92. 

  92.         assertEquals(4, heading.getEffectiveLevel());
    93. 

  93. 

  94.         heading = iterator.next();
    95. 

  95.         assertEquals("Level-6-Heading-1", heading.getName());
    96. 

  96.         assertEquals(5, heading.getEffectiveLevel());
    97. 

  97. 

  98.         heading = iterator.next();
    99. 

  99.         assertEquals("Level-4-Heading-1", heading.getName());
    100. 

  100.         assertEquals(3, heading.getEffectiveLevel());
    101. 

  101. 

  102.         heading = iterator.next();
    103. 

  103.         assertEquals("Level-2-Heading-3", heading.getName());
    104. 

  104.         assertEquals(2, heading.getEffectiveLevel());
    105. 

  105.     }
    106. 

  106. 

  107.     @Test
    108. 

  108.     public void testAnchorSupport() throws Exception {
    109. 

  109.         Iterator<Heading> iterator = staxBuilder.getOutline("<body><h2><a href=\"#\">Header</a> <i>with</i><a name='anchor1'> <b>two</b></a> <a name=\"anchor2\">anchors</a></h2></body>").iterator();
    110. 

  110.         Heading heading = iterator.next();
    111. 

  111. 

  112.         assertEquals("Header with two anchors", heading.getName());
    113. 

  113.         assertEquals("anchor1", heading.getAnchor());
    114. 

  114.     }
    115. 

  115. 

  116.     @Test
    117. 

  117.     public void testNumberedHeadingsWithAnchors() throws Exception {
    118. 

  118.         /*
    119. 

  119.          *  <h1 id="numberedheadings-Heading-1heading1">
    120. 

  120.          *      <span class="nh-number">1. </span>
    121. 

  121.          *      Heading-1
    122. 

  122.          *      <span class="confluence-anchor-link" id="numberedheadings-heading1"></span>
    123. 

  123.          *  </h1>
    124. 

  124.          */
    125. 

  125.         Iterator<Heading> iterator = staxBuilder.getOutline("<body><h1 id=\"numberedheadings-Heading-1heading1\"><span class=\"nh-number\">1. </span>Heading-1<span class=\"confluence-anchor-link\" id=\"numberedheadings-heading1\"></span></h1></body>").iterator();
    126. 

  126. 

  127.         Heading heading = iterator.next();
    128. 

  128.         assertEquals("1. Heading-1", heading.getName());
    129. 

  129.         assertEquals("numberedheadings-Heading-1heading1", heading.getAnchor());
    130. 

  130. 

  131.     }
    132. 

  132. 

  133.     @Test
    134. 

  134.     public void testHeadingsStartWithAnchors() throws Exception {
    135. 

  135.         /*
    136. 

  136.          * <h1 id="heading-h1Heading-1">
    137. 

  137.          *     <span class="confluence-anchor-link" id="heading-h1"></span>
    138. 

  138.          *     Heading-1
    139. 

  139.          *  </h1>
    140. 

  140.          */
    141. 

  141.         Iterator<Heading> iterator = staxBuilder.getOutline("<body><h1 id=\"heading-h1Heading-1\"><span class=\"confluence-anchor-link\" id=\"heading-h1\"></span>Heading-1</h1></body>").iterator();
    142. 

  142. 

  143.         Heading heading = iterator.next();
    144. 

  144.         assertEquals("Heading-1", heading.getName());
    145. 

  145.         assertEquals("heading-h1Heading-1", heading.getAnchor());
    146. 

  146.     }
    147. 

  147. 

  148.     @Test
    149. 

  149.     public void testHeadingsEndWithAnchors() throws Exception {
    150. 

  150.         /*
    151. 

  151.          * <h1 id="heading-Heading-1heading1">
    152. 

  152.          *     Heading-1&nbsp;
    153. 

  153.          *     <span class="confluence-anchor-link" id="heading-heading1"></span>
    154. 

  154.          * </h1>
    155. 

  155.          */
    156. 

  156.         Iterator<Heading> iterator = staxBuilder.getOutline("<body><h1 id=\"heading-Heading-1heading1\">Heading-1<span class=\"confluence-anchor-link\" id=\"heading-heading1\"></span></h1></body>").iterator();
    157. 

  157. 

  158.         Heading heading = iterator.next();
    159. 

  159.         assertEquals("Heading-1", heading.getName());
    160. 

  160.         assertEquals("heading-Heading-1heading1", heading.getAnchor());
    161. 

  161.     }
    162. 

  162. 

  163.     @Test
    164. 

  164.     public void testColorHeadings() throws Exception {
    165. 

  165.         /*
    166. 

  166.          *  <h1 id="colorheadings-Heading1withtext">
    167. 

  167.          *      <span style="color: rgb(255,0,0);">Heading 1</span>
    168. 

  168.          *      with text
    169. 

  169.          *  </h1>
    170. 

  170.          */
    171. 

  171.         Iterator<Heading> iterator = staxBuilder.getOutline("<body><h1 id=\"colorheadings-Heading1withtext\"><span style=\"color: rgb(255,0,0);\">Heading 1</span> with text</h1></body>").iterator();
    172. 

  172. 

  173.         Heading heading = iterator.next();
    174. 

  174.         assertEquals("Heading 1 with text", heading.getName());
    175. 

  175.         assertEquals("colorheadings-Heading1withtext", heading.getAnchor());
    176. 

  176.     }
    177. 

  177. 

  178. }
    179.