random number now accepts a range array [min, max].

2 files changed, 28 insertions, 2 deletions

diff --git a/lib/random.js b/lib/random.js
index fcd451d7..63347278 100644
--- a/lib/random.js
+++ b/lib/random.js
@@ -1,8 +1,12 @@
 var definitions = require('./definitions');
-
 var random = {
-    // returns a single random number based on a range
+    // returns a single random number based on a max number or range
     number: function (range) {
+        if(Array.isArray(range)) {
+          var min = range[0];
+          var max = range[1];
+          return Math.floor((Math.random() * max) + min);
+        }
         return Math.floor(Math.random() * range);
     },
 
diff --git a/test/random.unit.js b/test/random.unit.js
new file mode 100644
index 00000000..3894e1d9
--- /dev/null
+++ b/test/random.unit.js
@@ -0,0 +1,22 @@
+if (typeof module !== 'undefined') {
+    var assert = require('assert');
+    var sinon = require('sinon');
+    var Faker = require('../index');
+}
+
+describe("random.js", function () {
+  describe("number", function() {
+    it("returns a random number given a maximum value", function() {
+      var max = 10;
+      assert.ok(Faker.random.number(max) < max);
+    });
+    it("returns a random number between a range", function() {
+      var range = [1, 10];
+      var min = range[0];
+      var max = range[1]
+      var randomNumber = Faker.random.number(range);
+      assert.ok( randomNumber >= min);
+      assert.ok( randomNumber <= max);
+    });
+  });
+});