chore: better HTML normalization test helper (#16056)

* chore: better HTML normalization test helper

* simplify

* simplify/robustify

* tweak

* Apply suggestions from code review

Co-authored-by: Simon H <5968653+dummdidumm@users.noreply.github.com>

---------

Co-authored-by: Simon H <5968653+dummdidumm@users.noreply.github.com>
pull/16075/head
Rich Harris 3 months ago committed by GitHub
parent 335779dcf5
commit 2342c8719a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -1,8 +1,20 @@
import { assert } from 'vitest';
/** @param {Element} node */
function clean_children(node) {
/**
* @param {Element} node
* @param {{ preserveComments: boolean }} opts
*/
function clean_children(node, opts) {
let previous = null;
let has_element_children = false;
let template =
node.nodeName === 'TEMPLATE' ? /** @type {HTMLTemplateElement} */ (node) : undefined;
if (template) {
const div = document.createElement('div');
div.append(template.content);
node = div;
}
// sort attributes
const attributes = Array.from(node.attributes).sort((a, b) => {
@ -14,6 +26,11 @@ function clean_children(node) {
});
attributes.forEach((attr) => {
// Strip out the special onload/onerror hydration events from the test output
if ((attr.name === 'onload' || attr.name === 'onerror') && attr.value === 'this.__e=event') {
return;
}
node.setAttribute(attr.name, attr.value);
});
@ -27,24 +44,43 @@ function clean_children(node) {
node.tagName !== 'tspan'
) {
node.removeChild(child);
continue;
}
text.data = text.data.replace(/[ \t\n\r\f]+/g, '\n');
text.data = text.data.replace(/[^\S]+/g, ' ');
if (previous && previous.nodeType === 3) {
const prev = /** @type {Text} */ (previous);
prev.data += text.data;
prev.data = prev.data.replace(/[ \t\n\r\f]+/g, '\n');
node.removeChild(text);
text = prev;
text.data = text.data.replace(/[^\S]+/g, ' ');
continue;
}
}
} else if (child.nodeType === 8) {
if (child.nodeType === 8 && !opts.preserveComments) {
// comment
// do nothing
} else {
clean_children(/** @type {Element} */ (child));
child.remove();
continue;
}
// add newlines for better readability and potentially recurse into children
if (child.nodeType === 1 || child.nodeType === 8) {
if (previous?.nodeType === 3) {
const prev = /** @type {Text} */ (previous);
prev.data = prev.data.replace(/^[^\S]+$/, '\n');
} else if (previous?.nodeType === 1 || previous?.nodeType === 8) {
node.insertBefore(document.createTextNode('\n'), child);
}
if (child.nodeType === 1) {
has_element_children = true;
clean_children(/** @type {Element} */ (child), opts);
}
}
previous = child;
@ -53,37 +89,36 @@ function clean_children(node) {
// collapse whitespace
if (node.firstChild && node.firstChild.nodeType === 3) {
const text = /** @type {Text} */ (node.firstChild);
text.data = text.data.replace(/^[ \t\n\r\f]+/, '');
if (!text.data.length) node.removeChild(text);
text.data = text.data.trimStart();
}
if (node.lastChild && node.lastChild.nodeType === 3) {
const text = /** @type {Text} */ (node.lastChild);
text.data = text.data.replace(/[ \t\n\r\f]+$/, '');
if (!text.data.length) node.removeChild(text);
text.data = text.data.trimEnd();
}
// indent code for better readability
if (has_element_children && node.parentNode) {
node.innerHTML = `\n\ ${node.innerHTML.replace(/\n/g, '\n ')}\n`;
}
if (template) {
template.innerHTML = node.innerHTML;
}
}
/**
* @param {Window} window
* @param {string} html
* @param {{ removeDataSvelte?: boolean, preserveComments?: boolean }} param2
* @param {{ preserveComments?: boolean }} opts
*/
export function normalize_html(
window,
html,
{ removeDataSvelte = false, preserveComments = false }
) {
export function normalize_html(window, html, { preserveComments = false } = {}) {
try {
const node = window.document.createElement('div');
node.innerHTML = html
.replace(/(<!(--)?.*?\2>)/g, preserveComments ? '$1' : '')
.replace(/(data-svelte-h="[^"]+")/g, removeDataSvelte ? '' : '$1')
.replace(/>[ \t\n\r\f]+</g, '><')
// Strip out the special onload/onerror hydration events from the test output
.replace(/\s?onerror="this.__e=event"|\s?onload="this.__e=event"/g, '')
.trim();
clean_children(node);
node.innerHTML = html.trim();
clean_children(node, { preserveComments });
return node.innerHTML;
} catch (err) {
throw new Error(`Failed to normalize HTML:\n${html}\nCause: ${err}`);
@ -99,53 +134,47 @@ export function normalize_new_line(html) {
}
/**
* @param {{ removeDataSvelte?: boolean }} options
*/
export function setup_html_equal(options = {}) {
/**
* @param {string} actual
* @param {string} expected
* @param {string} [message]
*/
const assert_html_equal = (actual, expected, message) => {
export const assert_html_equal = (actual, expected, message) => {
try {
assert.deepEqual(
normalize_html(window, actual, options),
normalize_html(window, expected, options),
message
);
assert.deepEqual(normalize_html(window, actual), normalize_html(window, expected), message);
} catch (e) {
if (Error.captureStackTrace)
Error.captureStackTrace(/** @type {Error} */ (e), assert_html_equal);
throw e;
}
};
};
/**
/**
*
* @param {string} actual
* @param {string} expected
* @param {{ preserveComments?: boolean, withoutNormalizeHtml?: boolean }} param2
* @param {string} [message]
*/
const assert_html_equal_with_options = (
export const assert_html_equal_with_options = (
actual,
expected,
{ preserveComments, withoutNormalizeHtml },
message
) => {
) => {
try {
assert.deepEqual(
withoutNormalizeHtml
? normalize_new_line(actual.trim())
.replace(/(\sdata-svelte-h="[^"]+")/g, options.removeDataSvelte ? '' : '$1')
.replace(/(<!(--)?.*?\2>)/g, preserveComments !== false ? '$1' : '')
: normalize_html(window, actual.trim(), { ...options, preserveComments }),
? normalize_new_line(actual.trim()).replace(
/(<!(--)?.*?\2>)/g,
preserveComments !== false ? '$1' : ''
)
: normalize_html(window, actual.trim(), { preserveComments }),
withoutNormalizeHtml
? normalize_new_line(expected.trim())
.replace(/(\sdata-svelte-h="[^"]+")/g, options.removeDataSvelte ? '' : '$1')
.replace(/(<!(--)?.*?\2>)/g, preserveComments !== false ? '$1' : '')
: normalize_html(window, expected.trim(), { ...options, preserveComments }),
? normalize_new_line(expected.trim()).replace(
/(<!(--)?.*?\2>)/g,
preserveComments !== false ? '$1' : ''
)
: normalize_html(window, expected.trim(), { preserveComments }),
message
);
} catch (e) {
@ -153,13 +182,4 @@ export function setup_html_equal(options = {}) {
Error.captureStackTrace(/** @type {Error} */ (e), assert_html_equal_with_options);
throw e;
}
};
return {
assert_html_equal,
assert_html_equal_with_options
};
}
// Common case without options
export const { assert_html_equal, assert_html_equal_with_options } = setup_html_equal();
};

@ -25,7 +25,7 @@ export default test({
<p>selected: one</p>
<select>
<option${variant === 'hydrate' ? ' selected' : ''}>one</option$>
<option${variant === 'hydrate' ? ' selected' : ''}>one</option>
<option>two</option>
<option>three</option>
</select>
@ -54,7 +54,7 @@ export default test({
<p>selected: two</p>
<select>
<option${variant === 'hydrate' ? ' selected' : ''}>one</option$>
<option${variant === 'hydrate' ? ' selected' : ''}>one</option>
<option>two</option>
<option>three</option>
</select>

@ -4,7 +4,9 @@ export default test({
html: `
<input list='suggestions'>
<datalist id='suggestions'>
<option value='foo'/><option value='bar'/><option value='baz'/>
<option value='foo'></option>
<option value='bar'></option>
<option value='baz'></option>
</datalist>
`
});

@ -9,7 +9,7 @@ export default test({
<math>
<mrow></mrow>
</svg>
</math>
<div class="hi">hi</div>
`,

@ -7,7 +7,7 @@ export default test({
target.innerHTML,
`
<select>
<option${variant === 'hydrate' ? ' selected' : ''} value="a">A</option$>
<option${variant === 'hydrate' ? ' selected' : ''} value="a">A</option>
<option value="b">B</option>
</select>
selected: a
@ -23,7 +23,7 @@ export default test({
target.innerHTML,
`
<select>
<option${variant === 'hydrate' ? ' selected' : ''} value="a">A</option$>
<option${variant === 'hydrate' ? ' selected' : ''} value="a">A</option>
<option value="b">B</option>
</select>
selected: b

@ -7,7 +7,7 @@ import { flushSync, hydrate, mount, unmount } from 'svelte';
import { render } from 'svelte/server';
import { afterAll, assert, beforeAll } from 'vitest';
import { compile_directory, fragments } from '../helpers.js';
import { setup_html_equal } from '../html_equal.js';
import { assert_html_equal, assert_html_equal_with_options } from '../html_equal.js';
import { raf } from '../animation-helpers.js';
import type { CompileOptions } from '#compiler';
import { suite_with_variants, type BaseTest } from '../suite.js';
@ -86,10 +86,6 @@ function unhandled_rejection_handler(err: Error) {
const listeners = process.rawListeners('unhandledRejection');
const { assert_html_equal, assert_html_equal_with_options } = setup_html_equal({
removeDataSvelte: true
});
beforeAll(() => {
// @ts-expect-error TODO huh?
process.prependListener('unhandledRejection', unhandled_rejection_handler);

Loading…
Cancel
Save