diff --git a/src/composables/sidebarTabs/useAssetsSidebarTab.test.ts b/src/composables/sidebarTabs/useAssetsSidebarTab.test.ts new file mode 100644 index 000000000..0fabf6df6 --- /dev/null +++ b/src/composables/sidebarTabs/useAssetsSidebarTab.test.ts @@ -0,0 +1,46 @@ +import { describe, expect, it, vi } from 'vitest' + +import { useAssetsSidebarTab } from '@/composables/sidebarTabs/useAssetsSidebarTab' + +const { mockGetSetting, mockPendingTasks } = vi.hoisted(() => ({ + mockGetSetting: vi.fn(), + mockPendingTasks: [] as unknown[] +})) + +vi.mock('@/platform/settings/settingStore', () => ({ + useSettingStore: () => ({ + get: mockGetSetting + }) +})) + +vi.mock('@/components/sidebar/tabs/AssetsSidebarTab.vue', () => ({ + default: {} +})) + +vi.mock('@/stores/queueStore', () => ({ + useQueueStore: () => ({ + pendingTasks: mockPendingTasks + }) +})) + +describe('useAssetsSidebarTab', () => { + it('hides icon badge when QPO V2 is disabled', () => { + mockGetSetting.mockReturnValue(false) + mockPendingTasks.splice(0, mockPendingTasks.length, {}, {}) + + const sidebarTab = useAssetsSidebarTab() + + expect(typeof sidebarTab.iconBadge).toBe('function') + expect((sidebarTab.iconBadge as () => string | null)()).toBeNull() + }) + + it('shows pending task count when QPO V2 is enabled', () => { + mockGetSetting.mockReturnValue(true) + mockPendingTasks.splice(0, mockPendingTasks.length, {}, {}, {}) + + const sidebarTab = useAssetsSidebarTab() + + expect(typeof sidebarTab.iconBadge).toBe('function') + expect((sidebarTab.iconBadge as () => string | null)()).toBe('3') + }) +}) diff --git a/src/composables/sidebarTabs/useAssetsSidebarTab.ts b/src/composables/sidebarTabs/useAssetsSidebarTab.ts index cd8e8d0bb..15c1d1762 100644 --- a/src/composables/sidebarTabs/useAssetsSidebarTab.ts +++ b/src/composables/sidebarTabs/useAssetsSidebarTab.ts @@ -1,6 +1,7 @@ import { markRaw } from 'vue' import AssetsSidebarTab from '@/components/sidebar/tabs/AssetsSidebarTab.vue' +import { useSettingStore } from '@/platform/settings/settingStore' import { useQueueStore } from '@/stores/queueStore' import type { SidebarTabExtension } from '@/types/extensionTypes' @@ -14,6 +15,12 @@ export const useAssetsSidebarTab = (): SidebarTabExtension => { component: markRaw(AssetsSidebarTab), type: 'vue', iconBadge: () => { + const settingStore = useSettingStore() + + if (!settingStore.get('Comfy.Queue.QPOV2')) { + return null + } + const queueStore = useQueueStore() return queueStore.pendingTasks.length > 0 ? queueStore.pendingTasks.length.toString()