opsxshell/test/stores/tabStore.test.ts

/**
 * TabStore Tests - 标签页状态管理测试
 */

import { describe, it, expect, beforeEach, afterEach } from 'vitest'
import { useTabStore } from '../../src/stores/tabStore'

describe('TabStore', () => {
  beforeEach(() => {
    // Reset store to initial state
    useTabStore.setState({
      tabs: [{ id: 'welcome', title: '欢迎', type: 'welcome', isActive: true }],
      activeTabId: 'welcome',
    })
  })

  describe('初始状态', () => {
    it('应该包含默认的欢迎标签', () => {
      const { tabs, activeTabId } = useTabStore.getState()
      expect(tabs).toHaveLength(1)
      expect(tabs[0].id).toBe('welcome')
      expect(tabs[0].type).toBe('welcome')
      expect(activeTabId).toBe('welcome')
    })
  })

  describe('addTab', () => {
    it('应该添加新的终端标签', () => {
      const { addTab, tabs } = useTabStore.getState()
      addTab({ title: 'Terminal', type: 'terminal' })

      const newTabs = useTabStore.getState().tabs
      expect(newTabs).toHaveLength(tabs.length + 1)
      expect(newTabs.find((t) => t.type === 'terminal')).toBeDefined()
    })

    it('添加新标签后应该激活它', () => {
      const { addTab } = useTabStore.getState()
      addTab({ title: 'Terminal', type: 'terminal' })

      const { activeTabId, tabs } = useTabStore.getState()
      const activeTab = tabs.find((t) => t.id === activeTabId)
      expect(activeTab?.type).toBe('terminal')
    })

    it('添加 settings 类型时如果已存在则激活而不是创建新的', () => {
      const { addTab } = useTabStore.getState()
      addTab({ title: 'Settings', type: 'settings' })
      addTab({ title: 'Settings 2', type: 'settings' })

      const newTabs = useTabStore.getState().tabs
      expect(newTabs).toHaveLength(2) // welcome + settings
    })
  })

  describe('updateTab', () => {
    it('应该更新指定标签的标题', () => {
      const { updateTab } = useTabStore.getState()
      updateTab('welcome', { title: '新的标题' })

      const { tabs } = useTabStore.getState()
      expect(tabs[0].title).toBe('新的标题')
    })

    it('应该更新标签的连接状态', () => {
      const { updateTab } = useTabStore.getState()
      updateTab('welcome', { connectionStatus: 'connected' })

      const { tabs } = useTabStore.getState()
      expect(tabs[0].connectionStatus).toBe('connected')
    })
  })

  describe('closeTab', () => {
    it('不应该关闭最后一个欢迎页', () => {
      const { closeTab } = useTabStore.getState()
      const closed = closeTab('welcome')

      const { tabs } = useTabStore.getState()
      expect(tabs).toHaveLength(1)
      expect(tabs[0].type).toBe('welcome')
      expect(closed).toBeNull()
    })

    it('关闭最后一个会话标签应回到欢迎页', () => {
      const { addTab, closeOtherTabs, closeTab } = useTabStore.getState()
      addTab({ title: 'Tab 1', type: 'terminal' })
      const session = useTabStore.getState().tabs.find((t) => t.title === 'Tab 1')
      expect(session).toBeTruthy()
      closeOtherTabs(session!.id)

      expect(useTabStore.getState().tabs).toHaveLength(1)
      const closed = closeTab(session!.id)
      const { tabs, activeTabId } = useTabStore.getState()
      expect(closed?.title).toBe('Tab 1')
      expect(tabs).toHaveLength(1)
      expect(tabs[0].type).toBe('welcome')
      expect(activeTabId).toBe('welcome')
    })

    it('关闭标签后应该激活另一个标签', () => {
      const { addTab, closeTab } = useTabStore.getState()
      addTab({ title: 'Tab 1', type: 'terminal' })
      addTab({ title: 'Tab 2', type: 'sftp' })

      const { tabs } = useTabStore.getState()
      const tabToClose = tabs.find((t) => t.title === 'Tab 1')

      if (tabToClose) {
        closeTab(tabToClose.id)
        const { tabs: newTabs } = useTabStore.getState()
        expect(newTabs.find((t) => t.isActive)?.title).toBe('Tab 2')
      }
    })

    it('应该返回被关闭的标签', () => {
      const { addTab, closeTab } = useTabStore.getState()
      addTab({ title: 'Tab 1', type: 'terminal' })

      const { tabs } = useTabStore.getState()
      const tabToClose = tabs.find((t) => t.title === 'Tab 1')

      if (tabToClose) {
        const closed = closeTab(tabToClose.id)
        expect(closed?.title).toBe('Tab 1')
      }
    })
  })

  describe('closeOtherTabs', () => {
    it('应该关闭除指定标签外的所有标签', () => {
      const { addTab, closeOtherTabs } = useTabStore.getState()
      addTab({ title: 'Tab 1', type: 'terminal' })
      addTab({ title: 'Tab 2', type: 'sftp' })

      const tabToKeep = useTabStore.getState().tabs.find((t) => t.title === 'Tab 1')
      if (tabToKeep) {
        closeOtherTabs(tabToKeep.id)
        const { tabs: newTabs } = useTabStore.getState()
        expect(newTabs).toHaveLength(1)
        expect(newTabs[0].title).toBe('Tab 1')
      }
    })
  })

  describe('activateTab', () => {
    it('应该激活指定的标签', () => {
      const { addTab, activateTab } = useTabStore.getState()
      addTab({ title: 'Tab 1', type: 'terminal' })
      addTab({ title: 'Tab 2', type: 'sftp' })

      const { tabs } = useTabStore.getState()
      const tabToActivate = tabs.find((t) => t.title === 'Tab 1')

      if (tabToActivate) {
        activateTab(tabToActivate.id)
        const { tabs: newTabs, activeTabId } = useTabStore.getState()
        const activeTab = newTabs.find((t) => t.id === activeTabId)
        expect(activeTab?.title).toBe('Tab 1')
      }
    })
  })

  describe('getTabById', () => {
    it('应该返回指定 ID 的标签', () => {
      const { addTab, getTabById } = useTabStore.getState()
      addTab({ title: 'Test Tab', type: 'terminal' })

      const { tabs } = useTabStore.getState()
      const testTab = tabs.find((t) => t.title === 'Test Tab')

      if (testTab) {
        const found = getTabById(testTab.id)
        expect(found?.title).toBe('Test Tab')
      }
    })

    it('找不到时应该返回 undefined', () => {
      const { getTabById } = useTabStore.getState()
      const found = getTabById('non-existent-id')
      expect(found).toBeUndefined()
    })
  })

  describe('updateTabTitle', () => {
    it('应该更新标签标题', () => {
      const { updateTabTitle } = useTabStore.getState()
      updateTabTitle('welcome', 'Updated Welcome')

      const { tabs } = useTabStore.getState()
      expect(tabs[0].title).toBe('Updated Welcome')
    })
  })

  describe('持久化恢复', () => {
    afterEach(() => {
      localStorage.removeItem('opsxshell-tabs')
    })

    const seedTabs = (tabs: unknown[]) => {
      localStorage.setItem('opsxshell-tabs', JSON.stringify({ state: { tabs, activeTabId: null }, version: 0 }))
    }

    it('SSH 标签恢复后仍需手动重连', async () => {
      seedTabs([{ id: 'ssh-1', title: 'user@host', type: 'terminal', sessionId: 's1' }])

      await useTabStore.persist.rehydrate()

      const tab = useTabStore.getState().tabs.find((t) => t.id === 'ssh-1')
      expect(tab?.needsReconnect).toBe(true)
    })

    it('本地终端标签恢复后直接拉起 Shell，不弹重连横幅', async () => {
      seedTabs([{ id: 'local-1', title: 'Bash', type: 'terminal', isLocal: true, localShellPath: '/bin/bash' }])

      await useTabStore.persist.rehydrate()

      const tab = useTabStore.getState().tabs.find((t) => t.id === 'local-1')
      expect(tab?.needsReconnect).toBe(false)
      expect(tab?.localShellPath).toBe('/bin/bash')
    })
  })
})