File size: 2,887 Bytes
f0634fb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 | import { beforeEach, describe, expect, it } from 'vitest';
import { ScopeActivation } from '#/_base/di/instantiation';
import {
_clearScopedRegistryForTests,
registerScopedService,
} from '#/_base/di/scope';
import { createScopedTestHost, stubPair } from '#/_base/di/test';
import { IBootstrapService } from '#/app/bootstrap/bootstrap';
import { IFeatureManager } from '#/app/feature/featureManager';
import { FeatureManagerService } from '#/app/feature/featureManagerService';
import { LifecycleScope } from '#/app/scopes';
import { SessionInitFeature } from '#/features/sessionInit/sessionInitFeature';
import { ISessionInitService } from '#/features/sessionInit/sessionInit';
import { IHostEnvironment } from '#/os/interface/hostEnvironment';
import { IHostFileSystem } from '#/os/interface/hostFileSystem';
import { IFeatureAssemblyService } from '#/features/featureAssembly';
import { FeatureAssemblyService } from '#/features/featureAssemblyService';
import {
_clearFeatureRecipesForTests,
registerFeature,
} from '#/features/featureRegistry';
import { IAgentLifecycleService } from '#/session/agentLifecycle/agentLifecycle';
import { ISessionContext } from '#/session/sessionContext/sessionContext';
import { ISessionSubagentService } from '#/session/subagent/subagent';
describe('SessionInitFeature', () => {
beforeEach(() => {
_clearScopedRegistryForTests();
_clearFeatureRecipesForTests();
registerScopedService(
LifecycleScope.App,
IFeatureManager,
FeatureManagerService,
ScopeActivation.OnScopeCreated,
'feature',
);
registerScopedService(
LifecycleScope.App,
IFeatureAssemblyService,
FeatureAssemblyService,
ScopeActivation.OnScopeCreated,
'features',
);
registerFeature(SessionInitFeature);
});
it('withdraws and restores the Session service with the Feature', async () => {
const host = createScopedTestHost();
const session = host.child(LifecycleScope.Session, 'session-1', [
stubPair(IAgentLifecycleService, {} as IAgentLifecycleService),
stubPair(ISessionSubagentService, {} as ISessionSubagentService),
stubPair(IHostFileSystem, {} as IHostFileSystem),
stubPair(IHostEnvironment, {} as IHostEnvironment),
stubPair(IBootstrapService, {} as IBootstrapService),
stubPair(ISessionContext, {} as ISessionContext),
]);
const manager = host.app.accessor.get(IFeatureManager);
expect(session.accessor.get(ISessionInitService)).toBeDefined();
await manager.unprovideUnit('sessionInit');
await host.app.instantiation.cascade.whenIdle();
expect(() => session.accessor.get(ISessionInitService)).toThrow();
manager.provideUnit(SessionInitFeature);
await host.app.instantiation.cascade.whenIdle();
expect(session.accessor.get(ISessionInitService)).toBeDefined();
host.dispose();
});
});
|