diff --git a/packages/shared/__tests__/escapeHtml.spec.ts b/packages/shared/__tests__/escapeHtml.spec.ts
index 34505d3ea..30004706a 100644
--- a/packages/shared/__tests__/escapeHtml.spec.ts
+++ b/packages/shared/__tests__/escapeHtml.spec.ts
@@ -1,11 +1,31 @@
-import { escapeHtml } from '../src'
+import { escapeHtml, escapeHtmlComment } from '../src'
-test('ssr: escapeHTML', () => {
- expect(escapeHtml(`foo`)).toBe(`foo`)
- expect(escapeHtml(true)).toBe(`true`)
- expect(escapeHtml(false)).toBe(`false`)
- expect(escapeHtml(`a && b`)).toBe(`a && b`)
- expect(escapeHtml(`"foo"`)).toBe(`"foo"`)
- expect(escapeHtml(`'bar'`)).toBe(`'bar'`)
- expect(escapeHtml(`
`)).toBe(`<div>`)
+describe('escapeHtml', () => {
+ test('ssr: escapeHTML', () => {
+ expect(escapeHtml(`foo`)).toBe(`foo`)
+ expect(escapeHtml(true)).toBe(`true`)
+ expect(escapeHtml(false)).toBe(`false`)
+ expect(escapeHtml(`a && b`)).toBe(`a && b`)
+ expect(escapeHtml(`"foo"`)).toBe(`"foo"`)
+ expect(escapeHtml(`'bar'`)).toBe(`'bar'`)
+ expect(escapeHtml(`
`)).toBe(`<div>`)
+ })
+
+ test('ssr: escapeHTMLComment', () => {
+ const input = ''
+ const result = escapeHtmlComment(input)
+ expect(result).toEqual(' Hello World! ')
+ })
+
+ test('ssr: escapeHTMLComment', () => {
+ const input = ' Hello World!'
+ const result = escapeHtmlComment(input)
+ expect(result).toEqual(' Comment 1 Hello ! Comment 2 World!')
+ })
+
+ test('should not affect non-comment strings', () => {
+ const input = 'Hello World'
+ const result = escapeHtmlComment(input)
+ expect(result).toEqual(input)
+ })
})