1 files changed, 96 insertions, 0 deletions
diff --git a/spec/frontend/content_editor/services/table_of_contents_utils_spec.js b/spec/frontend/content_editor/services/table_of_contents_utils_spec.js
new file mode 100644
index 00000000000..7f63c2171c2
--- /dev/null
+++ b/spec/frontend/content_editor/services/table_of_contents_utils_spec.js
@@ -0,0 +1,96 @@
+import Heading from '~/content_editor/extensions/heading';
+import { toTree, getHeadings } from '~/content_editor/services/table_of_contents_utils';
+import { createTestEditor, createDocBuilder } from '../test_utils';
+
+describe('content_editor/services/table_of_content_utils', () => {
+  describe('toTree', () => {
+    it('should fills in gaps in heading levels and convert headings to a tree', () => {
+      expect(
+        toTree([
+          { level: 3, text: '3' },
+          { level: 2, text: '2' },
+        ]),
+      ).toEqual([
+        expect.objectContaining({
+          level: 1,
+          text: '',
+          subHeadings: [
+            expect.objectContaining({
+              level: 2,
+              text: '',
+              subHeadings: [expect.objectContaining({ level: 3, text: '3', subHeadings: [] })],
+            }),
+            expect.objectContaining({ level: 2, text: '2', subHeadings: [] }),
+          ],
+        }),
+      ]);
+    });
+  });
+
+  describe('getHeadings', () => {
+    const tiptapEditor = createTestEditor({
+      extensions: [Heading],
+    });
+
+    const {
+      builders: { heading, doc },
+    } = createDocBuilder({
+      tiptapEditor,
+      names: {
+        heading: { nodeType: Heading.name },
+      },
+    });
+
+    it('gets all headings as a tree in a tiptap document', () => {
+      const initialDoc = doc(
+        heading({ level: 1 }, 'Heading 1'),
+        heading({ level: 2 }, 'Heading 1.1'),
+        heading({ level: 3 }, 'Heading 1.1.1'),
+        heading({ level: 2 }, 'Heading 1.2'),
+        heading({ level: 3 }, 'Heading 1.2.1'),
+        heading({ level: 2 }, 'Heading 1.3'),
+        heading({ level: 2 }, 'Heading 1.4'),
+        heading({ level: 3 }, 'Heading 1.4.1'),
+        heading({ level: 1 }, 'Heading 2'),
+      );
+
+      tiptapEditor.commands.setContent(initialDoc.toJSON());
+
+      expect(getHeadings(tiptapEditor)).toEqual([
+        expect.objectContaining({
+          level: 1,
+          text: 'Heading 1',
+          subHeadings: [
+            expect.objectContaining({
+              level: 2,
+              text: 'Heading 1.1',
+              subHeadings: [
+                expect.objectContaining({ level: 3, text: 'Heading 1.1.1', subHeadings: [] }),
+              ],
+            }),
+            expect.objectContaining({
+              level: 2,
+              text: 'Heading 1.2',
+              subHeadings: [
+                expect.objectContaining({ level: 3, text: 'Heading 1.2.1', subHeadings: [] }),
+              ],
+            }),
+            expect.objectContaining({ level: 2, text: 'Heading 1.3', subHeadings: [] }),
+            expect.objectContaining({
+              level: 2,
+              text: 'Heading 1.4',
+              subHeadings: [
+                expect.objectContaining({ level: 3, text: 'Heading 1.4.1', subHeadings: [] }),
+              ],
+            }),
+          ],
+        }),
+        expect.objectContaining({
+          level: 1,
+          text: 'Heading 2',
+          subHeadings: [],
+        }),
+      ]);
+    });
+  });
+});