extension-js
diff --git a/‎programs/develop/webpack/plugin-extension/feature-html/__spec__/html-lib/get-assets-from-html.spec.ts
Lines changed: 46 additions & 34 deletions b/‎programs/develop/webpack/plugin-extension/feature-html/__spec__/html-lib/get-assets-from-html.spec.ts
Lines changed: 46 additions & 34 deletions
@@ -1,33 +1,11 @@
 import * as fs from 'fs'
 import {describe, it, expect, vi, beforeEach} from 'vitest'
-// @ts-ignore
-import parse5utils from 'parse5-utils'
 import {getAssetsFromHtml} from '../../html-lib/utils'
 
 vi.mock('fs', () => ({
   readFileSync: vi.fn()
 }))
 
-vi.mock('parse5-utils', () => ({
-  parse: vi.fn(),
-  getAttribute: vi.fn()
-}))
-
-const setupParseMock = (htmlDocument: any) => {
-  ;(parse5utils.parse as any).mockImplementation(() => htmlDocument)
-}
-
-function createMockHtmlDocument(childNodes: any[]): any {
-  return {
-    childNodes: [
-      {
-        nodeName: 'html',
-        childNodes
-      }
-    ]
-  }
-}
-
 describe('getAssetsFromHtml', () => {
   const htmlFilePath = '/path/to/index.html'
 
@@ -39,16 +17,14 @@ describe('getAssetsFromHtml', () => {
     const htmlContent =
       '<html><head><link rel="stylesheet" href="styles.css"></head></html>'
     ;(fs.readFileSync as any).mockReturnValue('This should not be read')
-    setupParseMock(createMockHtmlDocument([]))
 
-    getAssetsFromHtml(htmlFilePath, htmlContent)
+    const assets = getAssetsFromHtml(htmlFilePath, htmlContent)
 
     expect(fs.readFileSync).not.toHaveBeenCalled()
-    expect(parse5utils.parse).toHaveBeenCalledWith(htmlContent)
+    expect(assets?.css).toEqual(['/path/to/styles.css'])
   })
 
   it('should return empty arrays if HTML has no assets', () => {
-    setupParseMock(createMockHtmlDocument([]))
     ;(fs.readFileSync as any).mockReturnValue('<html></html>')
 
     const assets = getAssetsFromHtml(htmlFilePath)
@@ -61,17 +37,53 @@ describe('getAssetsFromHtml', () => {
   it('should ignore external URLs for assets', () => {
     const htmlContent =
       '<html><head><script src="https://cezaraugusto.com/script.js"></script></head></html>'
-    setupParseMock(
-      createMockHtmlDocument([
-        {
-          nodeName: 'script',
-          attrs: [{name: 'src', value: 'https://cezaraugusto.com/script.js'}]
-        }
-      ])
-    )
 
     const assets = getAssetsFromHtml(htmlFilePath, htmlContent)
 
     expect(assets?.js).toEqual([])
   })
+
+  it('should extract JavaScript files', () => {
+    const htmlContent =
+      '<html><head><script src="app.js"></script></head></html>'
+
+    const assets = getAssetsFromHtml(htmlFilePath, htmlContent)
+
+    expect(assets?.js).toEqual(['/path/to/app.js'])
+  })
+
+  it('should extract CSS files', () => {
+    const htmlContent =
+      '<html><head><link rel="stylesheet" href="styles.css"></head></html>'
+
+    const assets = getAssetsFromHtml(htmlFilePath, htmlContent)
+
+    expect(assets?.css).toEqual(['/path/to/styles.css'])
+  })
+
+  it('should extract static assets', () => {
+    const htmlContent = '<html><body><img src="image.png"></body></html>'
+
+    const assets = getAssetsFromHtml(htmlFilePath, htmlContent)
+
+    expect(assets?.static).toEqual(['/path/to/image.png'])
+  })
+
+  it('should handle public paths correctly', () => {
+    const htmlContent =
+      '<html><head><link rel="stylesheet" href="/public/styles.css"></head></html>'
+
+    const assets = getAssetsFromHtml(htmlFilePath, htmlContent, 'public')
+
+    expect(assets?.css).toEqual(['/public/styles.css'])
+  })
+
+  it('should handle relative paths correctly', () => {
+    const htmlContent =
+      '<html><head><link rel="stylesheet" href="./styles.css"></head></html>'
+
+    const assets = getAssetsFromHtml(htmlFilePath, htmlContent)
+
+    expect(assets?.css).toEqual(['/path/to/styles.css'])
+  })
 })