100 lines
6.1 KiB
JavaScript
100 lines
6.1 KiB
JavaScript
const { test } = require('node:test')
|
|
const assert = require('node:assert/strict')
|
|
const fs = require('node:fs')
|
|
const vm = require('node:vm')
|
|
const ts = require('typescript')
|
|
const React = require('react')
|
|
|
|
function compile(path, overrides = {}) {
|
|
const exports = {}
|
|
const source = fs.readFileSync(require.resolve(path), 'utf8')
|
|
vm.runInNewContext(ts.transpileModule(source, { compilerOptions: { module: ts.ModuleKind.CommonJS, target: ts.ScriptTarget.ES2022, jsx: ts.JsxEmit.ReactJSX } }).outputText,
|
|
{ exports, document: { body: {} }, require: name => overrides[name] ?? require(name) })
|
|
return exports
|
|
}
|
|
const models = compile('../src/renderer/src/features/workspace/agentModels.ts')
|
|
const options = [{ provider: 'OPENAI', model: 'gpt-6.1-sol', imagesEnabled: false }, { provider: 'QWEN', model: 'qwen3.8-flash', imagesEnabled: true }]
|
|
const plain = value => JSON.parse(JSON.stringify(value))
|
|
|
|
test('normalizes enabled models, validates IDs and preserves old single-model servers', () => {
|
|
const status = models.normalizeAgentStatus({ enabled: true, models: [options[1], options[0], options[0], { provider: 'FORGED', model: 'bad' }, { provider: 'GEMINI', model: '' }] })
|
|
assert.deepEqual(plain(status.models), options)
|
|
assert.equal(status.selectionSupported, true)
|
|
assert.deepEqual(plain(models.normalizeAgentStatus({ enabled: true, ...options[1] }).models), [options[1]])
|
|
assert.equal(models.normalizeAgentStatus({ enabled: true, ...options[1] }).selectionSupported, false)
|
|
assert.equal(models.normalizeAgentStatus({ enabled: false, models: options }).models.length, 0)
|
|
assert.equal(models.normalizeAgentStatus({ enabled: true, models: [] }).enabled, false)
|
|
})
|
|
|
|
test('selects the saved provider only while it is available, and never invents one', () => {
|
|
assert.equal(models.selectedAgentModel(options, 'QWEN'), options[1])
|
|
assert.equal(models.selectedAgentModel(options, 'ANTHROPIC'), options[0])
|
|
assert.equal(models.selectedAgentModel([options[0]], 'QWEN'), options[0])
|
|
assert.equal(models.selectedAgentModel([], 'QWEN'), null)
|
|
})
|
|
|
|
function elements(root, predicate) {
|
|
if (!React.isValidElement(root)) return []
|
|
return [...(predicate(root) ? [root] : []), ...React.Children.toArray(root.props.children).flatMap(child => elements(child, predicate))]
|
|
}
|
|
function selectorHarness(props) {
|
|
const state = [], refs = []
|
|
let stateIndex = 0, refIndex = 0
|
|
const { AgentModelSelector } = compile('../src/renderer/src/components/AgentModelSelector.tsx', {
|
|
'../i18n': { t: value => value }, '../features/workspace/agentModels': models,
|
|
'react-dom': { createPortal: child => child },
|
|
react: { ...React, useId: () => 'model-menu', useEffect() {}, useLayoutEffect() {},
|
|
useState(initial) { const index = stateIndex++; if (!(index in state)) state[index] = initial; return [state[index], value => { state[index] = typeof value === 'function' ? value(state[index]) : value }] },
|
|
useRef(initial) { return refs[refIndex++] ?? (refs[refIndex - 1] = { current: initial }) }
|
|
}
|
|
})
|
|
return { render(patch = {}) { stateIndex = 0; refIndex = 0; return AgentModelSelector({ ...props, ...patch }) } }
|
|
}
|
|
|
|
for (const dark of [true, false]) test(`model control lists only allowed providers and follows the theme (${dark})`, () => {
|
|
let selected
|
|
const h = selectorHarness({ models: options, provider: 'QWEN', dark, disabled: false, onChange: value => { selected = value } })
|
|
const tree = h.render()
|
|
const trigger = elements(tree, node => node.type === 'button')[0]
|
|
assert.equal(trigger.props['aria-haspopup'], 'menu')
|
|
assert.equal(trigger.props['aria-expanded'], false)
|
|
assert.equal(trigger.props.title, 'Qwen · qwen3.8-flash')
|
|
assert.equal(elements(tree, node => node.props.role === 'menu').length, 0)
|
|
assert.equal(elements(tree, node => node.type === 'select').length, 0)
|
|
trigger.props.onClick()
|
|
const menu = elements(h.render(), node => node.props.role === 'menu')[0]
|
|
assert.match(menu.props.className, dark ? /bg-\[#252526\]/ : /bg-white/)
|
|
assert.match(menu.props.className, /fixed/)
|
|
const items = elements(menu, node => node.props.role === 'menuitemradio')
|
|
assert.deepEqual(items.map(node => node.props.title), ['OpenAI · gpt-6.1-sol', 'Qwen · qwen3.8-flash'])
|
|
assert.deepEqual(items.map(node => node.props['aria-checked']), [false, true])
|
|
assert.ok(items.every(node => elements(node, item => item.type === 'span' && item.props.className.includes('truncate')).length === 1))
|
|
items[0].props.onClick()
|
|
assert.equal(selected, 'OPENAI')
|
|
assert.equal(elements(h.render(), node => node.props.role === 'menu').length, 0)
|
|
assert.ok(elements(h.render({ disabled: true }), node => node.type === 'button')[0].props.disabled)
|
|
const empty = elements(h.render({ models: [], provider: null }), node => node.type === 'button')[0]
|
|
assert.ok(empty.props.disabled)
|
|
assert.equal(empty.props.title, '사용 가능한 모델 없음')
|
|
})
|
|
|
|
test('model menu opens from arrow keys and closes when a request blocks the selector', () => {
|
|
const h = selectorHarness({ models: options, provider: 'QWEN', dark: true, disabled: false, onChange() {} })
|
|
const trigger = elements(h.render(), node => node.type === 'button')[0]
|
|
trigger.props.onKeyDown({ key: 'ArrowUp', preventDefault() {} })
|
|
assert.equal(elements(h.render(), node => node.props.role === 'menu').length, 1)
|
|
assert.equal(elements(h.render({ disabled: true }), node => node.props.role === 'menu').length, 0)
|
|
})
|
|
|
|
test('input footer places the model action before the attachment and clear controls', () => {
|
|
const source = fs.readFileSync(require.resolve('../src/renderer/src/components/ChatWidgets.tsx'), 'utf8')
|
|
const row = source.slice(source.indexOf('{(attachmentsEnabled || onClearConversation || footerAction)'))
|
|
assert.ok(row.indexOf('{footerAction}') < row.indexOf('title={t(\'파일 첨부\')}'))
|
|
assert.ok(row.indexOf('{footerAction}') < row.indexOf('title={t(\'대화내역 삭제\')}'))
|
|
})
|
|
|
|
test('A-Copilot has a single managed label', () => {
|
|
assert.equal(models.agentModelLabel({ provider: 'A_COPILOT', model: 'A-Copilot', imagesEnabled: false }), 'A-Copilot')
|
|
assert.equal(models.agentModelLabel(options[0]), 'OpenAI · gpt-6.1-sol')
|
|
})
|