-
Notifications
You must be signed in to change notification settings - Fork 22
/
studio.spec.ts
114 lines (97 loc) · 3.14 KB
/
studio.spec.ts
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/**********************************************************************
* Copyright (C) 2024 Red Hat, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* SPDX-License-Identifier: Apache-2.0
***********************************************************************/
/* eslint-disable @typescript-eslint/no-explicit-any */
import { afterEach, beforeEach, expect, test, vi } from 'vitest';
import { Studio } from './studio';
import { type ExtensionContext, EventEmitter } from '@podman-desktop/api';
import * as fs from 'node:fs';
vi.mock('./managers/modelsManager');
const mockedExtensionContext = {
subscriptions: [],
} as unknown as ExtensionContext;
const studio = new Studio(mockedExtensionContext);
const mocks = vi.hoisted(() => ({
listContainers: vi.fn(),
getContainerConnections: vi.fn(),
postMessage: vi.fn(),
}));
vi.mock('@podman-desktop/api', async () => {
return {
fs: {
createFileSystemWatcher: vi.fn(),
},
EventEmitter: vi.fn(),
Uri: class {
static joinPath = () => ({ fsPath: '.' });
},
window: {
createWebviewPanel: () => ({
webview: {
html: '',
onDidReceiveMessage: vi.fn(),
postMessage: mocks.postMessage,
},
onDidChangeViewState: vi.fn(),
}),
},
env: {
createTelemetryLogger: () => ({
logUsage: vi.fn(),
}),
},
containerEngine: {
onEvent: vi.fn(),
listContainers: mocks.listContainers,
},
provider: {
onDidRegisterContainerConnection: vi.fn(),
onDidUpdateContainerConnection: vi.fn(),
getContainerConnections: mocks.getContainerConnections,
},
};
});
/// mock console.log
const originalConsoleLog = console.log;
const consoleLogMock = vi.fn();
beforeEach(() => {
vi.clearAllMocks();
console.log = consoleLogMock;
vi.mocked(EventEmitter).mockReturnValue({
event: vi.fn(),
fire: vi.fn(),
} as unknown as EventEmitter<unknown>);
mocks.postMessage.mockResolvedValue(undefined);
});
afterEach(() => {
console.log = originalConsoleLog;
});
test('check activate ', async () => {
mocks.listContainers.mockReturnValue([]);
mocks.getContainerConnections.mockReturnValue([]);
vi.spyOn(fs.promises, 'readFile').mockImplementation(() => {
return Promise.resolve('<html></html>');
});
await studio.activate();
// expect the activate method to be called on the studio class
expect(consoleLogMock).toBeCalledWith('starting studio extension');
});
test('check deactivate ', async () => {
await studio.deactivate();
// expect the deactivate method to be called on the studio class
expect(consoleLogMock).toBeCalledWith('stopping studio extension');
});