| // Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. |
| // |
| // Use of this source code is governed by a BSD-style license |
| // that can be found in the LICENSE file in the root of the source |
| // tree. An additional intellectual property rights grant can be found |
| // in the file PATENTS. All contributing project authors may |
| // be found in the AUTHORS file in the root of the source tree. |
| // |
| // This script loads the test file in the virtual machine and runs it in a |
| // context that only exposes a test variable with methods for testing and to |
| // spawn bots. |
| // |
| // Note: an important part of this script is to keep nodejs-isms away from test |
| // code and isolate it from implementation details. |
| var fs = require('fs'); |
| var vm = require('vm'); |
| var BotManager = require('./botmanager.js'); |
| |
| function Test(botType) { |
| this.timeout_ = setTimeout( |
| this.fail.bind(this, "Test timeout!"), |
| 100000); |
| this.botType_ = botType; |
| } |
| |
| Test.prototype = { |
| log: function () { |
| console.log.apply(console.log, arguments); |
| }, |
| |
| abort: function (error) { |
| var error = new Error(error || "Test aborted"); |
| console.log(error.stack); |
| process.exit(1); |
| }, |
| |
| assert: function (value, message) { |
| if (value !== true) { |
| this.abort(message || "Assert failed."); |
| } |
| }, |
| |
| fail: function () { |
| this.assert(false, "Test failed."); |
| }, |
| |
| done: function () { |
| clearTimeout(this.timeout_); |
| console.log("Test succeeded"); |
| process.exit(0); |
| }, |
| |
| // Utility method to wait for multiple callbacks to be executed. |
| // functions - array of functions to call with a callback. |
| // doneCallback - called when all callbacks on the array have completed. |
| wait: function (functions, doneCallback) { |
| var result = new Array(functions.length); |
| var missingResult = functions.length; |
| for (var i = 0; i != functions.length; ++i) |
| functions[i](complete.bind(this, i)); |
| |
| function complete(index, value) { |
| missingResult--; |
| result[index] = value; |
| if (missingResult == 0) |
| doneCallback.apply(null, result); |
| } |
| }, |
| |
| spawnBot: function (name, doneCallback) { |
| // Lazy initialization of botmanager. |
| if (!this.botManager_) |
| this.botManager_ = new BotManager(); |
| this.botManager_.spawnNewBot(name, this.botType_, doneCallback); |
| }, |
| |
| createStatisticsReport: function (outputFileName) { |
| return new StatisticsReport(outputFileName); |
| }, |
| } |
| |
| StatisticsReport = function (outputFileName) { |
| this.output_ = []; |
| this.output_.push("Version: 1"); |
| this.outputFileName_ = outputFileName; |
| } |
| |
| StatisticsReport.prototype = { |
| collectStatsFromPeerConnection: function (prefix, pc) { |
| setInterval(this.addPeerConnectionStats.bind(this, prefix, pc), 100); |
| }, |
| |
| addPeerConnectionStats: function (prefix, pc) { |
| pc.getStats(onStatsReady.bind(this)); |
| |
| function onStatsReady(reports) { |
| for (index in reports) { |
| var stats = {}; |
| stats[reports[index].id] = collectStats(reports[index].stats); |
| |
| var data = {}; |
| data[prefix] = stats; |
| |
| this.output_.push({ |
| type: "UpdateCounters", |
| startTime: (new Date()).getTime(), |
| data: data, |
| }); |
| } |
| }; |
| |
| function collectStats(stats) { |
| var outputStats = {}; |
| for (index in stats) { |
| var statValue = parseFloat(stats[index].stat); |
| outputStats[stats[index].name] = isNaN(statValue)? |
| stats[index].stat : statValue; |
| } |
| return outputStats; |
| }; |
| }, |
| |
| finish: function (doneCallback) { |
| fs.writeFile("test/reports/" + this.outputFileName_ + "_" + |
| (new Date()).getTime() +".json", JSON.stringify(this.output_), |
| doneCallback); |
| }, |
| } |
| |
| function runTest(botType, testfile) { |
| console.log("Running test: " + testfile); |
| var script = vm.createScript(fs.readFileSync(testfile), testfile); |
| script.runInNewContext({ test: new Test(botType), setInterval: setInterval, |
| setTimeout: setTimeout }); |
| } |
| |
| runTest(process.argv[2], process.argv[3]); |