-
Notifications
You must be signed in to change notification settings - Fork 270
/
mock-run.ts
101 lines (89 loc) · 3.15 KB
/
mock-run.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
import ma = require('./mock-answer');
import im = require('./internal');
import mocker = require('./lib-mocker');
export class TaskMockRunner {
constructor(taskPath: string) {
this._taskPath = taskPath;
}
_taskPath: string;
_answers: ma.TaskLibAnswers | undefined;
_exports: {[key: string]: any} = { };
_moduleCount: number = 0;
public setInput(name: string, val: string) {
let key: string = im._getVariableKey(name);
process.env['INPUT_' + key] = val;
}
public setVariableName(name: string, val: string, isSecret?: boolean) {
let key: string = im._getVariableKey(name);
if (isSecret) {
process.env['SECRET_' + key] = val;
}
else {
process.env['VSTS_TASKVARIABLE_' + key] = val;
}
}
/**
* Register answers for the mock "azure-pipelines-task-lib/task" instance.
*
* @param answers Answers to be returned when the task lib functions are called.
*/
public setAnswers(answers: ma.TaskLibAnswers) {
this._answers = answers;
}
/**
* Register a mock module. When require() is called for the module name,
* the mock implementation will be returned instead.
*
* @param modName Module name to override.
* @param val Mock implementation of the module.
* @returns void
*/
public registerMock(modName: string, mod: any): void {
this._moduleCount++;
mocker.registerMock(modName, mod);
}
/**
* Registers an override for a specific function on the mock "azure-pipelines-task-lib/task" instance.
* This can be used in conjunction with setAnswers(), for cases where additional runtime
* control is needed for a specific function.
*
* @param key Function or field to override.
* @param val Function or field value.
* @returns void
*/
public registerMockExport(key: string, val: any): void {
this._exports[key] = val;
}
/**
* Runs a task script.
*
* @param noMockTask Indicates whether to mock "azure-pipelines-task-lib/task". Default is to mock.
* @returns void
*/
public run(noMockTask?: boolean): void {
// determine whether to enable mocker
if (!noMockTask || this._moduleCount) {
mocker.enable({warnOnUnregistered: false});
}
// answers and exports not compatible with "noMockTask" mode
if (noMockTask) {
if (this._answers || Object.keys(this._exports).length) {
throw new Error('setAnswers() and registerMockExport() is not compatible with "noMockTask" mode');
}
}
// register mock task lib
else {
var tlm = require('azure-pipelines-task-lib/mock-task');
if (this._answers) {
tlm.setAnswers(this._answers);
}
Object.keys(this._exports)
.forEach((key: string): void => {
tlm[key] = this._exports[key];
});
mocker.registerMock('azure-pipelines-task-lib/task', tlm);
}
// run it
require(this._taskPath);
}
}