forked from square/cube
-
Notifications
You must be signed in to change notification settings - Fork 1
/
test.js
76 lines (65 loc) · 2.18 KB
/
test.js
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
var mongodb = require("mongodb"),
assert = require("assert"),
util = require("util"),
http = require("http");
exports.port = 1083;
exports.batch = function(batch) {
return {
"": {
topic: function() {
var client = new mongodb.Server("localhost", 27017);
db = new mongodb.Db("cube_test", client),
cb = this.callback;
db.open(function(error) {
var collectionsRemaining = 2;
db.dropCollection("test_events", function(error) {
db.createCollection("test_events", function(error, events) {
events.ensureIndex({t: 1}, collectionReady);
});
});
db.dropCollection("test_metrics", function(error) {
db.createCollection("test_metrics", {capped: true, size: 1e6, autoIndexId: false}, function(error, metrics) {
var indexesRemaining = 3;
metrics.ensureIndex({e: 1, l: 1, t: 1, g: 1}, {unique: true}, indexReady);
metrics.ensureIndex({i: 1, e: 1, l: 1, t: 1}, indexReady);
metrics.ensureIndex({i: 1, l: 1, t: 1}, indexReady);
function indexReady() {
if (!--indexesRemaining) {
collectionReady();
}
}
});
});
function collectionReady() {
if (!--collectionsRemaining) {
cb(null, {client: client, db: db});
}
}
});
},
"": batch,
teardown: function(test) {
if (test.client.isConnected()) {
test.client.close();
}
}
}
};
};
exports.request = function(options, data) {
return function() {
var cb = this.callback;
options.host = "localhost";
var request = http.request(options, function(response) {
response.body = "";
response.setEncoding("utf8");
response.on("data", function(chunk) { response.body += chunk; });
response.on("end", function() { cb(null, response); });
});
request.on("error", function(e) { cb(e, null); });
if (data && data.length > 0) request.write(data);
request.end();
};
};
// Disable logging for tests.
util.log = function() {};