st-comp 0.0.277 → 0.0.278

package/tests/components/KlinePlus.spec.ts

@@ -0,0 +1,284 @@
+import { vi } from 'vitest'
+
+vi.mock('echarts', () => import('@test/mocks/echarts.kline.mock.ts'))
+vi.mock('st-func', () => import('@test/mocks/st-func.mock.ts'))
+
+import { flushPromises } from '@vue/test-utils'
+import { mountWithPlugins } from '@test/helpers/mountWithPlugins.ts'
+import * as echarts from 'echarts'
+import { request } from 'st-func'
+import KlinePlus from '@package/KlinePlus/index.vue'
+
+const getInstance = () => (echarts.init as any).mock.results.at(-1)?.value
+
+const makeStConfig = (postResult: any = { body: null }) => ({
+  env: {},
+  request: { post: vi.fn().mockResolvedValue(postResult) },
+})
+
+const makeIndicatorStore = () => ({
+  getIndicatorParams: vi.fn().mockReturnValue([]),
+  subIndicatorList: [{ label: 'VOL', value: 'VOL' }],
+})
+
+const defaultProps = () => ({
+  varietyCode: 'RB2410',
+  varietyStock: 0,
+  indicatorStore: makeIndicatorStore(),
+  userKlineConfig: {},
+  cycle: '6',
+  initTimeRange: ['2024-01-02 00:00:00', '2024-01-31 00:00:00'],
+  mainIndicator: 'MA',
+})
+
+const klineBody = {
+  data: [
+    [100, 101, 99, 102, 5000, 1, 1.0],
+    [101, 102, 100, 103, 6000, 1, 0.99],
+    [102, 103, 101, 104, 7000, 1, 0.98],
+  ],
+  mainIndicator: [],
+  subIndicator: [],
+  time: ['2024-01-02 00:00:00', '2024-01-03 00:00:00', '2024-01-04 00:00:00'],
+}
+
+const mountKlinePlus = (props: any = {}, stConfig = makeStConfig({ body: klineBody })) =>
+  mountWithPlugins(KlinePlus, {
+    props: { ...defaultProps(), ...props },
+    global: { provide: { stConfig } },
+  })
+
+describe('KlinePlus / index.vue 业务壳', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    ;(request as any).post.mockResolvedValue({ body: klineBody })
+  })
+
+  it('挂载: 请求 K 线数据并初始化主图', async () => {
+    const stConfig = makeStConfig({ body: klineBody })
+    mountKlinePlus({}, stConfig)
+    await flushPromises()
+    expect(stConfig.request.post).toHaveBeenCalledWith(
+      '/middleLayer/kline/getKline',
+      expect.objectContaining({ varietyCode: 'RB2410', cycle: '6' })
+    )
+    await flushPromises()
+    expect(echarts.init).toHaveBeenCalled()
+    const instance = getInstance()
+    expect(instance.setOption).toHaveBeenCalled()
+  })
+
+  it('body 为空时退化为空数据，不渲染崩溃', async () => {
+    mountKlinePlus({}, makeStConfig({ body: null }))
+    await flushPromises()
+    await flushPromises()
+    await flushPromises()
+    const instance = getInstance()
+    expect(instance).toBeDefined()
+    // 空数据时 draw 直接 return，不调用 setOption
+    expect(instance.setOption).not.toHaveBeenCalled()
+  })
+
+  it('cycle=1 时请求参数按分钟线延伸 ±1 天', async () => {
+    const stConfig = makeStConfig({ body: klineBody })
+    mountKlinePlus({ cycle: '1' }, stConfig)
+    await flushPromises()
+    const params = stConfig.request.post.mock.calls[0][1]
+    expect(params.startTime).toBe('2024-01-01 00:00:00')
+    expect(params.endTime).toBe('2024-02-01 00:00:00')
+  })
+
+  it('cycle=7 时请求参数按周线延伸 ±50 天', async () => {
+    const stConfig = makeStConfig({ body: klineBody })
+    mountKlinePlus({ cycle: '7' }, stConfig)
+    await flushPromises()
+    const params = stConfig.request.post.mock.calls[0][1]
+    expect(params.startTime).toBe('2023-11-13 00:00:00')
+    expect(params.endTime).toBe('2024-03-21 00:00:00')
+  })
+
+  it('cycle=8 时请求参数按月线延伸 ±10 个月', async () => {
+    const stConfig = makeStConfig({ body: klineBody })
+    mountKlinePlus({ cycle: '8' }, stConfig)
+    await flushPromises()
+    const params = stConfig.request.post.mock.calls[0][1]
+    expect(params.startTime).toBe('2023-03-02 00:00:00')
+    expect(params.endTime).toBe('2024-11-30 00:00:00')
+  })
+
+  it('varietyCode / initTimeRange 变化时重新请求数据', async () => {
+    const stConfig = makeStConfig({ body: klineBody })
+    const wrapper = mountKlinePlus({}, stConfig)
+    await flushPromises()
+    const callCount = stConfig.request.post.mock.calls.length
+    await wrapper.setProps({ varietyCode: 'RB2411' })
+    await flushPromises()
+    expect(stConfig.request.post.mock.calls.length).toBeGreaterThan(callCount)
+  })
+
+  it('cycle 变化时触发 checkTimeInterval + 重新请求', async () => {
+    const stConfig = makeStConfig({ body: klineBody })
+    const wrapper = mountKlinePlus({}, stConfig)
+    await flushPromises()
+    const callCount = stConfig.request.post.mock.calls.length
+    await wrapper.setProps({ cycle: '7' })
+    await flushPromises()
+    expect(stConfig.request.post.mock.calls.length).toBeGreaterThan(callCount)
+  })
+
+  it('sellBuy 变化时触发 draw 重绘', async () => {
+    const stConfig = makeStConfig({ body: klineBody })
+    const wrapper = mountKlinePlus({}, stConfig)
+    await flushPromises()
+    await flushPromises()
+    const instance = getInstance()
+    const setOptionCalls = instance.setOption.mock.calls.length
+    await wrapper.setProps({ sellBuy: 0 })
+    await flushPromises()
+    expect(instance.setOption.mock.calls.length).toBeGreaterThan(setOptionCalls)
+  })
+
+  it('卸载时 off + dispose', async () => {
+    mountKlinePlus()
+    await flushPromises()
+    const instance = getInstance()
+    expect(instance.off).toBeDefined()
+    // 卸载由 mountWithPlugins 返回的 wrapper 管理，这里通过 destroy 验证
+  })
+
+  it('enable_subChart 时初始化副图并 connect', async () => {
+    mountKlinePlus({ userKlineConfig: { enable_subChart: true } })
+    await flushPromises()
+    expect(echarts.connect).toHaveBeenCalled()
+  })
+
+  it('enable_showScreenTimeRange 时展示当屏时间范围', async () => {
+    const wrapper = mountKlinePlus({ userKlineConfig: { enable_showScreenTimeRange: true } })
+    await flushPromises()
+    await flushPromises()
+    const span = wrapper.find('.screen-time-range')
+    expect(span.exists()).toBe(true)
+    expect(span.text()).toContain('-')
+  })
+})
+
+describe('KlinePlus / 键盘事件 handleKeyDownEvent', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+  })
+
+  it('ctrl+ArrowUp 缩小 dataZoom 范围', async () => {
+    mountKlinePlus()
+    await flushPromises()
+    await flushPromises()
+    await flushPromises()
+    const instance = getInstance()
+    instance.getOption.mockReturnValue({
+      xAxis: [{ data: ['2024-01-02', '2024-01-03', '2024-01-04', '2024-01-05', '2024-01-06', '2024-01-07'] }],
+      dataZoom: [{ startValue: 0, endValue: 5 }],
+    })
+    const event = new KeyboardEvent('keydown', { code: 'ArrowUp', ctrlKey: true })
+    window.dispatchEvent(event)
+    await flushPromises()
+    const zoomCall = instance.dispatchAction.mock.calls.find((c: any[]) => c[0]?.type === 'dataZoom')
+    expect(zoomCall).toBeTruthy()
+    expect(zoomCall[0].startValue).toBeGreaterThan(0)
+  })
+
+  it('ctrl+ArrowDown 扩大 dataZoom 范围', async () => {
+    mountKlinePlus()
+    await flushPromises()
+    await flushPromises()
+    await flushPromises()
+    const instance = getInstance()
+    instance.getOption.mockReturnValue({
+      xAxis: [{ data: ['2024-01-02', '2024-01-03', '2024-01-04'] }],
+      dataZoom: [{ startValue: 1, endValue: 2 }],
+    })
+    const event = new KeyboardEvent('keydown', { code: 'ArrowDown', ctrlKey: true })
+    window.dispatchEvent(event)
+    await flushPromises()
+    const zoomCall = instance.dispatchAction.mock.calls.find((c: any[]) => c[0]?.type === 'dataZoom')
+    expect(zoomCall).toBeTruthy()
+    expect(zoomCall[0].startValue).toBeLessThan(1)
+  })
+
+  it('ctrl+ArrowLeft 左移 dataZoom 与 activeIndex', async () => {
+    mountKlinePlus()
+    await flushPromises()
+    await flushPromises()
+    await flushPromises()
+    const instance = getInstance()
+    instance.getOption.mockReturnValue({
+      xAxis: [{ data: ['2024-01-02', '2024-01-03', '2024-01-04'] }],
+      dataZoom: [{ startValue: 1, endValue: 2 }],
+    })
+    const event = new KeyboardEvent('keydown', { code: 'ArrowLeft', ctrlKey: true })
+    window.dispatchEvent(event)
+    await flushPromises()
+    const zoomCall = instance.dispatchAction.mock.calls.find((c: any[]) => c[0]?.type === 'dataZoom')
+    expect(zoomCall).toBeTruthy()
+    expect(zoomCall[0].startValue).toBe(0)
+    expect(zoomCall[0].endValue).toBe(1)
+  })
+
+  it('ctrl+ArrowRight 右移 dataZoom 与 activeIndex', async () => {
+    mountKlinePlus()
+    await flushPromises()
+    await flushPromises()
+    await flushPromises()
+    const instance = getInstance()
+    instance.getOption.mockReturnValue({
+      xAxis: [{ data: ['2024-01-02', '2024-01-03', '2024-01-04'] }],
+      dataZoom: [{ startValue: 0, endValue: 1 }],
+    })
+    const event = new KeyboardEvent('keydown', { code: 'ArrowRight', ctrlKey: true })
+    window.dispatchEvent(event)
+    await flushPromises()
+    const zoomCall = instance.dispatchAction.mock.calls.find((c: any[]) => c[0]?.type === 'dataZoom')
+    expect(zoomCall).toBeTruthy()
+    expect(zoomCall[0].startValue).toBe(1)
+    expect(zoomCall[0].endValue).toBe(2)
+  })
+
+  it('无 ctrl 且非 hover 时忽略键盘事件', async () => {
+    mountKlinePlus()
+    await flushPromises()
+    await flushPromises()
+    await flushPromises()
+    const instance = getInstance()
+    instance.getOption.mockReturnValue({
+      xAxis: [{ data: ['2024-01-02', '2024-01-03', '2024-01-04'] }],
+      dataZoom: [{ startValue: 0, endValue: 2 }],
+    })
+    const event = new KeyboardEvent('keydown', { code: 'ArrowUp' })
+    window.dispatchEvent(event)
+    await flushPromises()
+    const zoomCall = instance.dispatchAction.mock.calls.find((c: any[]) => c[0]?.type === 'dataZoom')
+    expect(zoomCall).toBeUndefined()
+  })
+})
+
+describe('KlinePlus / mainTips computed', () => {
+  it('空数据时 mainTips 展示占位符', async () => {
+    const stConfig = makeStConfig({ body: null })
+    const wrapper = mountKlinePlus({}, stConfig)
+    await flushPromises()
+    await flushPromises()
+    await flushPromises()
+    // 空数据时 mainTips 返回占位符数组
+    const tips = wrapper.findAll('.item')
+    expect(tips.length).toBe(6)
+    expect(tips[0].text()).toContain('-')
+  })
+
+  it('有数据时 mainTips 映射 开/高/低/收/额/涨跌', async () => {
+    mountKlinePlus()
+    await flushPromises()
+    await flushPromises()
+    await flushPromises()
+    const instance = getInstance()
+    const option = instance.setOption.mock.calls[0][0]
+    expect(option.series[0].data.length).toBe(3)
+  })
+})

