Convert random generator to es6 and add some specs for it directly

This commit is contained in:
John Crepezzi 2017-10-31 20:40:43 -04:00
parent e12805a8aa
commit e4e025f67e
3 changed files with 37 additions and 17 deletions

View file

@ -1,19 +1,20 @@
var RandomKeyGenerator = function(options) { module.exports = class RandomKeyGenerator {
if (!options) {
options = {};
}
this.keyspace = options.keyspace || 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
};
// Generate a random key // Initialize a new generator with the given keySpace
RandomKeyGenerator.prototype.createKey = function(keyLength) { constructor(options = {}) {
this.keyspace = options.keyspace || 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';
}
// Generate a key of the given length
createKey(keyLength) {
var text = ''; var text = '';
var index;
for (var i = 0; i < keyLength; i++) { for (var i = 0; i < keyLength; i++) {
index = Math.floor(Math.random() * this.keyspace.length); const index = Math.floor(Math.random() * this.keyspace.length);
text += this.keyspace.charAt(index); text += this.keyspace.charAt(index);
} }
return text;
};
module.exports = RandomKeyGenerator; return text;
}
};

View file

@ -46,6 +46,6 @@
}, },
"scripts": { "scripts": {
"start": "node server.js", "start": "node server.js",
"test": "mocha" "test": "mocha --recursive"
} }
} }

View file

@ -0,0 +1,19 @@
/* global describe, it */
const assert = require('assert');
const Generator = require('../../lib/key_generators/random');
describe('RandomKeyGenerator', function() {
describe('randomKey', function() {
it('should return a key of the proper length', function() {
var gen = new Generator();
assert.equal(6, gen.createKey(6).length);
});
it('should use a key from the given keyset if given', () => {
var gen = new Generator({keyspace: 'A'});
assert.equal('AAAAAA', gen.createKey(6));
});
});
});