end.js
1.84 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
var util = require('util');
var events = require('events');
var Utils = require('../../util/utils.js');
var Logger = require('../../util/logger.js');
/**
* Ends the session. Uses session protocol command.
*
* ```
* this.demoTest = function (browser) {
* browser.end();
* };
* ```
*
* @method end
* @param {function} [callback] Optional callback function to be called when the command finishes.
* @see session
* @api commands
*/
function End() {
events.EventEmitter.call(this);
}
util.inherits(End, events.EventEmitter);
End.prototype.command = function(callback) {
var self = this;
var client = this.client;
if (client.sessionId) {
if (this.testFailuresExist() && this.shouldTakeScreenshot()) {
var fileNamePath = Utils.getScreenshotFileName(client.api.currentTest, false, client.options.screenshots.path);
Logger.info('We have failures in "' + client.api.currentTest.name + '". Taking screenshot...');
client.api.saveScreenshot(fileNamePath, function(result, err) {
if (err || result.status !== 0) {
Logger.warn('Error saving screenshot...', err || result);
}
});
}
client.api.session('delete', function(result) {
client.sessionId = client.api.sessionId = null;
self.complete(callback, result);
});
} else {
setImmediate(function() {
self.complete(callback, null);
});
}
return this.client.api;
};
End.prototype.testFailuresExist = function() {
return this.client.results.errors > 0 || this.client.results.failed > 0;
};
End.prototype.shouldTakeScreenshot = function() {
return this.client.options.screenshots.enabled && this.client.options.screenshots.on_failure;
};
End.prototype.complete = function(callback, result) {
if (typeof callback === 'function') {
callback.call(this, result);
}
this.emit('complete');
};
module.exports = End;