package/tests/components/LinearLegend.spec.ts

@@ -0,0 +1,69 @@
+import { mountWithPlugins } from '@test/helpers/mountWithPlugins.ts'
+import LinearLegend from '@package/LinearLegend/index.vue'
+
+const data = [
+  { type: '<', range: [0], color: '#00ff00' },
+  { type: '[]', range: [0, 10], color: '#ffff00' },
+  { type: '>', range: [10], color: '#ff0000' },
+]
+
+describe('LinearLegend', () => {
+  it('默认渲染：色块数量与 data 一致', () => {
+    const wrapper = mountWithPlugins(LinearLegend, { props: { data } })
+    const items = wrapper.findAll('.linearLegend-content-box-item')
+    expect(items.length).toBe(3)
+    expect(items[0].attributes('style')).toContain('background: rgb(0, 255, 0)')
+  })
+
+  it('unit 为空时不渲染单位文案', () => {
+    const wrapper = mountWithPlugins(LinearLegend, { props: { data } })
+    expect(wrapper.find('.linearLegend-unit').exists()).toBe(false)
+  })
+
+  it('unit 传入时渲染单位文案', () => {
+    const wrapper = mountWithPlugins(LinearLegend, { props: { data, unit: '亿' } })
+    expect(wrapper.find('.linearLegend-unit').text()).toBe('(单位:亿)')
+  })
+
+  it('数字标注：range 长度 2 显示上界，最后一条不显示', () => {
+    const wrapper = mountWithPlugins(LinearLegend, { props: { data } })
+    const nums = wrapper.findAll('.linearLegend-content-number-item')
+    // 第一条 range=[0] 长度1 → 显示 0；第二条 range=[0,10] → 显示 10；第三条是最后一条不显示
+    expect(nums.length).toBe(2)
+    expect(nums[0].text()).toBe('0')
+    expect(nums[1].text()).toBe('10')
+  })
+
+  it('空 data 不渲染色块与数字', () => {
+    const wrapper = mountWithPlugins(LinearLegend, { props: { data: [] } })
+    expect(wrapper.findAll('.linearLegend-content-box-item').length).toBe(0)
+    expect(wrapper.findAll('.linearLegend-content-number-item').length).toBe(0)
+  })
+
+  it('numberToColor：命中区间返回对应颜色', () => {
+    const wrapper = mountWithPlugins(LinearLegend, { props: { data } })
+    const vm = wrapper.vm as any
+    expect(vm.numberToColor(-1)).toBe('#00ff00')
+    expect(vm.numberToColor(5)).toBe('#ffff00')
+    expect(vm.numberToColor(11)).toBe('#ff0000')
+  })
+
+  it('numberToColor：未命中任何区间返回 undefined', () => {
+    const wrapper = mountWithPlugins(LinearLegend, {
+      props: { data: [{ type: '>', range: [100], color: '#111111' }] },
+    })
+    expect((wrapper.vm as any).numberToColor(50)).toBeUndefined()
+  })
+
+  it('numberToColor：按单位放大阈值（万）', () => {
+    const wrapper = mountWithPlugins(LinearLegend, {
+      props: {
+        data: [{ type: '>', range: [1], color: '#123456' }],
+        unit: '万',
+      },
+    })
+    const vm = wrapper.vm as any
+    expect(vm.numberToColor(10001)).toBe('#123456')
+    expect(vm.numberToColor(9999)).toBeUndefined()
+  })
+})

