From fa65cb6af1e8b126c2b340f71274b8b9dea26c88 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=B6=E8=BF=9C=E6=96=B9?= <yangpanteng@gmail.com>
Date: Fri, 10 Nov 2023 15:07:24 +0800
Subject: [PATCH] test(shared): add test case for escapeHtmlComment (#8065)

---
 packages/shared/__tests__/escapeHtml.spec.ts | 38 +++++++++++++++-----
 1 file changed, 29 insertions(+), 9 deletions(-)

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 &amp;&amp; b`)
-  expect(escapeHtml(`"foo"`)).toBe(`&quot;foo&quot;`)
-  expect(escapeHtml(`'bar'`)).toBe(`&#39;bar&#39;`)
-  expect(escapeHtml(`<div>`)).toBe(`&lt;div&gt;`)
+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 &amp;&amp; b`)
+    expect(escapeHtml(`"foo"`)).toBe(`&quot;foo&quot;`)
+    expect(escapeHtml(`'bar'`)).toBe(`&#39;bar&#39;`)
+    expect(escapeHtml(`<div>`)).toBe(`&lt;div&gt;`)
+  })
+
+  test('ssr: escapeHTMLComment', () => {
+    const input = '<!-- Hello --><!-- World! -->'
+    const result = escapeHtmlComment(input)
+    expect(result).toEqual(' Hello  World! ')
+  })
+
+  test('ssr: escapeHTMLComment', () => {
+    const input = '<!-- Comment 1 --> Hello <!--! Comment 2 --> 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)
+  })
 })