fix(chat): 保留页面切换后的提供商与模型选择

2026-09-04 08:24:49 +08:00
parent 9f621371b8
commit 150cf0d994
3 changed files with 83 additions and 3 deletions
@@ -0,0 +1,66 @@
+// @vitest-environment happy-dom
+import { beforeEach, expect, it, vi } from 'vitest'
+import { flushPromises, mount } from '@vue/test-utils'
+import { createPinia, setActivePinia } from 'pinia'
+import { useChatStore } from '@/stores/chat'
+import { useProviderStore } from '@/stores/provider'
+import { useSkillStore } from '@/stores/skill'
+import ChatView from './ChatView.vue'
+
+vi.mock('vue-router', () => ({ useRouter: () => ({ push: vi.fn() }) }))
+vi.mock('@/stores/editor', () => ({ useEditorStore: () => ({}) }))
+vi.mock('@/stores/workspace', () => ({ useWorkspaceStore: () => ({}) }))
+vi.mock('@/components/common/MarkdownContent.vue', () => ({ default: { template: '<div />' } }))
+
+beforeEach(() => {
+  setActivePinia(createPinia())
+  const providers = useProviderStore()
+  providers.providers = ['a', 'b'].map(id => ({
+    provider_id: id, provider_type: 'openai_compatible', name: id,
+    default_model: `${id}-default`, enabled: true, capabilities: { chat: true }, has_credential: false,
+  }))
+  providers.defaultProviderId = 'a'
+  vi.spyOn(providers, 'loadProviders').mockResolvedValue(undefined)
+  vi.spyOn(providers, 'loadModels').mockResolvedValue([])
+  vi.spyOn(useSkillStore(), 'loadSkills').mockResolvedValue(undefined)
+})
+
+it('preserves the selected provider and manual model after leaving and returning to chat', async () => {
+  const chat = useChatStore()
+  const first = mount(ChatView)
+  await flushPromises()
+  await first.get('select').setValue('b')
+  await first.get('input[list="chat-models"]').setValue('b-manual')
+  first.unmount()
+  const returned = mount(ChatView)
+  await flushPromises()
+  expect(chat.selectedProviderId).toBe('b')
+  expect(chat.selectedModel).toBe('b-manual')
+  expect(useProviderStore().loadModels).toHaveBeenLastCalledWith('b')
+  returned.unmount()
+})
+
+it.each(['missing', 'disabled', 'unselected'])('uses the default when the selected provider is %s', async state => {
+  const chat = useChatStore()
+  chat.selectedProviderId = state === 'unselected' ? '' : state === 'missing' ? 'deleted' : 'b'
+  chat.selectedModel = 'old-model'
+  if (state === 'disabled') useProviderStore().providers[1]!.enabled = false
+  const wrapper = mount(ChatView)
+  await flushPromises()
+  expect(chat.selectedProviderId).toBe('a')
+  expect(chat.selectedModel).toBe('a-default')
+  wrapper.unmount()
+})
+
+it('preserves the selection when provider discovery fails', async () => {
+  const chat = useChatStore()
+  chat.selectedProviderId = 'b'
+  chat.selectedModel = 'b-manual'
+  useProviderStore().error = 'offline'
+  const wrapper = mount(ChatView)
+  await flushPromises()
+  expect(chat.selectedProviderId).toBe('b')
+  expect(chat.selectedModel).toBe('b-manual')
+  expect(wrapper.get('.error-banner').text()).toBe('offline')
+  wrapper.unmount()
+})