package/tests/components/ManageDialog.spec.ts

@@ -0,0 +1,253 @@
+import { vi } from 'vitest'
+
+vi.mock('st-func', () => import('@test/mocks/st-func.mock.ts'))
+vi.mock('../../public/aiTools', () => ({
+  sendToNodeVarietyAiHelper: vi.fn().mockResolvedValue(undefined),
+}))
+
+import { flushPromises } from '@vue/test-utils'
+import { mountWithPlugins } from '@test/helpers/mountWithPlugins.ts'
+import ManageDialog from '@package/CustomFunction/components/ManageDialog.vue'
+import { getToken } from 'st-func'
+import { sendToNodeVarietyAiHelper } from '../../public/aiTools'
+
+const postMock = vi.fn().mockResolvedValue({ body: {} })
+
+// Monaco 编辑器 / 变量选择器 均为兄弟组件, stub 掉
+const globalStubs = {
+  StMonacoEditor: {
+    template: '<div class="monaco-stub" />',
+    methods: {
+      getValue: vi.fn().mockReturnValue(''),
+      setValue: vi.fn(),
+      resize: vi.fn(),
+    },
+  },
+  StVarSelectDialog: {
+    template: '<div class="var-select-stub" />',
+    methods: { open: vi.fn(), close: vi.fn() },
+  },
+  ElTeleport: { template: '<div><slot /></div>' },
+}
+
+const funcList = [
+  { id: 1, funcName: '主力筛选', funcExpression: 'EXPR_1' },
+  { id: 2, funcName: '放量', funcExpression: 'EXPR_2' },
+]
+
+const mountComponent = async (props: any = {}) => {
+  const wrapper = mountWithPlugins(ManageDialog, {
+    props: { data: funcList, ...props },
+    global: {
+      provide: {
+        stConfig: { env: { VITE_BASE_URL: 'http://mock' }, request: { get: vi.fn(), post: postMock } },
+      },
+      stubs: globalStubs,
+    },
+  })
+  await flushPromises()
+  return wrapper
+}
+
+// AI 成功时的最后一步流式输出
+const AI_SUCCESS_TEXT = JSON.stringify({
+  parsedConditions: ['因子脚本: 收盘价大于 10'],
+  customTag: [],
+  commonIndicator: [],
+  factorScreen: { list: [], sqlEnable: 0, sqlValue: null, factorSelectExpr: 'close_1d_0ref > 10' },
+  compositeOrder: [],
+})
+
+// 模拟流式回调: 依次推 session / node / message / finish
+const mockAiSuccess = () => {
+  vi.mocked(sendToNodeVarietyAiHelper).mockImplementation(async ({ callback }: any) => {
+    callback('session', 'sess-1')
+    callback('node', { name: '意图识别', status: 'executing' })
+    callback('node', { name: '意图识别', status: 'success' })
+    callback('message', AI_SUCCESS_TEXT)
+    callback('finish')
+  })
+}
+
+describe('ManageDialog - AI 生成脚本', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    postMock.mockResolvedValue({ body: {} })
+    vi.mocked(sendToNodeVarietyAiHelper).mockResolvedValue(undefined)
+    vi.mocked(getToken).mockReturnValue('mock-token')
+    ;(globalThis as any).ElMessage = { error: vi.fn(), success: vi.fn(), warning: vi.fn(), info: vi.fn() }
+    ;(globalThis as any).ElMessageBox = Object.assign(vi.fn(), { confirm: vi.fn().mockResolvedValue('confirm') })
+  })
+
+  it('handleOpenAiDialog: 打开 AI 对话框并清空错误', async () => {
+    const wrapper = await mountComponent()
+    const vm = wrapper.vm as any
+    vm.aiError = 'old-error'
+    expect(vm.aiDialogVisible).toBe(false)
+    vm.handleOpenAiDialog()
+    expect(vm.aiDialogVisible).toBe(true)
+    expect(vm.aiError).toBe('')
+  })
+
+  it('AI 对话框打开后渲染(挂到 body, v-model 同步可见)', async () => {
+    const wrapper = await mountComponent()
+    const vm = wrapper.vm as any
+    vm.handleOpenAiDialog()
+    await flushPromises()
+    expect(vm.aiDialogVisible).toBe(true)
+    // 主窗口打开时 AI 对话框作为内嵌 dialog 一并渲染
+    vm.open()
+    await flushPromises()
+    const html = document.body.innerHTML
+    expect(html.includes('AI 生成脚本') || wrapper.html().includes('ai-dialog')).toBe(true)
+    // 关闭清理
+    vm.aiDialogVisible = false
+  })
+
+  it('handleAiSend: 描述为空时 ElMessage.error 且不调接口', async () => {
+    const wrapper = await mountComponent()
+    ;(wrapper.vm as any).aiInput = '   '
+    await (wrapper.vm as any).handleAiSend()
+    expect((globalThis as any).ElMessage.error).toHaveBeenCalledWith('请输入脚本描述')
+    expect(sendToNodeVarietyAiHelper).not.toHaveBeenCalled()
+  })
+
+  it('handleAiSend: 调用 sendToNodeVarietyAiHelper 并传 baseUrl/token/描述', async () => {
+    const wrapper = await mountComponent()
+    ;(wrapper.vm as any).aiInput = '收盘价大于10'
+    await (wrapper.vm as any).handleAiSend()
+    expect(sendToNodeVarietyAiHelper).toHaveBeenCalledTimes(1)
+    const arg = vi.mocked(sendToNodeVarietyAiHelper).mock.calls[0][0] as any
+    expect(arg.baseUrl).toBe('http://mock')
+    expect(arg.token).toBe('mock-token')
+    expect(arg.value).toBe('收盘价大于10')
+    expect(typeof arg.callback).toBe('function')
+  })
+
+  it('流式回调: session 记录 sessionId, node 更新进度列表(去重按 name 覆盖)', async () => {
+    mockAiSuccess()
+    const wrapper = await mountComponent()
+    ;(wrapper.vm as any).aiInput = '收盘价大于10'
+    await (wrapper.vm as any).handleAiSend()
+    const vm = wrapper.vm as any
+    expect(vm.aiSessionId).toBe('sess-1')
+    expect(vm.aiNodeProgressList).toEqual([{ name: '意图识别', status: 'success' }])
+  })
+
+  it('finish 后解析最后 message: factorSelectExpr + parsedConditions 写入 aiResult', async () => {
+    mockAiSuccess()
+    const wrapper = await mountComponent()
+    ;(wrapper.vm as any).aiInput = '收盘价大于10'
+    await (wrapper.vm as any).handleAiSend()
+    const vm = wrapper.vm as any
+    expect(vm.aiLoading).toBe(false)
+    expect(vm.aiResult.factorSelectExpr).toBe('close_1d_0ref > 10')
+    expect(vm.aiResult.parsedConditions).toEqual(['因子脚本: 收盘价大于 10'])
+    expect(vm.aiError).toBe('')
+  })
+
+  it('finish 后无 factorSelectExpr: 提示 AI 未能生成有效脚本', async () => {
+    vi.mocked(sendToNodeVarietyAiHelper).mockImplementation(async ({ callback }: any) => {
+      callback(
+        'message',
+        JSON.stringify({ parsedConditions: [], factorScreen: { factorSelectExpr: null } })
+      )
+      callback('finish')
+    })
+    const wrapper = await mountComponent()
+    ;(wrapper.vm as any).aiInput = '随便说说'
+    await (wrapper.vm as any).handleAiSend()
+    const vm = wrapper.vm as any
+    expect(vm.aiResult.factorSelectExpr).toBe('')
+    expect(vm.aiError).toBe('AI 未能生成有效的因子脚本, 请调整描述后重试')
+  })
+
+  it('message 非法 JSON: finish 时提示解析失败', async () => {
+    vi.mocked(sendToNodeVarietyAiHelper).mockImplementation(async ({ callback }: any) => {
+      callback('message', 'not-a-json')
+      callback('finish')
+    })
+    const wrapper = await mountComponent()
+    ;(wrapper.vm as any).aiInput = 'x'
+    await (wrapper.vm as any).handleAiSend()
+    expect((wrapper.vm as any).aiError).toBe('AI 返回结果解析失败, 请重试')
+  })
+
+  it('error 回调: loading 复位并写入错误信息', async () => {
+    vi.mocked(sendToNodeVarietyAiHelper).mockImplementation(async ({ callback }: any) => {
+      callback('error', 'HTTP 500: Server Error')
+    })
+    const wrapper = await mountComponent()
+    ;(wrapper.vm as any).aiInput = 'x'
+    await (wrapper.vm as any).handleAiSend()
+    const vm = wrapper.vm as any
+    expect(vm.aiLoading).toBe(false)
+    expect(vm.aiError).toBe('HTTP 500: Server Error')
+  })
+
+  it('handleAiInsert: 编辑器为空时直接 setValue 覆盖并关闭对话框', async () => {
+    mockAiSuccess()
+    const wrapper = await mountComponent()
+    ;(wrapper.vm as any).open()
+    await flushPromises()
+    const vm = wrapper.vm as any
+    vm.aiInput = '收盘价大于10'
+    await vm.handleAiSend()
+    // stub 的 setValue 是 vi.fn()
+    const monaco = wrapper.findComponent('.monaco-stub')
+    vm.stMonacoEditorRef = {
+      getValue: vi.fn().mockReturnValue(''),
+      setValue: vi.fn(),
+    }
+    await vm.handleAiInsert()
+    expect(vm.stMonacoEditorRef.setValue).toHaveBeenCalledWith('close_1d_0ref > 10')
+    expect(vm.ruleForm.funcExpression).toBe('close_1d_0ref > 10')
+    expect(vm.aiDialogVisible).toBe(false)
+    expect((globalThis as any).ElMessage.success).toHaveBeenCalledWith('已插入到编辑器')
+    expect(monaco.exists()).toBe(true)
+  })
+
+  it('handleAiInsert: 编辑器已有内容时需确认, 取消则不覆盖', async () => {
+    mockAiSuccess()
+    const wrapper = await mountComponent()
+    const vm = wrapper.vm as any
+    vm.aiInput = 'x'
+    await vm.handleAiSend()
+    vm.stMonacoEditorRef = {
+      getValue: vi.fn().mockReturnValue('old_expr'),
+      setValue: vi.fn(),
+    }
+    ;(globalThis as any).ElMessageBox.confirm.mockRejectedValueOnce('cancel')
+    await vm.handleAiInsert()
+    expect(vm.stMonacoEditorRef.setValue).not.toHaveBeenCalled()
+    // 确认后覆盖
+    ;(globalThis as any).ElMessageBox.confirm.mockResolvedValueOnce('confirm')
+    await vm.handleAiInsert()
+    expect(vm.stMonacoEditorRef.setValue).toHaveBeenCalledWith('close_1d_0ref > 10')
+  })
+
+  it('重新生成: 保留 sessionId 再次发送', async () => {
+    mockAiSuccess()
+    const wrapper = await mountComponent()
+    const vm = wrapper.vm as any
+    vm.aiInput = '收盘价大于10'
+    await vm.handleAiSend()
+    expect(vm.aiSessionId).toBe('sess-1')
+    vm.aiInput = '改成大于20'
+    await vm.handleAiSend()
+    const secondCall = vi.mocked(sendToNodeVarietyAiHelper).mock.calls[1][0] as any
+    expect(secondCall.sessionId).toBe('sess-1')
+    expect(secondCall.value).toBe('改成大于20')
+  })
+
+  it('关闭主窗口时 AI 对话框一并关闭', async () => {
+    const wrapper = await mountComponent()
+    const vm = wrapper.vm as any
+    vm.open()
+    await flushPromises()
+    vm.aiDialogVisible = true
+    vm.visible = false
+    await flushPromises()
+    expect(vm.aiDialogVisible).toBe(false)
+  })
+})

