vscode-python/src/test/terminals/codeExecution/terminalReplWatcher.unit.test.ts at 8ed74f27c1fb7fac8700cf5d3ee26e73e1eebd15 · microsoft/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
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
115
116
117
118
119
120
121
122
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.

'use strict';

import { expect } from 'chai';
import * as sinon from 'sinon';
import * as windowApis from '../../../client/common/vscodeApis/windowApis';
import * as telemetryModule from '../../../client/telemetry';
import { EventName } from '../../../client/telemetry/constants';
import { registerTriggerForTerminalREPL } from '../../../client/terminals/codeExecution/terminalReplWatcher';

suite('Terminal REPL Watcher', () => {
    let windowApisStub: sinon.SinonStub;
    let telemetryStub: sinon.SinonStub;

    setup(() => {
        windowApisStub = sinon.stub(windowApis, 'onDidStartTerminalShellExecution').returns({
            dispose: () => {
                // Do nothing
            },
        });
        telemetryStub = sinon.stub(telemetryModule, 'sendTelemetryEvent');
    });

    teardown(() => {
        sinon.restore();
    });

    test('Should send REPL telemetry when Python is invoked', () => {
        windowApisStub.callsFake((callback) => {
            callback({
                execution: {
                    commandLine: {
                        value: 'python script.py',
                        isTrusted: true,
                    },
                },
            });
            return {
                dispose: () => {
                    // Do nothing
                },
            };
        });

        registerTriggerForTerminalREPL([]);

        expect(telemetryStub.calledOnce).to.be.true;
        expect(telemetryStub.args[0][0]).to.equal(EventName.REPL);
        expect(telemetryStub.args[0][2]).to.deep.equal({ replType: 'runningScript' });
    });

    test('Should send unittest CLI telemetry when pytest is invoked', () => {
        windowApisStub.callsFake((callback) => {
            callback({
                execution: {
                    commandLine: {
                        value: 'python -m pytest',
                        isTrusted: true,
                    },
                },
            });
            return {
                dispose: () => {
                    // Do nothing
                },
            };
        });

        registerTriggerForTerminalREPL([]);

        expect(telemetryStub.calledOnce).to.be.true;
        expect(telemetryStub.args[0][0]).to.equal(EventName.UNITTEST_RUN_CLI);
    });

    test('Should send unittest CLI telemetry when unittest is invoked', () => {
        windowApisStub.callsFake((callback) => {
            callback({
                execution: {
                    commandLine: {
                        value: 'python -m unittest discover',
                        isTrusted: true,
                    },
                },
            });
            return {
                dispose: () => {
                    // Do nothing
                },
            };
        });

        registerTriggerForTerminalREPL([]);

        expect(telemetryStub.calledOnce).to.be.true;
        expect(telemetryStub.args[0][0]).to.equal(EventName.UNITTEST_RUN_CLI);
    });

    test('Should send unittest CLI telemetry when py.test is invoked', () => {
        windowApisStub.callsFake((callback) => {
            callback({
                execution: {
                    commandLine: {
                        value: 'py.test',
                        isTrusted: true,
                    },
                },
            });
            return {
                dispose: () => {
                    // Do nothing
                },
            };
        });

        registerTriggerForTerminalREPL([]);

        expect(telemetryStub.calledOnce).to.be.true;
        expect(telemetryStub.args[0][0]).to.equal(EventName.UNITTEST_RUN_CLI);
    });
});