import { unified } from 'unified';
import remarkParse from 'remark-parse';
import remark2rehype from 'remark-rehype';
import { rehype } from 'rehype';
import rehypeRaw from 'rehype-raw';
import stringify from 'rehype-stringify';
import rehypeRewrite, { getCodeString } from '../index.js';
describe('getCodeString test case', () => {
it('selector options', async () => {
const str = getCodeString([
{
type: 'text',
value: 'Hello'
},
{
type: 'element',
tagName: 'span',
properties: {
class: 'octicon octicon-link',
},
children: [
{
type: 'text',
value: 'World'
},
]
}
]);
expect(str).toEqual('HelloWorld');
expect(getCodeString()).toEqual('');
expect(getCodeString(undefined, 'good')).toEqual('good');
});
})
describe('rehype-rewrite test case', () => {
it('selector options', async () => {
const html = `header
`;
const expected = `header
`
const htmlStr = rehype()
.data('settings', { fragment: true })
.use(rehypeRewrite, {
selector: 'h1',
rewrite: (node) => {
if (node.type === 'element') {
node.properties = { ...node.properties, class: 'test' }
}
}
})
.use(stringify)
.processSync(html)
.toString()
expect(htmlStr).toEqual(expected);
});
it('selector options', async () => {
const html = `header
header
header
`;
const expected = `header
header
header
`
const htmlStr = rehype()
.data('settings', { fragment: true })
.use(rehypeRewrite, {
selector: 'h1',
rewrite: (node) => {
if (node.type === 'element') {
node.properties!.className = 'test';
}
}
})
.use(stringify)
.processSync(html)
.toString()
expect(htmlStr).toEqual(expected);
});
it('selector = undefined', async () => {
const html = `header
`;
const expected = `header
`
const htmlStr = rehype()
.data('settings', { fragment: true })
.use(rehypeRewrite, {
selector: 'h1.good',
rewrite: (node) => {
if (node.type === 'element') {
node.properties = { ...node.properties, class: 'test' }
}
}
})
.use(stringify)
.processSync(html)
.toString()
expect(htmlStr).toEqual(expected);
});
it('selector = undefined', async () => {
const html = `header
`;
const expected = `header
`
const htmlStr = rehype()
.data('settings', { fragment: true })
.use(rehypeRewrite)
.use(stringify)
.processSync(html)
.toString()
expect(htmlStr).toEqual(expected);
});
it('remove text', async () => {
const html = `header
`;
const expected = ``
const htmlStr = rehype()
.data('settings', { fragment: true })
.use(rehypeRewrite, {
rewrite: (node) => {
if(node.type == 'text' && node.value == 'header') {
node.value = ''
}
}
})
.use(stringify)
.processSync(html)
.toString()
expect(htmlStr).toEqual(expected);
});
it('modify properties', async () => {
const html = `header
`;
const expected = `header
`
const htmlStr = rehype()
.use(rehypeRewrite, {
rewrite: (node) => {
if (node.type == 'element' && node.tagName == 'body') {
node.properties = { ...node.properties, style: 'color:red;'}
}
}
})
.use(stringify)
.processSync(html)
.toString()
expect(htmlStr).toEqual(expected);
});
it('add element', async () => {
const html = `hello
`;
const expected = `hello world
`
const htmlStr = rehype()
.data('settings', { fragment: true })
.use(rehypeRewrite, {
rewrite: (node) => {
if (node.type == 'element' && node.tagName == 'h1') {
node.children = [ ...node.children, {
type: 'element',
tagName: 'span',
properties: {},
children: [
{type: 'text', value: ' world'}
]
}]
}
}
})
.use(stringify)
.processSync(html)
.toString()
expect(htmlStr).toEqual(expected);
});
it('use unified', async () => {
const markdown = "Hello World!"
const expected = `Hello World!
`
const htmlStr = unified()
.use(remarkParse)
.use(remark2rehype, { allowDangerousHtml: true })
.use(rehypeRaw)
.use(rehypeRewrite, {
rewrite: (node) => {
if (node.type == 'element' && node.tagName == 'p') {
node.properties = { ...node.properties, style: 'color:red;' }
}
}
})
.use(stringify)
.processSync(markdown)
.toString()
expect(htmlStr).toEqual(expected);
});
it('options parameter test', async () => {
const markdown = "Hello World!"
const expected = `Hello World!`
const htmlStr = rehype()
.data('settings', { fragment: true })
.use(rehypeRewrite, {} as any)
.use(stringify)
.processSync(markdown)
.toString()
expect(htmlStr).toEqual(expected);
});
});