vscode-python/src/test/activation/defaultLanguageServer.unit.test.ts at 9b8a62e8a436329767fc5d1e3b581c53adb3c455 · kinto0/vscode-python · GitHub

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
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

'use strict';

import { expect } from 'chai';
import { anything, instance, mock, when, verify } from 'ts-mockito';
import { Extension } from 'vscode';
import { setDefaultLanguageServer } from '../../client/activation/common/defaultlanguageServer';
import { LanguageServerType } from '../../client/activation/types';
import { PYLANCE_EXTENSION_ID, PYREFLY_EXTENSION_ID } from '../../client/common/constants';
import { IDefaultLanguageServer, IExtensions } from '../../client/common/types';
import { ServiceManager } from '../../client/ioc/serviceManager';
import { IServiceManager } from '../../client/ioc/types';

suite('Activation - setDefaultLanguageServer()', () => {
    let extensions: IExtensions;
    let extension: Extension<unknown>;
    let serviceManager: IServiceManager;
    setup(() => {
        extensions = mock();
        extension = mock();
        serviceManager = mock(ServiceManager);
    });

    test('Pylance not installed', async () => {
        let defaultServerType;

        when(extensions.getExtension(PYLANCE_EXTENSION_ID)).thenReturn(undefined);
        when(serviceManager.addSingletonInstance<IDefaultLanguageServer>(IDefaultLanguageServer, anything())).thenCall(
            (_symbol, value: IDefaultLanguageServer) => {
                defaultServerType = value.defaultLSType;
            },
        );

        await setDefaultLanguageServer(instance(extensions), instance(serviceManager));

        verify(extensions.getExtension(PYLANCE_EXTENSION_ID)).once();
        verify(serviceManager.addSingletonInstance<IDefaultLanguageServer>(IDefaultLanguageServer, anything())).once();
        expect(defaultServerType).to.deep.equal({type: "always", languageServerType: LanguageServerType.Jedi});
    });

    test('Pylance installed', async () => {
        let defaultServerType;

        when(extensions.getExtension(PYLANCE_EXTENSION_ID)).thenReturn(instance(extension));
        when(serviceManager.addSingletonInstance<IDefaultLanguageServer>(IDefaultLanguageServer, anything())).thenCall(
            (_symbol, value: IDefaultLanguageServer) => {
                defaultServerType = value.defaultLSType;
            },
        );

        await setDefaultLanguageServer(instance(extensions), instance(serviceManager));

        verify(extensions.getExtension(PYLANCE_EXTENSION_ID)).once();
        verify(serviceManager.addSingletonInstance<IDefaultLanguageServer>(IDefaultLanguageServer, anything())).once();
        expect(defaultServerType).to.deep.equal({type: "always", languageServerType: LanguageServerType.Node});
    });

    test('Pyrefly installed', async () => {
        let defaultServerType;

        when(extensions.getExtension(PYREFLY_EXTENSION_ID)).thenReturn(instance(extension));
        when(serviceManager.addSingletonInstance<IDefaultLanguageServer>(IDefaultLanguageServer, anything())).thenCall(
            (_symbol, value: IDefaultLanguageServer) => {
                defaultServerType = value.defaultLSType;
            },
        );

        await setDefaultLanguageServer(instance(extensions), instance(serviceManager));

        verify(extensions.getExtension(PYREFLY_EXTENSION_ID)).once();
        verify(serviceManager.addSingletonInstance<IDefaultLanguageServer>(IDefaultLanguageServer, anything())).once();
        expect(defaultServerType).to.deep.equal({type: "none or (if pyrefly language services disabled)", languageServerType: LanguageServerType.Jedi});
    });
});