summaryrefslogtreecommitdiff
path: root/packages/integrations/mdx/test/mdx-rehype-plugins.test.js
blob: d8761b9fbb1163d58f1d7fef9beaab9b9b98148b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import mdx from '@astrojs/mdx';
import { jsToTreeNode } from '../dist/utils.js';

import { expect } from 'chai';
import { parseHTML } from 'linkedom';
import getReadingTime from 'reading-time';
import { toString } from 'mdast-util-to-string';

import { loadFixture } from '../../../astro/test/test-utils.js';

export function rehypeReadingTime() {
	return function (tree) {
		const readingTime = getReadingTime(toString(tree));
		tree.children.unshift(
			jsToTreeNode(`export const readingTime = ${JSON.stringify(readingTime)}`)
		);
	};
}

const FIXTURE_ROOT = new URL('./fixtures/mdx-rehype-plugins/', import.meta.url);

describe('MDX rehype plugins', () => {
	describe('without "extends"', () => {
		let fixture;
		before(async () => {
			fixture = await loadFixture({
				root: FIXTURE_ROOT,
				integrations: [
					mdx({
						rehypePlugins: [rehypeReadingTime],
					}),
				],
			});
			await fixture.build();
		});

		it('removes default getHeadings', async () => {
			const html = await fixture.readFile('/space-ipsum/index.html');
			const { document } = parseHTML(html);

			const headings = [...document.querySelectorAll('h1, h2')];
			expect(headings.length).to.be.greaterThan(0);
			for (const heading of headings) {
				expect(heading.id).to.be.empty;
			}
		});

		it('supports custom rehype plugins - reading time', async () => {
			const readingTime = JSON.parse(await fixture.readFile('/reading-time.json'));

			expect(readingTime).to.not.be.null;
			expect(readingTime.text).to.match(/^\d+ min read/);
		});
	});

	describe('with "extends"', () => {
		let fixture;
		before(async () => {
			fixture = await loadFixture({
				root: FIXTURE_ROOT,
				integrations: [
					mdx({
						rehypePlugins: { extends: [rehypeReadingTime] },
					}),
				],
			});
			await fixture.build();
		});

		it('preserves default getHeadings', async () => {
			const html = await fixture.readFile('/space-ipsum/index.html');
			const { document } = parseHTML(html);

			const headings = [...document.querySelectorAll('h1, h2')];
			expect(headings.length).to.be.greaterThan(0);
			for (const heading of headings) {
				expect(heading.id).to.not.be.empty;
			}
		});
	});
});