mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-03 06:47:33 +00:00
8b8f3538bf
Fixes issue where query params from https://github.com/Comfy-Org/ComfyUI_frontend/pull/6593 are stripped during the login/signup views/flow by storing initial params in session storage via router plugin. https://github.com/user-attachments/assets/51642e8c-af5c-43ef-ab7d-133bc7e511aa ┆Issue is synchronized with this [Notion page](https://www.notion.so/PR-6677-fix-template-query-param-stripped-during-login-views-2aa6d73d365081a1bdc7d22b35f72a77) by [Unito](https://www.unito.io)
74 lines
2.2 KiB
TypeScript
74 lines
2.2 KiB
TypeScript
import { beforeEach, describe, expect, it } from 'vitest'
|
|
|
|
import {
|
|
capturePreservedQuery,
|
|
clearPreservedQuery,
|
|
hydratePreservedQuery,
|
|
mergePreservedQueryIntoQuery
|
|
} from '@/platform/navigation/preservedQueryManager'
|
|
import { PRESERVED_QUERY_NAMESPACES } from '@/platform/navigation/preservedQueryNamespaces'
|
|
|
|
const NAMESPACE = PRESERVED_QUERY_NAMESPACES.TEMPLATE
|
|
|
|
describe('preservedQueryManager', () => {
|
|
beforeEach(() => {
|
|
sessionStorage.clear()
|
|
clearPreservedQuery(NAMESPACE)
|
|
})
|
|
|
|
it('captures specified keys from the route query', () => {
|
|
capturePreservedQuery(NAMESPACE, { template: 'flux', source: 'custom' }, [
|
|
'template',
|
|
'source'
|
|
])
|
|
|
|
hydratePreservedQuery(NAMESPACE)
|
|
const merged = mergePreservedQueryIntoQuery(NAMESPACE)
|
|
|
|
expect(merged).toEqual({ template: 'flux', source: 'custom' })
|
|
expect(sessionStorage.getItem('Comfy.PreservedQuery.template')).toBeTruthy()
|
|
})
|
|
|
|
it('hydrates cached payload from sessionStorage once', () => {
|
|
sessionStorage.setItem(
|
|
'Comfy.PreservedQuery.template',
|
|
JSON.stringify({ template: 'flux', source: 'default' })
|
|
)
|
|
|
|
hydratePreservedQuery(NAMESPACE)
|
|
const merged = mergePreservedQueryIntoQuery(NAMESPACE)
|
|
|
|
expect(merged).toEqual({ template: 'flux', source: 'default' })
|
|
})
|
|
|
|
it('merges stored payload only when query lacks the keys', () => {
|
|
capturePreservedQuery(NAMESPACE, { template: 'flux' }, ['template'])
|
|
|
|
const merged = mergePreservedQueryIntoQuery(NAMESPACE, {
|
|
foo: 'bar'
|
|
})
|
|
|
|
expect(merged).toEqual({ foo: 'bar', template: 'flux' })
|
|
})
|
|
|
|
it('returns undefined when merge does not change query', () => {
|
|
capturePreservedQuery(NAMESPACE, { template: 'flux' }, ['template'])
|
|
|
|
const merged = mergePreservedQueryIntoQuery(NAMESPACE, {
|
|
template: 'existing'
|
|
})
|
|
|
|
expect(merged).toBeUndefined()
|
|
})
|
|
|
|
it('clears cached payload', () => {
|
|
capturePreservedQuery(NAMESPACE, { template: 'flux' }, ['template'])
|
|
|
|
clearPreservedQuery(NAMESPACE)
|
|
|
|
const merged = mergePreservedQueryIntoQuery(NAMESPACE)
|
|
expect(merged).toBeUndefined()
|
|
expect(sessionStorage.getItem('Comfy.PreservedQuery.template')).toBeNull()
|
|
})
|
|
})
|