mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-04-28 18:22:40 +00:00
[Bug] Fix this binding in useChainCallback (#3252)
This commit is contained in:
@@ -9,8 +9,8 @@ export const useChainCallback = <T extends (...args: any[]) => void>(
|
|||||||
originalCallback: T | undefined,
|
originalCallback: T | undefined,
|
||||||
...callbacks: ((...args: Parameters<T>) => void)[]
|
...callbacks: ((...args: Parameters<T>) => void)[]
|
||||||
) => {
|
) => {
|
||||||
return (...args: Parameters<T>) => {
|
return function (this: unknown, ...args: Parameters<T>) {
|
||||||
originalCallback?.(...args)
|
originalCallback?.call(this, ...args)
|
||||||
callbacks.forEach((callback) => callback(...args))
|
callbacks.forEach((callback) => callback.call(this, ...args))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,55 @@
|
|||||||
|
import { describe, expect, it } from 'vitest'
|
||||||
|
|
||||||
|
import { useChainCallback } from '@/composables/functional/useChainCallback'
|
||||||
|
|
||||||
|
describe('useChainCallback', () => {
|
||||||
|
it('preserves "this" context in original callback and chained callbacks', () => {
|
||||||
|
class TestClass {
|
||||||
|
value = 'test'
|
||||||
|
|
||||||
|
constructor() {
|
||||||
|
this.method = useChainCallback(this.method, function (this: TestClass) {
|
||||||
|
expect(this.value).toBe('test')
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
method() {
|
||||||
|
expect(this.value).toBe('test')
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const instance = new TestClass()
|
||||||
|
instance.method()
|
||||||
|
})
|
||||||
|
|
||||||
|
it('handles undefined original callback', () => {
|
||||||
|
const context = { value: 'test' }
|
||||||
|
const chainedFn = useChainCallback(
|
||||||
|
undefined,
|
||||||
|
function (this: typeof context) {
|
||||||
|
expect(this.value).toBe('test')
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
chainedFn.call(context)
|
||||||
|
})
|
||||||
|
|
||||||
|
it('passes arguments to all callbacks', () => {
|
||||||
|
const originalCalls: number[] = []
|
||||||
|
const chainedCalls: number[] = []
|
||||||
|
|
||||||
|
const original = function (this: unknown, num: number) {
|
||||||
|
originalCalls.push(num)
|
||||||
|
}
|
||||||
|
|
||||||
|
const chained = function (this: unknown, num: number) {
|
||||||
|
chainedCalls.push(num)
|
||||||
|
}
|
||||||
|
|
||||||
|
const chainedFn = useChainCallback(original, chained)
|
||||||
|
chainedFn(42)
|
||||||
|
|
||||||
|
expect(originalCalls).toEqual([42])
|
||||||
|
expect(chainedCalls).toEqual([42])
|
||||||
|
})
|
||||||
|
})
|
||||||
Reference in New Issue
Block a user