Skip to content

Commit

Permalink
Prevent ENETUNREACH errors from being thrown in MetadataService test
Browse files Browse the repository at this point in the history
  • Loading branch information
@jeskew
jeskew committed May 4, 2017
1 parent 6c58abf commit a860695
Showing 1 changed file with 145 additions and 145 deletions.
290 changes: 145 additions & 145 deletions test/metadata_service.spec.js
View file
Original file line number Diff line number Diff line change
@@ -1,166 +1,166 @@
// Generated by CoffeeScript 1.12.3
(function() {
var AWS, helpers, http, url;
var AWS, helpers, http, url;

helpers = require('./helpers');
helpers = require('./helpers');

url = require('url');
url = require('url');

http = require('http');
http = require('http');

AWS = helpers.AWS;
AWS = helpers.AWS;

if (AWS.util.isNode()) {
describe('AWS.MetadataService', function() {
return describe('loadCredentials', function() {
var forceTimeout, httpClient, port, ref, server, service;
ref = [null, 1024 + parseInt(Math.random() * 100), null, null], server = ref[0], port = ref[1], service = ref[2], forceTimeout = ref[3];
httpClient = AWS.HttpClient.getInstance();
beforeEach(function() {
forceTimeout = false;
helpers.spyOn(http.ClientRequest.prototype, 'setTimeout').andCallFake(function(timeout, cb) {
if (forceTimeout) {
return process.nextTick(cb);
}
});
service = new AWS.MetadataService({
host: '127.0.0.1:' + port
});
server = http.createServer(function(req, res) {
var data, match, re;
re = new RegExp('^/latest/meta-data/iam/security-credentials/(.*)$');
match = url.parse(req.url).pathname.match(re);
if (match) {
res.writeHead(200, {
'Content-Type': 'text/plain'
});
if (match[1] === '') {
res.write('TestingRole\n');
res.write('TestingRole2\n');
} else {
data = '{"Code":"Success","AccessKeyId":"KEY","SecretAccessKey":"SECRET","Token":"TOKEN"}';
res.write(data);
}
if (AWS.util.isNode()) {
describe('AWS.MetadataService', function() {
describe('loadCredentials', function() {
var forceTimeout, httpClient, port, ref, server, service;
ref = [null, 1024 + parseInt(Math.random() * 100), null, null], server = ref[0], port = ref[1], service = ref[2], forceTimeout = ref[3];
httpClient = AWS.HttpClient.getInstance();

beforeEach(function() {
forceTimeout = false;
helpers.spyOn(http.ClientRequest.prototype, 'setTimeout').andCallFake(function(timeout, cb) {
if (forceTimeout) {
return process.nextTick(cb);
}
});
service = new AWS.MetadataService({
host: '127.0.0.1:' + port
});
server = http.createServer(function(req, res) {
var data, match, re;
re = new RegExp('^/latest/meta-data/iam/security-credentials/(.*)$');
match = url.parse(req.url).pathname.match(re);
if (match) {
res.writeHead(200, {
'Content-Type': 'text/plain'
});
if (match[1] === '') {
res.write('TestingRole\n');
res.write('TestingRole2\n');
} else {
res.writeHead(404, {});
data = '{"Code":"Success","AccessKeyId":"KEY","SecretAccessKey":"SECRET","Token":"TOKEN"}';
res.write(data);
}
return res.end();
});
return server.listen(port);
});
afterEach(function() {
if (server) {
return server.close();
} else {
res.writeHead(404, {});
}
res.end();
});
it('should load credentials from metadata service', function(done) {
return service.loadCredentials(function(err, data) {
server.listen(port);
});

afterEach(function() {
if (server) {
server.close();
}
});

it('should load credentials from metadata service', function(done) {
service.loadCredentials(function(err, data) {
expect(err).to.equal(null);
expect(data.Code).to.equal('Success');
expect(data.AccessKeyId).to.equal('KEY');
expect(data.SecretAccessKey).to.equal('SECRET');
expect(data.Token).to.equal('TOKEN');
done();
});
});

it('should load credentials from metadata service', function(done) {
var concurrency, countdown, i, ref1, results, x;
concurrency = countdown = 10;
results = [];
for (x = i = 1, ref1 = concurrency; 1 <= ref1 ? i <= ref1 : i >= ref1; x = 1 <= ref1 ? ++i : --i) {
results.push(service.loadCredentials(function(err, data) {
expect(err).to.equal(null);
expect(data.Code).to.equal('Success');
expect(data.AccessKeyId).to.equal('KEY');
expect(data.SecretAccessKey).to.equal('SECRET');
expect(data.Token).to.equal('TOKEN');
return done();
});
countdown--;
if (countdown === 0) {
done();
}
}));
}
return results;
});

it('should fail if server is not up', function(done) {
server.close();
server = null;
service.loadCredentials(function(err, data) {
expect(err).to.be.instanceOf(Error);
expect(data).not.to.exist;
done();
});
it('should load credentials from metadata service', function(done) {
var concurrency, countdown, i, ref1, results, x;
concurrency = countdown = 10;
results = [];
for (x = i = 1, ref1 = concurrency; 1 <= ref1 ? i <= ref1 : i >= ref1; x = 1 <= ref1 ? ++i : --i) {
results.push(service.loadCredentials(function(err, data) {
expect(err).to.equal(null);
expect(data.Code).to.equal('Success');
expect(data.AccessKeyId).to.equal('KEY');
expect(data.SecretAccessKey).to.equal('SECRET');
expect(data.Token).to.equal('TOKEN');
countdown--;
if (countdown === 0) {
return done();
}
}));
});

it('should retry when request times out', function(done) {
var firstTry, options, spy;
options = {
host: '127.0.0.1:' + port,
maxRetries: 5
};
firstTry = true;
spy = helpers.spyOn(httpClient, 'handleRequest').andCallFake(function() {
if (firstTry) {
forceTimeout = true;
firstTry = false;
} else {
forceTimeout = false;
}
return results;
spy.origMethod.apply(httpClient, arguments);
});
it('should fail if server is not up', function(done) {
server.close();
server = null;
service = new AWS.MetadataService({
host: '255.255.255.255'
});
service.httpOptions.timeout = 10;
return service.loadCredentials(function(err, data) {
expect(err).to.be.instanceOf(Error);
expect(data).not.to.exist;
return done();
});
service = new AWS.MetadataService(options);
service.loadCredentials(function(err, data) {
expect(err).to.be["null"];
expect(data.AccessKeyId).to.equal('KEY');
expect(spy.calls.length).to.equal(3);
done();
});
it('should retry when request times out', function(done) {
var firstTry, options, spy;
options = {
host: '127.0.0.1:' + port,
maxRetries: 5
};
firstTry = true;
spy = helpers.spyOn(httpClient, 'handleRequest').andCallFake(function() {
if (firstTry) {
forceTimeout = true;
firstTry = false;
} else {
forceTimeout = false;
}
return spy.origMethod.apply(httpClient, arguments);
});
service = new AWS.MetadataService(options);
return service.loadCredentials(function(err, data) {
expect(err).to.be["null"];
expect(data.AccessKeyId).to.equal('KEY');
expect(spy.calls.length).to.equal(3);
return done();
});
});
it('should retry up to the specified maxRetries when requests time out', function(done) {
var options, spy;
options = {
host: '127.0.0.1:' + port,
maxRetries: 5
};
forceTimeout = true;
spy = helpers.spyOn(httpClient, 'handleRequest').andCallThrough();
service = new AWS.MetadataService(options);
return service.loadCredentials(function(err, data) {
expect(data).to.be.undefined;
expect(err).to.not.be["null"];
expect(err.code).to.equal('TimeoutError');
expect(spy.calls.length).to.equal(6);
return done();
});
});
return it('makes only one pair of requests when multiple calls are made before first one finishes', function(done) {
var concurrency, countdown, i, options, ref1, results, services, spy, x;
options = {
host: '127.0.0.1:' + port
};
spy = helpers.spyOn(httpClient, 'handleRequest').andCallThrough();
concurrency = countdown = 10;
services = [];
results = [];
for (x = i = 1, ref1 = concurrency; 1 <= ref1 ? i <= ref1 : i >= ref1; x = 1 <= ref1 ? ++i : --i) {
services[x - 1] = new AWS.MetadataService(options);
results.push(services[x - 1].loadCredentials(function(err, data) {
expect(err).to.equal(null);
expect(data.AccessKeyId).to.equal('KEY');
countdown--;
if (countdown === 0) {
expect(spy.calls.length).to.equal(2);
return done();
}
}));
}
return results;
});

it('should retry up to the specified maxRetries when requests time out', function(done) {
var options, spy;
options = {
host: '127.0.0.1:' + port,
maxRetries: 5
};
forceTimeout = true;
spy = helpers.spyOn(httpClient, 'handleRequest').andCallThrough();
service = new AWS.MetadataService(options);
service.loadCredentials(function(err, data) {
expect(data).to.be.undefined;
expect(err).to.not.be["null"];
expect(err.code).to.equal('TimeoutError');
expect(spy.calls.length).to.equal(6);
done();
});
});
});
}

}).call(this);
it('makes only one pair of requests when multiple calls are made before first one finishes', function(done) {
var concurrency, countdown, i, options, ref1, results, services, spy, x;
options = {
host: '127.0.0.1:' + port
};
spy = helpers.spyOn(httpClient, 'handleRequest').andCallThrough();
concurrency = countdown = 10;
services = [];
results = [];
for (x = i = 1, ref1 = concurrency; 1 <= ref1 ? i <= ref1 : i >= ref1; x = 1 <= ref1 ? ++i : --i) {
services[x - 1] = new AWS.MetadataService(options);
results.push(services[x - 1].loadCredentials(function(err, data) {
expect(err).to.equal(null);
expect(data.AccessKeyId).to.equal('KEY');
countdown--;
if (countdown === 0) {
expect(spy.calls.length).to.equal(2);
return done();
}
}));
}
return results;
});
});
});
}

0 comments on commit a860695

Please sign in to comment.