Remove legacy slow jest tests (#2092)

This commit is contained in:
Chenlei Hu
2024-12-29 16:25:06 -05:00
committed by GitHub
parent 95a4623c37
commit 642094ead1
26 changed files with 4 additions and 2127 deletions

View File

@@ -0,0 +1,63 @@
import { getNodeSource, NodeSourceType } from '@/types/nodeSource'
describe('getNodeSource', () => {
it('should return UNKNOWN_NODE_SOURCE when python_module is undefined', () => {
const result = getNodeSource(undefined)
expect(result).toEqual({
type: NodeSourceType.Unknown,
className: 'comfy-unknown',
displayText: 'Unknown',
badgeText: '?'
})
})
it('should identify core nodes from nodes module', () => {
const result = getNodeSource('nodes.some_module')
expect(result).toEqual({
type: NodeSourceType.Core,
className: 'comfy-core',
displayText: 'Comfy Core',
badgeText: '🦊'
})
})
it('should identify core nodes from comfy_extras module', () => {
const result = getNodeSource('comfy_extras.some_module')
expect(result).toEqual({
type: NodeSourceType.Core,
className: 'comfy-core',
displayText: 'Comfy Core',
badgeText: '🦊'
})
})
it('should identify custom nodes and format their names', () => {
const result = getNodeSource('custom_nodes.ComfyUI-Example')
expect(result).toEqual({
type: NodeSourceType.CustomNodes,
className: 'comfy-custom-nodes',
displayText: 'Example',
badgeText: 'Example'
})
})
it('should identify custom nodes with version and format their names', () => {
const result = getNodeSource('custom_nodes.ComfyUI-Example@1.0.0')
expect(result).toEqual({
type: NodeSourceType.CustomNodes,
className: 'comfy-custom-nodes',
displayText: 'Example',
badgeText: 'Example'
})
})
it('should return UNKNOWN_NODE_SOURCE for unrecognized modules', () => {
const result = getNodeSource('unknown_module.something')
expect(result).toEqual({
type: NodeSourceType.Unknown,
className: 'comfy-unknown',
displayText: 'Unknown',
badgeText: '?'
})
})
})