test(teams): stabilize runtime launch validation mocks
This commit is contained in:
parent
95b62d6013
commit
728603d788
3 changed files with 90 additions and 7 deletions
|
|
@ -23,6 +23,7 @@ vi.mock('@main/services/team/ClaudeBinaryResolver', () => ({
|
||||||
|
|
||||||
vi.mock('@main/utils/childProcess', () => ({
|
vi.mock('@main/utils/childProcess', () => ({
|
||||||
spawnCli: vi.fn(),
|
spawnCli: vi.fn(),
|
||||||
|
execCli: vi.fn(),
|
||||||
killProcessTree: vi.fn(),
|
killProcessTree: vi.fn(),
|
||||||
}));
|
}));
|
||||||
|
|
||||||
|
|
@ -39,7 +40,7 @@ vi.mock('@main/utils/pathDecoder', async (importOriginal) => {
|
||||||
|
|
||||||
import { TeamProvisioningService } from '@main/services/team/TeamProvisioningService';
|
import { TeamProvisioningService } from '@main/services/team/TeamProvisioningService';
|
||||||
import { ClaudeBinaryResolver } from '@main/services/team/ClaudeBinaryResolver';
|
import { ClaudeBinaryResolver } from '@main/services/team/ClaudeBinaryResolver';
|
||||||
import { spawnCli } from '@main/utils/childProcess';
|
import { execCli, spawnCli } from '@main/utils/childProcess';
|
||||||
import { setAppDataBasePath } from '@main/utils/pathDecoder';
|
import { setAppDataBasePath } from '@main/utils/pathDecoder';
|
||||||
|
|
||||||
function createFakeChild() {
|
function createFakeChild() {
|
||||||
|
|
@ -78,6 +79,56 @@ async function setupRunningTeam(teamName: string) {
|
||||||
vi.mocked(ClaudeBinaryResolver.resolve).mockResolvedValue('/fake/claude');
|
vi.mocked(ClaudeBinaryResolver.resolve).mockResolvedValue('/fake/claude');
|
||||||
const { child, writeSpy } = createFakeChild();
|
const { child, writeSpy } = createFakeChild();
|
||||||
vi.mocked(spawnCli).mockReturnValue(child as any);
|
vi.mocked(spawnCli).mockReturnValue(child as any);
|
||||||
|
vi.mocked(execCli).mockImplementation(async (_binaryPath, args) => {
|
||||||
|
const providerIndex = args.indexOf('--provider');
|
||||||
|
const providerId = providerIndex >= 0 ? args[providerIndex + 1] : 'anthropic';
|
||||||
|
if (args[0] === 'model' && args[1] === 'list') {
|
||||||
|
return {
|
||||||
|
stdout: JSON.stringify({
|
||||||
|
providers: {
|
||||||
|
[providerId ?? 'anthropic']: {
|
||||||
|
defaultModel: providerId === 'codex' ? 'gpt-5.4' : 'opus[1m]',
|
||||||
|
models:
|
||||||
|
providerId === 'codex'
|
||||||
|
? ['gpt-5.4']
|
||||||
|
: ['opus[1m]', 'opus', 'claude-opus-4-6', 'sonnet', 'haiku'],
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
stderr: '',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
if (args[0] === 'runtime' && args[1] === 'status') {
|
||||||
|
return {
|
||||||
|
stdout: JSON.stringify({
|
||||||
|
providers: {
|
||||||
|
[providerId ?? 'anthropic']: {
|
||||||
|
runtimeCapabilities:
|
||||||
|
providerId === 'codex'
|
||||||
|
? {
|
||||||
|
reasoningEffort: {
|
||||||
|
supported: true,
|
||||||
|
values: ['minimal', 'low', 'medium', 'high', 'xhigh'],
|
||||||
|
configPassthrough: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
: {
|
||||||
|
fastMode: {
|
||||||
|
supported: false,
|
||||||
|
available: false,
|
||||||
|
reason: 'Test runtime does not expose fast mode.',
|
||||||
|
source: 'test',
|
||||||
|
},
|
||||||
|
},
|
||||||
|
modelCatalog: null,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}),
|
||||||
|
stderr: '',
|
||||||
|
};
|
||||||
|
}
|
||||||
|
return { stdout: '{}', stderr: '' };
|
||||||
|
});
|
||||||
|
|
||||||
const svc = new TeamProvisioningService();
|
const svc = new TeamProvisioningService();
|
||||||
(svc as any).buildProvisioningEnv = vi.fn(async () => ({
|
(svc as any).buildProvisioningEnv = vi.fn(async () => ({
|
||||||
|
|
|
||||||
|
|
@ -25,7 +25,7 @@ async function createSubagentLog(
|
||||||
return filePath;
|
return filePath;
|
||||||
}
|
}
|
||||||
|
|
||||||
async function waitForCondition(check: () => void, attempts = 20): Promise<void> {
|
async function waitForCondition(check: () => void, attempts = 100): Promise<void> {
|
||||||
let lastError: unknown;
|
let lastError: unknown;
|
||||||
for (let attempt = 0; attempt < attempts; attempt += 1) {
|
for (let attempt = 0; attempt < attempts; attempt += 1) {
|
||||||
try {
|
try {
|
||||||
|
|
@ -161,6 +161,8 @@ describe('TeammateToolTracker', () => {
|
||||||
tracker.handleLogSourceChange('my-team');
|
tracker.handleLogSourceChange('my-team');
|
||||||
await waitForCondition(() => {
|
await waitForCondition(() => {
|
||||||
expect(events).toHaveLength(1);
|
expect(events).toHaveLength(1);
|
||||||
|
const fileState = (tracker as any).stateByTeam.get('my-team')?.filesByPath.get(filePath);
|
||||||
|
expect(fileState?.lineCarry).toBe(resultLine.slice(0, splitAt).trim());
|
||||||
});
|
});
|
||||||
|
|
||||||
await appendFile(filePath, `${resultLine.slice(splitAt)}\n`, 'utf8');
|
await appendFile(filePath, `${resultLine.slice(splitAt)}\n`, 'utf8');
|
||||||
|
|
|
||||||
|
|
@ -140,11 +140,41 @@ describe('teamModelAvailability', () => {
|
||||||
|
|
||||||
it('keeps both Anthropic Opus 4.7 and explicit Opus 4.6 in the fallback selector options', () => {
|
it('keeps both Anthropic Opus 4.7 and explicit Opus 4.6 in the fallback selector options', () => {
|
||||||
expect(getAvailableTeamProviderModelOptions('anthropic')).toEqual([
|
expect(getAvailableTeamProviderModelOptions('anthropic')).toEqual([
|
||||||
{ value: '', label: 'Default', badgeLabel: 'Default' },
|
{
|
||||||
{ value: 'opus', label: 'Opus 4.7', badgeLabel: 'Opus 4.7' },
|
value: '',
|
||||||
{ value: 'claude-opus-4-6', label: 'Opus 4.6', badgeLabel: 'Opus 4.6' },
|
label: 'Default',
|
||||||
{ value: 'sonnet', label: 'Sonnet 4.6', badgeLabel: 'Sonnet 4.6' },
|
badgeLabel: 'Default',
|
||||||
{ value: 'haiku', label: 'Haiku 4.5', badgeLabel: 'Haiku 4.5' },
|
availabilityStatus: undefined,
|
||||||
|
availabilityReason: undefined,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
value: 'opus',
|
||||||
|
label: 'Opus 4.7',
|
||||||
|
badgeLabel: 'Opus 4.7',
|
||||||
|
availabilityStatus: 'available',
|
||||||
|
availabilityReason: null,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
value: 'claude-opus-4-6',
|
||||||
|
label: 'Opus 4.6',
|
||||||
|
badgeLabel: 'Opus 4.6',
|
||||||
|
availabilityStatus: 'available',
|
||||||
|
availabilityReason: null,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
value: 'sonnet',
|
||||||
|
label: 'Sonnet 4.6',
|
||||||
|
badgeLabel: 'Sonnet 4.6',
|
||||||
|
availabilityStatus: 'available',
|
||||||
|
availabilityReason: null,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
value: 'haiku',
|
||||||
|
label: 'Haiku 4.5',
|
||||||
|
badgeLabel: 'Haiku 4.5',
|
||||||
|
availabilityStatus: 'available',
|
||||||
|
availabilityReason: null,
|
||||||
|
},
|
||||||
]);
|
]);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue