• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1// Copyright (c) 2014 The WebRTC project authors. All Rights Reserved.
2//
3// Use of this source code is governed by a BSD-style license
4// that can be found in the LICENSE file in the root of the source
5// tree. An additional intellectual property rights grant can be found
6// in the file PATENTS.  All contributing project authors may
7// be found in the AUTHORS file in the root of the source tree.
8//
9// Provides a Test class that exposes api to the tests.
10// Read test.prototype to see what methods are exposed.
11var fs = require('fs');
12var request = require('request');
13var BotManager = require('./botmanager.js');
14
15function Test() {
16  this.timeout_ = setTimeout(
17      this.fail.bind(this, "Test timeout!"),
18      100000);
19}
20
21Test.prototype = {
22  log: function () {
23    console.log.apply(console.log, arguments);
24  },
25
26  abort: function (error) {
27    var error = new Error(error || "Test aborted");
28    console.log(error.stack);
29    process.exit(1);
30  },
31
32  assert: function (value, message) {
33    if (value !== true) {
34      this.abort(message || "Assert failed.");
35    }
36  },
37
38  fail: function () {
39    this.assert(false, "Test failed.");
40  },
41
42  done: function () {
43    clearTimeout(this.timeout_);
44    console.log("Test succeeded");
45    process.exit(0);
46  },
47
48  // Utility method to wait for multiple callbacks to be executed.
49  //  functions - array of functions to call with a callback.
50  //  doneCallback - called when all callbacks on the array have completed.
51  wait: function (functions, doneCallback) {
52    var result = new Array(functions.length);
53    var missingResult = functions.length;
54    for (var i = 0; i != functions.length; ++i)
55      functions[i](complete.bind(this, i));
56
57    function complete(index, value) {
58      missingResult--;
59      result[index] = value;
60      if (missingResult == 0)
61        doneCallback.apply(null, result);
62    }
63  },
64
65  spawnBot: function (name, botType, doneCallback) {
66    // Lazy initialization of botmanager.
67    if (!this.botManager_)
68      this.botManager_ = new BotManager();
69    this.botManager_.spawnNewBot(name, botType, doneCallback);
70  },
71
72  createStatisticsReport: function (outputFileName) {
73    return new StatisticsReport(outputFileName);
74  },
75
76  // Ask computeengineondemand to give us TURN server credentials and URIs.
77  createTurnConfig: function (onSuccess, onError) {
78    request('https://computeengineondemand.appspot.com/turn?username=1234&key=5678',
79        function (error, response, body) {
80          if (error || response.statusCode != 200) {
81            onError('TURN request failed');
82            return;
83          }
84
85          var response = JSON.parse(body);
86          var iceServer = {
87            'username': response.username,
88            'credential': response.password,
89            'urls': response.uris
90          };
91          onSuccess({ 'iceServers': [ iceServer ] });
92        }
93      );
94  },
95}
96
97StatisticsReport = function (outputFileName) {
98  this.output_ = [];
99  this.output_.push("Version: 1");
100  this.outputFileName_ = outputFileName;
101}
102
103StatisticsReport.prototype = {
104  collectStatsFromPeerConnection: function (prefix, pc) {
105    setInterval(this.addPeerConnectionStats.bind(this, prefix, pc), 100);
106  },
107
108  addPeerConnectionStats: function (prefix, pc) {
109    pc.getStats(onStatsReady.bind(this));
110
111    function onStatsReady(reports) {
112      for (index in reports) {
113        var stats = {};
114        stats[reports[index].id] = collectStats(reports[index].stats);
115
116        var data = {};
117        data[prefix] = stats;
118
119        this.output_.push({
120            type: "UpdateCounters",
121            startTime: (new Date()).getTime(),
122            data: data,
123          });
124      }
125    };
126
127    function collectStats(stats) {
128      var outputStats = {};
129      for (index in stats) {
130        var statValue = parseFloat(stats[index].stat);
131        outputStats[stats[index].name] = isNaN(statValue)?
132            stats[index].stat : statValue;
133      }
134      return outputStats;
135    };
136  },
137
138  finish: function (doneCallback) {
139    fs.exists("test/reports/", function (exists) {
140      if(exists) {
141        writeFile.bind(this)();
142      } else {
143        fs.mkdir("test/reports/", 0777, writeFile.bind(this));
144      }
145    }.bind(this));
146
147    function writeFile () {
148      fs.writeFile("test/reports/" + this.outputFileName_ + "_" +
149        (new Date()).getTime() +".json", JSON.stringify(this.output_),
150        doneCallback);
151    }
152  },
153};
154
155module.exports = Test;
156