forked from DonJayamanne/pythonVSCode
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathfixtures.ts
More file actions
87 lines (76 loc) · 2.28 KB
/
fixtures.ts
File metadata and controls
87 lines (76 loc) · 2.28 KB
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
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
import * as fs from 'fs-extra';
import { sleep } from '../client/common/utils/async';
import { PYTHON_PATH } from './common';
import { Proc, spawn } from './proc';
export type CleanupFunc = (() => void) | (() => Promise<void>);
export class CleanupFixture {
private cleanups: CleanupFunc[];
constructor() {
this.cleanups = [];
}
public addCleanup(cleanup: CleanupFunc) {
this.cleanups.push(cleanup);
}
public addFSCleanup(filename: string) {
this.addCleanup(async () => {
try {
await fs.unlink(filename);
} catch {
// The file is already gone.
}
});
}
public async cleanUp() {
const cleanups = this.cleanups;
this.cleanups = [];
return Promise.all(
cleanups.map(async (cleanup, i) => {
try {
const res = cleanup();
if (res) {
await res;
}
} catch (err) {
console.error(`cleanup ${i + 1} failed: ${err}`);
console.error('moving on...');
}
}),
);
}
}
export class PythonFixture extends CleanupFixture {
public readonly python: string;
constructor(
// If not provided, we will use the global default.
python?: string,
) {
super();
if (python) {
this.python = python;
} else {
this.python = PYTHON_PATH;
}
}
public runScript(filename: string, ...args: string[]): Proc {
return this.spawn(filename, ...args);
}
public runModule(name: string, ...args: string[]): Proc {
return this.spawn('-m', name, ...args);
}
private spawn(...args: string[]) {
const proc = spawn(this.python, ...args);
this.addCleanup(async () => {
if (!proc.exited) {
await sleep(1000); // Wait a sec before the hammer falls.
try {
proc.raw.kill();
} catch {
// It already finished.
}
}
});
return proc;
}
}