| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150 |
- /**
- * The staged card form: what a draft shows before it is written, which wire
- * call a save reaches, and what happens to drafts the Host did not accept.
- */
- import { describe, expect, it, vi } from 'vitest'
- import type { SettingsPathOpView } from '@deepseek-ai/dsh-api-remotes/client'
- import { RemoteError, stubSettingsScope, type StubSettingsScope } from '@deepseek-ai/dsh-client-test-runtime'
- import { CardForm, numberField, textField } from '../src/client/card-form.ts'
- import { SubagentLimitsCardController, type SubagentLimitsSettings } from '../src/client/subagent-limits-card-controller.ts'
- import { subagentCardFace, subagentCardShell } from '../src/client/subagent-card-controller.ts'
- import { AgentLoopCardController, type AgentLoopSettings } from '../src/client/agent-loop-card-controller.ts'
- import { BashCardController, type BashSettings } from '../src/client/bash-card-controller.ts'
- import {
- SubagentModelSelectionCardController,
- subagentModelCandidates,
- type SubagentModelSelectionSettings,
- } from '../src/client/subagent-model-selection-card-controller.ts'
- import { WebSearchCardController, type WebSearchSettings } from '../src/client/web-search-card-controller.ts'
- /** Make the stub behave like a Host that accepts every write. */
- function acceptWrites<T>(host: StubSettingsScope<T>): void {
- const section = (): Record<string, unknown> => ({ ...host.scope.getSnapshot().value as object })
- const layer = (): Record<string, unknown> => ({ ...host.scope.getSnapshot().user as object })
- host.set.mockImplementation((field: string, value: unknown) => {
- host.publish({ value: { ...section(), [field]: value } as T, user: { ...layer(), [field]: value } })
- })
- host.mutate.mockImplementation((ops: readonly SettingsPathOpView[]) => {
- const value = { ...section() }
- const user = { ...layer() }
- for (const op of ops) {
- const field = op.path[0]!
- if (op.op === 'set') {
- value[field] = op.value
- user[field] = op.value
- }
- }
- host.publish({ value: value as T, user })
- })
- host.unset.mockImplementation((field: string) => {
- const user = Object.fromEntries(Object.entries(layer()).filter(([key]) => key !== field))
- const base = host.scope.getSnapshot().base as Record<string, unknown> | undefined
- host.publish({ value: { ...section(), [field]: base?.[field] } as T, user })
- })
- }
- /** The card plugin's context, scripted down to the namespaces a card reaches. */
- function ctxWith(namespaces: object) {
- return { remote: namespaces } as never
- }
- function credentialsApi(configured: boolean) {
- const describe = vi.fn(() => Promise.resolve({
- ok: true as const,
- value: { DEEPSEEK_API_KEY: { configured, writable: true } },
- }))
- const set = vi.fn(() => Promise.resolve({ ok: true as const, value: undefined }))
- return { ctx: ctxWith({ credentials: { describe, set } }), describe, set }
- }
- function modelsApi(options: {
- groups?: readonly {
- id: string
- name: string
- models: readonly { id: string; name: string }[]
- }[]
- failures?: readonly { id: string; name: string; message: string }[]
- error?: string
- } = {}) {
- const models = vi.fn(() => Promise.resolve({
- ...(options.error === undefined
- ? { ok: true as const, value: { groups: options.groups ?? [], failures: options.failures ?? [] } }
- : { ok: false as const, error: new RemoteError('gateway/internal', options.error, {}) }),
- }))
- return { ctx: ctxWith({ session: { modelCatalog: models } }), models }
- }
- function deferred<T>() {
- let resolve!: (value: T) => void
- let reject!: (error: unknown) => void
- const promise = new Promise<T>((accept, fail) => {
- resolve = accept
- reject = fail
- })
- return { promise, resolve, reject }
- }
- describe('CardForm', () => {
- function form() {
- const host = stubSettingsScope<Record<string, unknown>>()
- const subject = new CardForm(host.scope, [numberField('timeoutMs'), textField('baseURL')])
- host.publish({
- status: 'ready',
- writable: true,
- value: { timeoutMs: 60_000, baseURL: 'https://search.test/v1' },
- base: { timeoutMs: 60_000, baseURL: 'https://search.test/v1' },
- user: {},
- })
- return { host, subject }
- }
- it('shows the effective value and stays clean until something is staged', () => {
- const { subject } = form()
- expect(subject.field('timeoutMs')).toEqual({ text: '60000', overridden: false, invalid: false })
- expect(subject.shell()).toMatchObject({ available: true, writable: true, dirty: false, invalid: false })
- })
- it('marks a field the user layer carries as overridden', () => {
- const { host, subject } = form()
- host.publish({ value: { timeoutMs: 60_000 }, user: { timeoutMs: 60_000 } })
- // An override equal to the composition default is still an override.
- expect(subject.field('timeoutMs').overridden).toBe(true)
- })
- it('writes nothing until the form is saved', async () => {
- const { host, subject } = form()
- acceptWrites(host)
- subject.actions().edit('timeoutMs', '9000')
- expect(subject.field('timeoutMs')).toEqual({ text: '9000', overridden: true, invalid: false })
- expect(subject.shell().dirty).toBe(true)
- expect(host.set).not.toHaveBeenCalled()
- await subject.save()
- expect(host.set.mock.calls).toEqual([['timeoutMs', 9_000]])
- expect(subject.shell()).toMatchObject({ dirty: false, failed: false, saving: false })
- })
- it('drops a draft that settles back on the value already shown', async () => {
- const { host, subject } = form()
- subject.actions().edit('timeoutMs', '9000')
- subject.actions().edit('timeoutMs', '60000')
- expect(subject.shell().dirty).toBe(false)
- await subject.save()
- expect(host.set).not.toHaveBeenCalled()
- })
- it('refuses to save while a draft is not a value the field accepts', async () => {
- const { host, subject } = form()
- subject.actions().edit('timeoutMs', 'soon')
- expect(subject.field('timeoutMs')).toEqual({ text: 'soon', overridden: false, invalid: true })
- expect(subject.shell()).toMatchObject({ dirty: true, invalid: true })
- await subject.save()
- expect(host.set).not.toHaveBeenCalled()
- expect(subject.field('timeoutMs').text).toBe('soon')
- })
- it('stages a reset that clears the field only once saved', async () => {
- const { host, subject } = form()
- acceptWrites(host)
- host.publish({ value: { timeoutMs: 9_000 }, user: { timeoutMs: 9_000 } })
- subject.actions().resetField('timeoutMs')
- // The badge previews the save: the field will no longer be overridden.
- expect(subject.field('timeoutMs')).toEqual({ text: '60000', overridden: false, invalid: false })
- expect(host.unset).not.toHaveBeenCalled()
- await subject.save()
- expect(host.unset.mock.calls).toEqual([['timeoutMs']])
- expect(subject.shell()).toMatchObject({ dirty: false, failed: false })
- })
- it('treats resetting an inherited field as no change at all', async () => {
- const { host, subject } = form()
- subject.actions().resetField('timeoutMs')
- expect(subject.shell().dirty).toBe(false)
- await subject.save()
- expect(host.unset).not.toHaveBeenCalled()
- })
- it('clears a number field by emptying it', async () => {
- const { host, subject } = form()
- acceptWrites(host)
- host.publish({ user: { timeoutMs: 9_000 } })
- subject.actions().edit('timeoutMs', '')
- expect(subject.field('timeoutMs')).toEqual({ text: '', overridden: false, invalid: false })
- await subject.save()
- expect(host.unset.mock.calls).toEqual([['timeoutMs']])
- })
- it('clears a text field by emptying it', async () => {
- const { host, subject } = form()
- acceptWrites(host)
- host.publish({ user: { baseURL: 'https://search.test/v1' } })
- subject.actions().edit('baseURL', ' ')
- await subject.save()
- expect(host.unset.mock.calls).toEqual([['baseURL']])
- })
- it('writes the trimmed text of a text field', async () => {
- const { host, subject } = form()
- acceptWrites(host)
- subject.actions().edit('baseURL', ' https://other.test ')
- await subject.save()
- expect(host.set.mock.calls).toEqual([['baseURL', 'https://other.test']])
- })
- it('keeps the drafts a save did not land, and reports the failure', async () => {
- const { host, subject } = form()
- subject.actions().edit('timeoutMs', '9000')
- await subject.save()
- // The stub Host accepted the call without storing it, exactly as a
- // validator that refuses the value does.
- expect(host.set).toHaveBeenCalledWith('timeoutMs', 9_000)
- expect(subject.shell()).toMatchObject({ dirty: true, failed: true, saving: false })
- expect(subject.field('timeoutMs').text).toBe('9000')
- })
- it('reports a reset the Host did not apply as a failure', async () => {
- const { host, subject } = form()
- host.publish({ user: { timeoutMs: 9_000 } })
- subject.actions().resetField('timeoutMs')
- await subject.save()
- expect(host.unset).toHaveBeenCalledWith('timeoutMs')
- expect(subject.shell().failed).toBe(true)
- })
- it('clears the failure as soon as the user edits again', async () => {
- const { subject } = form()
- subject.actions().edit('timeoutMs', '9000')
- await subject.save()
- expect(subject.shell().failed).toBe(true)
- subject.actions().edit('timeoutMs', '9001')
- expect(subject.shell().failed).toBe(false)
- })
- it('discards every staged edit', async () => {
- const { host, subject } = form()
- subject.actions().edit('timeoutMs', '9000')
- subject.actions().discard()
- expect(subject.field('timeoutMs').text).toBe('60000')
- expect(subject.shell()).toMatchObject({ dirty: false, failed: false })
- // A discard with nothing staged publishes nothing.
- const before = subject.shell()
- subject.actions().discard()
- expect(subject.shell()).toEqual(before)
- await subject.save()
- expect(host.set).not.toHaveBeenCalled()
- })
- it('refuses a second save while one is in flight', async () => {
- const { host, subject } = form()
- acceptWrites(host)
- subject.actions().edit('timeoutMs', '9000')
- const first = subject.save()
- expect(subject.shell().saving).toBe(true)
- const second = subject.save()
- await Promise.all([first, second])
- expect(host.set).toHaveBeenCalledTimes(1)
- })
- it('publishes a projection whenever the scope or a draft changes', () => {
- const { host, subject } = form()
- const store = subject.bind(() => subject.field('timeoutMs').text)
- expect(store.getSnapshot()).toBe('60000')
- host.publish({ value: { timeoutMs: 1_000 } })
- expect(store.getSnapshot()).toBe('1000')
- subject.actions().edit('timeoutMs', '2000')
- expect(store.getSnapshot()).toBe('2000')
- })
- it('refuses to address a field the card never declared', () => {
- const { subject } = form()
- expect(() => subject.field('nope')).toThrow('plugin card has no field nope')
- })
- it('renders an absent section value as an empty draft', () => {
- const host = stubSettingsScope<Record<string, unknown>>()
- const subject = new CardForm(host.scope, [numberField('timeoutMs'), textField('baseURL')])
- host.publish({ status: 'ready', writable: true, value: {}, base: {}, user: undefined })
- expect(subject.field('timeoutMs').text).toBe('')
- expect(subject.field('baseURL').text).toBe('')
- expect(subject.shell().available).toBe(true)
- })
- it('stays unavailable while the namespace is not served', () => {
- const host = stubSettingsScope<Record<string, unknown>>()
- const subject = new CardForm(host.scope, [numberField('timeoutMs')])
- host.publish({ status: 'unavailable' })
- expect(subject.shell()).toMatchObject({ available: false, writable: false })
- })
- })
- describe('BashCardController', () => {
- it('projects both fields and saves them in one write pass', async () => {
- const host = stubSettingsScope<BashSettings>()
- acceptWrites(host)
- const controller = new BashCardController(host.scope)
- host.publish({
- status: 'ready',
- writable: true,
- value: { timeoutMs: 5_000, maxOutputBytes: 64_000 },
- base: { timeoutMs: 60_000, maxOutputBytes: 64_000 },
- user: { timeoutMs: 5_000 },
- })
- const face = controller.inject()
- expect(face.hooks.bashCard.getSnapshot()).toMatchObject({
- available: true,
- writable: true,
- dirty: false,
- timeoutMs: { text: '5000', overridden: true },
- maxOutputBytes: { text: '64000', overridden: false },
- })
- face.edit('timeoutMs', '9000')
- face.edit('maxOutputBytes', '1024')
- expect(face.hooks.bashCard.getSnapshot().dirty).toBe(true)
- face.save()
- await vi.waitFor(() => { expect(host.set).toHaveBeenCalledTimes(2) })
- expect(host.set.mock.calls).toEqual([['timeoutMs', 9_000], ['maxOutputBytes', 1_024]])
- expect(face.hooks.bashCard.getSnapshot().dirty).toBe(false)
- })
- it('stages a reset and applies it on save', async () => {
- const host = stubSettingsScope<BashSettings>()
- acceptWrites(host)
- const controller = new BashCardController(host.scope)
- host.publish({
- status: 'ready',
- writable: true,
- value: { timeoutMs: 5_000 },
- base: { timeoutMs: 60_000 },
- user: { timeoutMs: 5_000 },
- })
- const face = controller.inject()
- face.resetField('timeoutMs')
- expect(face.hooks.bashCard.getSnapshot().timeoutMs.text).toBe('60000')
- face.save()
- await vi.waitFor(() => { expect(host.unset).toHaveBeenCalledWith('timeoutMs') })
- expect(face.hooks.bashCard.getSnapshot()).toMatchObject({
- dirty: false,
- timeoutMs: { text: '60000', overridden: false },
- })
- })
- it('discards staged edits without writing', () => {
- const host = stubSettingsScope<BashSettings>()
- const controller = new BashCardController(host.scope)
- host.publish({ status: 'ready', writable: true, value: { timeoutMs: 5_000 }, user: {} })
- const face = controller.inject()
- face.edit('timeoutMs', '9000')
- face.discard()
- expect(face.hooks.bashCard.getSnapshot().timeoutMs.text).toBe('5000')
- expect(host.set).not.toHaveBeenCalled()
- })
- })
- describe('AgentLoopCardController', () => {
- it('saves the only field it owns', async () => {
- const host = stubSettingsScope<AgentLoopSettings>()
- acceptWrites(host)
- const controller = new AgentLoopCardController(host.scope)
- host.publish({
- status: 'ready',
- writable: true,
- value: { maxParallelToolCalls: 10 },
- base: { maxParallelToolCalls: 10 },
- user: {},
- })
- const face = controller.inject()
- face.edit('maxParallelToolCalls', '4')
- face.save()
- await vi.waitFor(() => { expect(host.set).toHaveBeenCalledWith('maxParallelToolCalls', 4) })
- expect(face.hooks.agentLoopCard.getSnapshot()).toMatchObject({
- dirty: false,
- maxParallelToolCalls: { text: '4', overridden: true },
- })
- })
- it('reports a read-only document so the card can disable its controls', () => {
- const host = stubSettingsScope<AgentLoopSettings>()
- const controller = new AgentLoopCardController(host.scope)
- host.publish({ status: 'ready', writable: false, value: { maxParallelToolCalls: 10 } })
- expect(controller.inject().hooks.agentLoopCard.getSnapshot().writable).toBe(false)
- })
- })
- describe('SubagentModelSelectionCardController', () => {
- it('joins stored routes with the live catalog without dropping unavailable choices', () => {
- const candidates = subagentModelCandidates(
- [{ id: 'alpha', name: 'Alpha API', models: [{ id: 'fast', name: 'Fast' }] }],
- [{ provider: 'legacy', model: 'old' }],
- new Set(['legacy\0old']),
- )
- expect(candidates).toEqual([
- {
- key: 'alpha\0fast', provider: 'alpha', model: 'fast', providerName: 'Alpha API',
- modelName: 'Fast', available: true, selected: false,
- },
- {
- key: 'legacy\0old', provider: 'legacy', model: 'old', providerName: 'legacy',
- modelName: 'old', available: false, selected: true,
- },
- ])
- })
- it('loads adapter models and saves the switch and routes atomically', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- acceptWrites(host)
- const models = modelsApi({
- groups: [{ id: 'alpha', name: 'Alpha API', models: [{ id: 'fast', name: 'Fast' }] }],
- })
- const controller = new SubagentModelSelectionCardController(host.scope, models.ctx)
- host.publish({
- status: 'ready', writable: true, revision: 3,
- value: { enabled: false, allowedModels: [] }, user: {},
- })
- const face = controller.inject()
- expect(face.hooks.subagentModelSelectionCard.getSnapshot().enabled).toBe(false)
- face.toggleEnabled()
- await vi.waitFor(() => {
- expect(face.hooks.subagentModelSelectionCard.getSnapshot().candidates).toHaveLength(1)
- })
- face.toggleModel('alpha\0fast')
- face.save()
- await vi.waitFor(() => {
- expect(host.mutate).toHaveBeenCalledWith([
- { op: 'set', path: ['enabled'], value: true },
- { op: 'set', path: ['allowedModels'], value: [{ provider: 'alpha', model: 'fast' }] },
- ], 3)
- })
- expect(face.hooks.subagentModelSelectionCard.getSnapshot()).toMatchObject({
- enabled: true,
- dirty: false,
- saving: false,
- failed: false,
- })
- })
- it('starts an empty draft when a ready test scope has no decoded value', () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- const controller = new SubagentModelSelectionCardController(host.scope, modelsApi().ctx)
- host.publish({ status: 'ready', writable: true, revision: 0, value: undefined })
- const face = controller.inject()
- face.toggleEnabled()
- expect(face.hooks.subagentModelSelectionCard.getSnapshot()).toMatchObject({
- enabled: true, dirty: true, invalid: true,
- })
- })
- it('keeps the Host value and reports a rejected write', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- const models = modelsApi({
- groups: [{ id: 'alpha', name: 'Alpha API', models: [{ id: 'fast', name: 'Fast' }] }],
- })
- const controller = new SubagentModelSelectionCardController(host.scope, models.ctx)
- host.publish({ status: 'ready', writable: true, value: { enabled: false, allowedModels: [] }, user: {} })
- const face = controller.inject()
- face.toggleEnabled()
- await vi.waitFor(() => {
- expect(face.hooks.subagentModelSelectionCard.getSnapshot().candidates).toHaveLength(1)
- })
- face.toggleModel('alpha\0fast')
- face.save()
- await vi.waitFor(() => {
- expect(face.hooks.subagentModelSelectionCard.getSnapshot().failed).toBe(true)
- })
- expect(face.hooks.subagentModelSelectionCard.getSnapshot()).toMatchObject({
- enabled: true,
- dirty: true,
- saving: false,
- })
- })
- it('loads stored routes, stages removal and disablement, and discards both', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- const models = modelsApi({
- groups: [{ id: 'alpha', name: 'Alpha API', models: [{ id: 'fast', name: 'Fast' }] }],
- failures: [{ id: 'beta', name: 'Beta', message: 'offline' }],
- })
- const controller = new SubagentModelSelectionCardController(host.scope, models.ctx)
- host.publish({
- status: 'ready', writable: true, revision: 5,
- value: { enabled: true, allowedModels: [{ provider: 'alpha', model: 'fast' }] }, user: {},
- })
- const face = controller.inject()
- const state = () => face.hooks.subagentModelSelectionCard.getSnapshot()
- await vi.waitFor(() => { expect(state().catalogStatus).toBe('ready') })
- expect(state().catalogPartial).toBe(true)
- face.toggleModel('missing')
- expect(state().dirty).toBe(false)
- face.toggleModel('alpha\0fast')
- expect(state()).toMatchObject({ dirty: true, invalid: true })
- face.discard()
- expect(state()).toMatchObject({ dirty: false, invalid: false, enabled: true })
- face.toggleEnabled()
- expect(state()).toMatchObject({ dirty: true, enabled: false })
- face.toggleEnabled()
- expect(state()).toMatchObject({ dirty: false, enabled: true })
- })
- it('retains selected routes when disabling and loads an already-ready enabled card', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- acceptWrites(host)
- host.publish({
- status: 'ready', writable: true, revision: 5,
- value: { enabled: true, allowedModels: [{ provider: 'alpha', model: 'fast' }] }, user: {},
- })
- const models = modelsApi({
- groups: [{ id: 'alpha', name: 'Alpha API', models: [{ id: 'fast', name: 'Fast' }] }],
- })
- const controller = new SubagentModelSelectionCardController(host.scope, models.ctx)
- const face = controller.inject()
- await vi.waitFor(() => { expect(models.models).toHaveBeenCalledOnce() })
- face.toggleEnabled()
- face.save()
- await vi.waitFor(() => {
- expect(host.mutate).toHaveBeenCalledWith([
- { op: 'set', path: ['enabled'], value: false },
- { op: 'set', path: ['allowedModels'], value: [{ provider: 'alpha', model: 'fast' }] },
- ], 5)
- })
- expect(face.hooks.subagentModelSelectionCard.getSnapshot()).toMatchObject({
- enabled: false, dirty: false,
- })
- })
- it('reports a directory error and retries it', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- const models = modelsApi({ error: 'offline' })
- const controller = new SubagentModelSelectionCardController(host.scope, models.ctx)
- host.publish({ status: 'ready', writable: true, value: { enabled: false, allowedModels: [] }, user: {} })
- const face = controller.inject()
- const state = () => face.hooks.subagentModelSelectionCard.getSnapshot()
- face.toggleEnabled()
- await vi.waitFor(() => { expect(state().catalogStatus).toBe('error') })
- face.retryCatalog()
- await vi.waitFor(() => { expect(models.models).toHaveBeenCalledTimes(2) })
- })
- it('rejects a draft after the Host revision changes', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- const models = modelsApi({
- groups: [{ id: 'alpha', name: 'Alpha API', models: [{ id: 'fast', name: 'Fast' }] }],
- })
- const controller = new SubagentModelSelectionCardController(host.scope, models.ctx)
- host.publish({
- status: 'ready', writable: true, revision: 4,
- value: { enabled: false, allowedModels: [] }, user: {},
- })
- const face = controller.inject()
- face.toggleEnabled()
- await vi.waitFor(() => {
- expect(face.hooks.subagentModelSelectionCard.getSnapshot().candidates).toHaveLength(1)
- })
- face.toggleModel('alpha\0fast')
- host.publish({
- revision: 5,
- value: { enabled: true, allowedModels: [{ provider: 'other', model: 'new' }] },
- })
- expect(face.hooks.subagentModelSelectionCard.getSnapshot()).toMatchObject({
- conflicted: true, failed: false, dirty: true,
- })
- face.save()
- await Promise.resolve()
- expect(host.mutate).not.toHaveBeenCalled()
- face.discard()
- expect(face.hooks.subagentModelSelectionCard.getSnapshot()).toMatchObject({
- conflicted: false, failed: false, dirty: false, enabled: true,
- })
- })
- it('settles a draft when a newer Host revision already contains it', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- const models = modelsApi({
- groups: [{ id: 'alpha', name: 'Alpha', models: [{ id: 'fast', name: 'Fast' }] }],
- })
- const controller = new SubagentModelSelectionCardController(host.scope, models.ctx)
- host.publish({
- status: 'ready', writable: true, revision: 4,
- value: { enabled: false, allowedModels: [] }, user: {},
- })
- const face = controller.inject()
- face.toggleEnabled()
- await vi.waitFor(() => { expect(face.hooks.subagentModelSelectionCard.getSnapshot().candidates).toHaveLength(1) })
- face.toggleModel('alpha\0fast')
- host.publish({
- revision: 5,
- value: { enabled: true, allowedModels: [{ provider: 'alpha', model: 'fast' }] },
- })
- expect(face.hooks.subagentModelSelectionCard.getSnapshot()).toMatchObject({
- conflicted: false, dirty: false, enabled: true,
- })
- })
- it('retains unsaved routes across a catalog refresh', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- acceptWrites(host)
- host.publish({
- status: 'ready', writable: true, revision: 2,
- value: { enabled: false, allowedModels: [] }, user: {},
- })
- const refreshed = deferred<never>()
- const models = vi.fn()
- .mockResolvedValueOnce({
- ok: true, value: {
- groups: [{ id: 'alpha', name: 'Alpha', models: [{ id: 'fast', name: 'Fast' }] }],
- failures: [],
- },
- })
- .mockImplementationOnce(() => refreshed.promise)
- const controller = new SubagentModelSelectionCardController(
- host.scope, ctxWith({ session: { modelCatalog: models } }),
- )
- const face = controller.inject()
- const state = () => face.hooks.subagentModelSelectionCard.getSnapshot()
- face.toggleEnabled()
- await vi.waitFor(() => { expect(state().candidates).toHaveLength(1) })
- face.toggleModel('alpha\0fast')
- controller.refreshCatalog()
- expect(state()).toMatchObject({
- catalogStatus: 'loading',
- candidates: [expect.objectContaining({ key: 'alpha\0fast', selected: true })],
- })
- refreshed.resolve({
- ok: true, value: { groups: [], failures: [] },
- } as never)
- await vi.waitFor(() => { expect(state().catalogStatus).toBe('ready') })
- expect(state().candidates).toEqual([
- expect.objectContaining({ key: 'alpha\0fast', available: false, selected: true }),
- ])
- face.save()
- await vi.waitFor(() => {
- expect(host.mutate).toHaveBeenCalledWith([
- { op: 'set', path: ['enabled'], value: true },
- { op: 'set', path: ['allowedModels'], value: [{ provider: 'alpha', model: 'fast' }] },
- ], 2)
- })
- })
- it('drops a draft when the connection generation changes', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- const models = modelsApi({
- groups: [{ id: 'alpha', name: 'Alpha', models: [{ id: 'fast', name: 'Fast' }] }],
- })
- host.publish({
- status: 'ready', writable: true, revision: 4,
- value: { enabled: false, allowedModels: [] }, user: {},
- })
- const controller = new SubagentModelSelectionCardController(host.scope, models.ctx)
- const face = controller.inject()
- face.toggleEnabled()
- await vi.waitFor(() => { expect(face.hooks.subagentModelSelectionCard.getSnapshot().candidates).toHaveLength(1) })
- face.toggleModel('alpha\0fast')
- controller.resetConnection()
- host.publish({
- revision: 4,
- value: { enabled: true, allowedModels: [{ provider: 'other', model: 'new' }] },
- })
- expect(face.hooks.subagentModelSelectionCard.getSnapshot()).toMatchObject({
- conflicted: false, dirty: false, enabled: true,
- })
- face.save()
- await Promise.resolve()
- expect(host.mutate).not.toHaveBeenCalled()
- })
- it('reloads the model catalog after invalidation', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- host.publish({
- status: 'ready', writable: true, revision: 1,
- value: { enabled: true, allowedModels: [] }, user: {},
- })
- const models = vi.fn()
- .mockResolvedValueOnce({
- ok: true, value: {
- groups: [{ id: 'alpha', name: 'Alpha', models: [{ id: 'fast', name: 'Fast' }] }],
- failures: [],
- },
- })
- .mockResolvedValueOnce({
- ok: true, value: {
- groups: [{ id: 'beta', name: 'Beta', models: [{ id: 'new', name: 'New' }] }],
- failures: [],
- },
- })
- const controller = new SubagentModelSelectionCardController(
- host.scope, ctxWith({ session: { modelCatalog: models } }),
- )
- const state = () => controller.inject().hooks.subagentModelSelectionCard.getSnapshot()
- await vi.waitFor(() => { expect(state().candidates[0]?.provider).toBe('alpha') })
- controller.refreshCatalog()
- await vi.waitFor(() => { expect(state().candidates[0]?.provider).toBe('beta') })
- expect(models).toHaveBeenCalledTimes(2)
- })
- it('suppresses duplicate actions and late save settlements', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- const catalog = modelsApi({
- groups: [{ id: 'alpha', name: 'Alpha API', models: [{ id: 'fast', name: 'Fast' }] }],
- })
- const write = deferred<undefined>()
- const mutate = vi.fn(async (ops: readonly SettingsPathOpView[]) => {
- await write.promise
- const enabled = ops.find(op => op.path[0] === 'enabled')
- const allowedModels = ops.find(op => op.path[0] === 'allowedModels')
- host.publish({ value: {
- enabled: enabled?.op === 'set' ? enabled.value as boolean : false,
- allowedModels: allowedModels?.op === 'set' ? allowedModels.value as never[] : [],
- } })
- })
- const controller = new SubagentModelSelectionCardController({ ...host.scope, mutate }, catalog.ctx)
- const face = controller.inject()
- face.save()
- face.toggleModel('alpha\0fast')
- host.publish({ status: 'ready', writable: true, value: { enabled: false, allowedModels: [] }, user: {} })
- face.save()
- face.toggleEnabled()
- await vi.waitFor(() => { expect(face.hooks.subagentModelSelectionCard.getSnapshot().catalogStatus).toBe('ready') })
- face.save()
- face.toggleModel('alpha\0fast')
- face.save()
- expect(face.hooks.subagentModelSelectionCard.getSnapshot().saving).toBe(true)
- face.toggleEnabled()
- face.toggleModel('alpha\0fast')
- face.save()
- face.discard()
- controller.dispose()
- write.resolve(undefined)
- await write.promise
- expect(mutate).toHaveBeenCalledOnce()
- })
- it('suppresses duplicate directory loads and late settlements', async () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- host.publish({ status: 'ready', writable: true, value: { enabled: false, allowedModels: [] }, user: {} })
- const pending = deferred<never>()
- const models = vi.fn(() => pending.promise)
- const controller = new SubagentModelSelectionCardController(host.scope, ctxWith({ session: { modelCatalog: models } }))
- const face = controller.inject()
- face.toggleEnabled()
- face.retryCatalog()
- expect(models).toHaveBeenCalledOnce()
- controller.dispose()
- pending.resolve({ ok: false, error: new RemoteError('gateway/internal', 'late failure', {}) } as never)
- await pending.promise
- const pendingResolve = deferred<never>()
- const resolving = new SubagentModelSelectionCardController(
- host.scope,
- ctxWith({ session: { modelCatalog: () => pendingResolve.promise } }),
- )
- const resolvingFace = resolving.inject()
- resolvingFace.toggleEnabled()
- resolving.dispose()
- pendingResolve.resolve({
- ok: true, value: { groups: [], failures: [] },
- } as never)
- await pendingResolve.promise
- })
- it('ignores writes while read-only and scope notifications after disposal', () => {
- const host = stubSettingsScope<SubagentModelSelectionSettings>()
- const controller = new SubagentModelSelectionCardController(host.scope, modelsApi().ctx)
- host.publish({ status: 'ready', writable: false, value: { enabled: false, allowedModels: [] }, user: {} })
- const face = controller.inject()
- face.toggleEnabled()
- face.toggleModel('alpha\0fast')
- face.save()
- expect(host.mutate).not.toHaveBeenCalled()
- controller.dispose()
- controller.refreshCatalog()
- controller.resetConnection()
- face.toggleEnabled()
- face.retryCatalog()
- face.save()
- host.publish({ value: { enabled: true, allowedModels: [{ provider: 'alpha', model: 'fast' }] } })
- expect(host.mutate).not.toHaveBeenCalled()
- expect(face.hooks.subagentModelSelectionCard.getSnapshot().enabled).toBe(false)
- })
- })
- describe('WebSearchCardController', () => {
- it('reads the credential state for the reference the tab names', async () => {
- const host = stubSettingsScope<WebSearchSettings>()
- const credentials = credentialsApi(true)
- const controller = new WebSearchCardController(host.scope, credentials.ctx)
- const state = () => controller.inject().hooks.webSearchCard.getSnapshot()
- await vi.waitFor(() => { expect(credentials.describe).toHaveBeenCalled() })
- host.publish({ status: 'ready', writable: true, value: { baseURL: 'https://search.test/v1' }, user: {} })
- await vi.waitFor(() => { expect(state().apiKeyConfigured).toBe(true) })
- expect(state()).toMatchObject({
- baseURL: { text: 'https://search.test/v1', overridden: false },
- apiKey: { text: '', overridden: false },
- })
- })
- it('writes the staged key through the credentials domain, never the settings section', async () => {
- const host = stubSettingsScope<WebSearchSettings>()
- const credentials = credentialsApi(false)
- const controller = new WebSearchCardController(host.scope, credentials.ctx)
- host.publish({ status: 'ready', writable: true, value: {}, user: {} })
- const face = controller.inject()
- face.edit('apiKey', ' ds-secret ')
- expect(face.hooks.webSearchCard.getSnapshot().dirty).toBe(true)
- expect(credentials.set).not.toHaveBeenCalled()
- credentials.describe.mockImplementation(() => Promise.resolve({
- ok: true as const,
- value: { DEEPSEEK_API_KEY: { configured: true, writable: true } },
- }))
- face.save()
- await vi.waitFor(() => { expect(credentials.set).toHaveBeenCalled() })
- expect(credentials.set).toHaveBeenCalledWith('DEEPSEEK_API_KEY', 'ds-secret')
- expect(host.set).not.toHaveBeenCalled()
- await vi.waitFor(() => {
- expect(face.hooks.webSearchCard.getSnapshot()).toMatchObject({ dirty: false, apiKeyConfigured: true })
- })
- })
- it('keeps the stored key when the draft is left blank', () => {
- const host = stubSettingsScope<WebSearchSettings>()
- const credentials = credentialsApi(true)
- const controller = new WebSearchCardController(host.scope, credentials.ctx)
- host.publish({ status: 'ready', writable: true, value: {}, user: {} })
- const face = controller.inject()
- face.edit('apiKey', ' ')
- expect(face.hooks.webSearchCard.getSnapshot().dirty).toBe(false)
- face.save()
- expect(credentials.set).not.toHaveBeenCalled()
- })
- it('re-reads when the Host reports the watched reference changed', async () => {
- const host = stubSettingsScope<WebSearchSettings>()
- const credentials = credentialsApi(false)
- const controller = new WebSearchCardController(host.scope, credentials.ctx)
- host.publish({ status: 'ready', writable: true, value: {}, user: {} })
- await vi.waitFor(() => { expect(credentials.describe).toHaveBeenCalled() })
- credentials.describe.mockClear()
- // Another reference is not this card's business.
- controller.refreshCredential('OTHER_KEY')
- expect(credentials.describe).not.toHaveBeenCalled()
- // A key written on another surface reaches this card only through this signal.
- credentials.describe.mockImplementation(() => Promise.resolve({
- ok: true as const,
- value: { DEEPSEEK_API_KEY: { configured: true, writable: true } },
- }))
- controller.refreshCredential('DEEPSEEK_API_KEY')
- await vi.waitFor(() => {
- expect(controller.inject().hooks.webSearchCard.getSnapshot().apiKeyConfigured).toBe(true)
- })
- })
- it('addresses the reference the tab declares rather than the default', async () => {
- const host = stubSettingsScope<WebSearchSettings>()
- const credentials = credentialsApi(false)
- const controller = new WebSearchCardController(host.scope, credentials.ctx)
- host.publish({ status: 'ready', writable: true, value: { apiKeyEnv: 'SEARCH_KEY' }, user: {} })
- const face = controller.inject()
- face.edit('apiKey', 'ds-secret')
- face.save()
- await vi.waitFor(() => { expect(credentials.set).toHaveBeenCalled() })
- expect(credentials.set).toHaveBeenCalledWith('SEARCH_KEY', 'ds-secret')
- })
- it('reports a key the Host did not store as a failed save', async () => {
- const host = stubSettingsScope<WebSearchSettings>()
- const credentials = credentialsApi(false)
- const controller = new WebSearchCardController(host.scope, credentials.ctx)
- host.publish({ status: 'ready', writable: true, value: {}, user: {} })
- const face = controller.inject()
- face.edit('apiKey', 'ds-secret')
- face.save()
- await vi.waitFor(() => {
- expect(face.hooks.webSearchCard.getSnapshot()).toMatchObject({ failed: true, dirty: true })
- })
- })
- it('keeps the card usable when the credential read is refused', async () => {
- const host = stubSettingsScope<WebSearchSettings>()
- const refusal = () => Promise.resolve({
- ok: false as const,
- error: new RemoteError('credential/rejected', 'offline', { ref: 'DEEPSEEK_API_KEY' }),
- })
- const describe = vi.fn(refusal)
- const set = vi.fn(refusal)
- const controller = new WebSearchCardController(host.scope, ctxWith({ credentials: { describe, set } }))
- const face = controller.inject()
- await vi.waitFor(() => { expect(describe).toHaveBeenCalled() })
- host.publish({ status: 'ready', writable: true, value: { baseURL: 'https://search.test/v1' }, user: {} })
- face.edit('apiKey', 'ds-secret')
- face.save()
- await vi.waitFor(() => { expect(set).toHaveBeenCalled() })
- expect(face.hooks.webSearchCard.getSnapshot()).toMatchObject({
- available: true,
- apiKeyConfigured: false,
- baseURL: { text: 'https://search.test/v1' },
- })
- })
- it('ignores a credential read the Host refused', async () => {
- const host = stubSettingsScope<WebSearchSettings>()
- const describe = vi.fn(() => Promise.resolve({
- ok: false as const,
- error: new RemoteError('gateway/internal', 'no credential provider', {}),
- }))
- const controller = new WebSearchCardController(host.scope, ctxWith({
- credentials: { describe, set: vi.fn() },
- }))
- await vi.waitFor(() => { expect(describe).toHaveBeenCalled() })
- expect(controller.inject().hooks.webSearchCard.getSnapshot().apiKeyConfigured).toBe(false)
- })
- it('saves the endpoint and the search budget together', async () => {
- const host = stubSettingsScope<WebSearchSettings>()
- acceptWrites(host)
- const credentials = credentialsApi(true)
- const controller = new WebSearchCardController(host.scope, credentials.ctx)
- host.publish({ status: 'ready', writable: true, value: {}, base: {}, user: {} })
- const face = controller.inject()
- face.edit('baseURL', 'https://other.test')
- face.edit('maxUses', '3')
- face.save()
- await vi.waitFor(() => { expect(host.set).toHaveBeenCalledTimes(2) })
- expect(host.set.mock.calls).toEqual([['baseURL', 'https://other.test'], ['maxUses', 3]])
- expect(credentials.set).not.toHaveBeenCalled()
- })
- })
- describe('SubagentLimitsCardController', () => {
- it('validates staged limits, saves them, and restores composed defaults', async () => {
- const host = stubSettingsScope<SubagentLimitsSettings>()
- const face = new SubagentLimitsCardController(host.scope).inject()
- const state = () => face.hooks.subagentLimitsCard.getSnapshot()
- host.publish({ status: 'ready', writable: true, value: { maxDepth: 3, maxActiveSubagents: 8 }, base: { maxDepth: 3, maxActiveSubagents: 8 }, user: {} })
- acceptWrites(host)
- expect(state().maxActiveSubagents.text).toBe('8')
- for (const draft of ['-1', '1.5', '9007199254740992', 'wat', '-0']) {
- face.edit('maxDepth', draft)
- expect(state().invalid).toBe(true)
- }
- face.edit('maxDepth', '0')
- face.edit('maxActiveSubagents', '0')
- expect(state().invalid).toBe(true)
- face.edit('maxActiveSubagents', '12')
- expect(host.set).not.toHaveBeenCalled()
- face.save()
- await vi.waitFor(() => { expect(state().saving).toBe(false) })
- expect(host.scope.getSnapshot().value).toEqual({ maxDepth: 0, maxActiveSubagents: 12 })
- face.resetField('maxDepth')
- face.edit('maxActiveSubagents', '')
- expect(state().invalid).toBe(false)
- face.save()
- await vi.waitFor(() => { expect(state().saving).toBe(false) })
- expect(host.scope.getSnapshot().value).toEqual({ maxDepth: 3, maxActiveSubagents: 8 })
- })
- })
- describe('shared Subagent card actions', () => {
- function card() {
- const limits = stubSettingsScope<SubagentLimitsSettings>()
- const models = stubSettingsScope<SubagentModelSelectionSettings>()
- const limitFace = new SubagentLimitsCardController(limits.scope).inject()
- const modelFace = new SubagentModelSelectionCardController(models.scope, modelsApi().ctx).inject()
- limits.publish({
- status: 'ready', writable: true, revision: 2,
- value: { maxDepth: 3, maxActiveSubagents: 8 },
- base: { maxDepth: 3, maxActiveSubagents: 8 }, user: {},
- })
- models.publish({
- status: 'ready', writable: true, revision: 5,
- value: { enabled: false, allowedModels: [{ provider: 'alpha', model: 'fast' }] }, user: {},
- })
- acceptWrites(limits)
- acceptWrites(models)
- const face = subagentCardFace(limitFace, modelFace)
- const state = () => subagentCardShell(
- face.hooks.subagentLimitsCard.getSnapshot(),
- face.hooks.subagentModelSelectionCard.getSnapshot(),
- )
- return { limits, models, face, state }
- }
- it('saves both drafts through their existing namespaces from one action', async () => {
- const { limits, models, face, state } = card()
- face.editLimit('maxDepth', '2')
- face.toggleEnabled()
- face.save()
- await vi.waitFor(() => { expect(state()).toMatchObject({ saving: false, dirty: false, failed: false }) })
- expect(limits.set).toHaveBeenCalledWith('maxDepth', 2)
- expect(models.mutate).toHaveBeenCalledWith([
- { op: 'set', path: ['enabled'], value: true },
- { op: 'set', path: ['allowedModels'], value: [{ provider: 'alpha', model: 'fast' }] },
- ], 5)
- })
- it('saves a limit-only draft without rewriting model authorization', async () => {
- const { limits, models, face, state } = card()
- face.editLimit('maxDepth', '2')
- face.save()
- await vi.waitFor(() => { expect(state()).toMatchObject({ saving: false, dirty: false, failed: false }) })
- expect(limits.scope.getSnapshot().value?.maxDepth).toBe(2)
- expect(models.mutate).not.toHaveBeenCalled()
- expect(models.scope.getSnapshot().value?.enabled).toBe(false)
- })
- it('retains the pending draft when discard is requested before both writes finish', async () => {
- const { limits, face, state } = card()
- const pending = deferred<undefined>()
- const set = vi.spyOn(limits.scope, 'set').mockImplementationOnce(async () => {
- await pending.promise
- limits.publish({ value: { maxDepth: 2, maxActiveSubagents: 8 }, user: { maxDepth: 2 } })
- })
- face.editLimit('maxDepth', '2')
- face.toggleEnabled()
- face.save()
- try {
- await vi.waitFor(() => {
- expect(face.hooks.subagentModelSelectionCard.getSnapshot()).toMatchObject({ saving: false, dirty: false })
- })
- expect(state().saving).toBe(true)
- expect(set).toHaveBeenCalledWith('maxDepth', 2)
- face.discard()
- expect(face.hooks.subagentLimitsCard.getSnapshot()).toMatchObject({ dirty: true, maxDepth: { text: '2' } })
- } finally {
- pending.resolve(undefined)
- await vi.waitFor(() => { expect(state().saving).toBe(false) })
- }
- expect(state()).toMatchObject({ dirty: false, failed: false })
- expect(limits.scope.getSnapshot().value?.maxDepth).toBe(2)
- })
- it('writes neither namespace when either draft is invalid and discards both', () => {
- const { limits, models, face, state } = card()
- face.editLimit('maxDepth', '1.5')
- face.toggleEnabled()
- face.save()
- expect(limits.set).not.toHaveBeenCalled()
- expect(models.mutate).not.toHaveBeenCalled()
- face.discard()
- expect(state()).toMatchObject({ dirty: false, invalid: false })
- expect(face.hooks.subagentLimitsCard.getSnapshot().maxDepth.text).toBe('3')
- expect(face.hooks.subagentModelSelectionCard.getSnapshot().enabled).toBe(false)
- })
- it('retains a rejected model draft after limits save, and retries only that draft', async () => {
- const { limits, models, face, state } = card()
- models.mutate.mockImplementationOnce(() => {})
- face.editLimit('maxDepth', '2')
- face.toggleEnabled()
- face.save()
- await vi.waitFor(() => { expect(state()).toMatchObject({ saving: false, dirty: true, failed: true }) })
- expect(face.hooks.subagentLimitsCard.getSnapshot().dirty).toBe(false)
- expect(face.hooks.subagentModelSelectionCard.getSnapshot()).toMatchObject({ enabled: true, dirty: true })
- face.save()
- await vi.waitFor(() => { expect(state()).toMatchObject({ saving: false, dirty: false, failed: false }) })
- expect(limits.set).toHaveBeenCalledOnce()
- expect(models.mutate).toHaveBeenCalledTimes(2)
- })
- })
|