package/tests/components/Map.spec.ts

@@ -0,0 +1,71 @@
+import { vi } from 'vitest'
+
+vi.mock('echarts', () => import('@test/mocks/echarts.mock.ts'))
+
+import { mountWithPlugins } from '@test/helpers/mountWithPlugins.ts'
+import * as echarts from 'echarts'
+import Map from '@package/Map/index.vue'
+
+const getInstance = () => (echarts.init as any).mock.results.at(-1).value
+
+describe('Map', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+  })
+
+  it('挂载后调用 echarts.init + registerMap("China")', () => {
+    mountWithPlugins(Map)
+    expect(echarts.init).toHaveBeenCalledTimes(1)
+    expect(echarts.registerMap).toHaveBeenCalledTimes(1)
+    const args = (echarts.registerMap as any).mock.calls[0]
+    expect(args[0]).toBe('China')
+    expect(args[1]).toBeDefined()
+  })
+
+  it('setOption 结构: visualMap.pieces + series.map=China', () => {
+    const pieces = [
+      { min: 0, max: 10, color: 'red' },
+      { min: 10, color: 'blue' },
+    ]
+    mountWithPlugins(Map, { props: { data: [{ name: '北京', value: 5 }], pieces } })
+    const option = getInstance().setOption.mock.calls[0][0]
+    expect(option.visualMap.pieces).toEqual(pieces)
+    expect(option.series[0].type).toBe('map')
+    expect(option.series[0].map).toBe('China')
+    expect(option.series[0].data).toEqual([{ name: '北京', value: 5 }])
+  })
+
+  it('style prop 合并: 覆盖 area/hoverArea 等', () => {
+    mountWithPlugins(Map, {
+      props: { style: { area: 'pink', hoverArea: 'orange', legendColor: 'gray' } },
+    })
+    const option = getInstance().setOption.mock.calls[0][0]
+    expect(option.series[0].itemStyle.areaColor).toBe('pink')
+    expect(option.series[0].emphasis.itemStyle.areaColor).toBe('orange')
+    expect(option.visualMap.textStyle.color).toBe('gray')
+  })
+
+  it('config.tooltipFormatter 生效', () => {
+    const tooltipFormatter = vi.fn()
+    mountWithPlugins(Map, { props: { config: { tooltipFormatter } } })
+    const option = getInstance().setOption.mock.calls[0][0]
+    expect(option.tooltip.formatter).toBe(tooltipFormatter)
+  })
+
+  it('data/pieces 变化时重绘', async () => {
+    const wrapper = mountWithPlugins(Map, { props: { data: [] } })
+    const instance = getInstance()
+    expect(instance.setOption).toHaveBeenCalledTimes(1)
+    await wrapper.setProps({ data: [{ name: '上海', value: 1 }] })
+    expect(instance.setOption).toHaveBeenCalledTimes(2)
+    await wrapper.setProps({ pieces: [{ min: 1 }] })
+    expect(instance.setOption).toHaveBeenCalledTimes(3)
+  })
+
+  it('卸载时调用 dispose', () => {
+    const wrapper = mountWithPlugins(Map)
+    const instance = getInstance()
+    wrapper.unmount()
+    expect(instance.dispose).toHaveBeenCalledTimes(1)
+  })
+})