Skip to content

Commit a5f9985

Browse files
committed
test: unit tests (to be continued)
Signed-off-by: Philippe Martin <[email protected]>
1 parent 628c119 commit a5f9985

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

packages/backend/src/managers/application/applicationManager.spec.ts

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ import { VMType } from '@shared/models/IPodman';
3131
import { POD_LABEL_MODEL_ID, POD_LABEL_RECIPE_ID } from '../../utils/RecipeConstants';
3232
import type { InferenceServer } from '@shared/models/IInference';
3333
import type { RpcExtension } from '@shared/messages/MessageProxy';
34+
import type { LlamaStackManager } from '../llama-stack/llamaStackManager';
3435

3536
const taskRegistryMock = {
3637
createTask: vi.fn(),
@@ -75,6 +76,10 @@ const recipeManager = {
7576
buildRecipe: vi.fn(),
7677
} as unknown as RecipeManager;
7778

79+
const llamaStackManager = {
80+
getLlamaStackContainer: vi.fn(),
81+
} as unknown as LlamaStackManager;
82+
7883
vi.mock('@podman-desktop/api', () => ({
7984
window: {
8085
withProgress: vi.fn(),
@@ -139,6 +144,11 @@ beforeEach(() => {
139144
id: 'fake-task',
140145
}));
141146
vi.mocked(modelsManagerMock.uploadModelToPodmanMachine).mockResolvedValue('downloaded-model-path');
147+
vi.mocked(llamaStackManager.getLlamaStackContainer).mockResolvedValue({
148+
containerId: 'container1',
149+
port: 10001,
150+
playgroundPort: 10002,
151+
});
142152
});
143153

144154
function getInitializedApplicationManager(): ApplicationManager {
@@ -151,6 +161,7 @@ function getInitializedApplicationManager(): ApplicationManager {
151161
telemetryMock,
152162
podManager,
153163
recipeManager,
164+
llamaStackManager,
154165
);
155166

156167
manager.init();

0 commit comments

Comments
 (0)