mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-01-26 19:09:52 +00:00
a630caa9d5
* Add support for custom iconify using tailwind plugin - Register svgs from custom icons folder - Update existing custom icons to remove padding - Swap component icons for classes in sidebar tabs - Update browse templates in menu to use custom icon * Add basic check for custom SVG icons * Remove unused iconify packages
58 lines
1.7 KiB
TypeScript
58 lines
1.7 KiB
TypeScript
import { expect } from '@playwright/test'
|
|
import type { Locator } from '@playwright/test'
|
|
|
|
import { comfyPageFixture as test } from '../fixtures/ComfyPage'
|
|
|
|
async function verifyCustomIconSvg(iconElement: Locator) {
|
|
const svgVariable = await iconElement.evaluate((element) => {
|
|
const styles = getComputedStyle(element)
|
|
return styles.getPropertyValue('--svg')
|
|
})
|
|
|
|
expect(svgVariable).toBeTruthy()
|
|
const dataUrlMatch = svgVariable.match(
|
|
/url\("data:image\/svg\+xml,([^"]+)"\)/
|
|
)
|
|
expect(dataUrlMatch).toBeTruthy()
|
|
|
|
const encodedSvg = dataUrlMatch![1]
|
|
const decodedSvg = decodeURIComponent(encodedSvg)
|
|
|
|
// Check for SVG header to confirm it's a valid SVG
|
|
expect(decodedSvg).toContain("<svg xmlns='http://www.w3.org/2000/svg'")
|
|
}
|
|
|
|
test.describe('Custom Icons', () => {
|
|
test.beforeEach(async ({ comfyPage }) => {
|
|
await comfyPage.setSetting('Comfy.UseNewMenu', 'Top')
|
|
})
|
|
|
|
test('sidebar tab icons use custom SVGs', async ({ comfyPage }) => {
|
|
// Find the icon in the sidebar
|
|
const icon = comfyPage.page.locator(
|
|
'.icon-\\[comfy--ai-model\\].side-bar-button-icon'
|
|
)
|
|
await expect(icon).toBeVisible()
|
|
|
|
// Verify the custom SVG content
|
|
await verifyCustomIconSvg(icon)
|
|
})
|
|
|
|
test('Browse Templates menu item uses custom template icon', async ({
|
|
comfyPage
|
|
}) => {
|
|
// Open the topbar menu
|
|
await comfyPage.menu.topbar.openTopbarMenu()
|
|
const menuItem = comfyPage.menu.topbar.getMenuItem('Browse Templates')
|
|
|
|
// Find the icon as a previous sibling of the menu item label
|
|
const templateIcon = menuItem
|
|
.locator('..')
|
|
.locator('.icon-\\[comfy--template\\]')
|
|
await expect(templateIcon).toBeVisible()
|
|
|
|
// Verify the custom SVG content
|
|
await verifyCustomIconSvg(templateIcon)
|
|
})
|
|
})
|