Merge pull request #37 from BryanDonovan/main

Refactored with 100% test coverage

1 files changed, 31 insertions, 32 deletions

diff --git a/lib/internet.js b/lib/internet.js
index e34dc48d..e08308d0 100644
--- a/lib/internet.js
+++ b/lib/internet.js
@@ -1,40 +1,39 @@
-var Helpers = require('./helpers');
-var definitions = require('./definitions');
+var Faker = require('../index');
 
-exports.email = function() {
-	return this.userName() + "@" + this.domainName();
-};
+var internet = {
+    email: function () {
+        return this.userName() + "@" + this.domainName();
+    },
 
-exports.userName = function() {
-	var ret;
-	switch (Helpers.randomNumber(2)) {
-		case 0:
-			ret = Helpers.randomize(definitions.first_name());
-			break;
-		case 1:
-			ret = Helpers.randomize(definitions.first_name()) + Helpers.randomize([".", "_"]) + Helpers.randomize(definitions.last_name());
-			break;
-	}
-	return ret;
-};
+    userName: function () {
+        switch (Faker.random.number(2)) {
+        case 0:
+            return Faker.random.first_name();
+        case 1:
+            return Faker.random.first_name() + Faker.random.array_element([".", "_"]) + Faker.random.last_name();
+        }
+    },
 
-exports.domainName = function() {
-	return this.domainWord() + "." + Helpers.randomize(definitions.domain_suffix());
-};
+    domainName: function () {
+        return this.domainWord() + "." + Faker.random.domain_suffix();
+    },
 
-exports.domainWord = function() {
-	return Helpers.randomize(definitions.first_name()).toLowerCase();
-};
+    domainWord:  function () {
+        return Faker.random.first_name().toLowerCase();
+    },
 
-exports.ip = function() {
-	var randNum = function() {
-		return (Math.random() * 254 + 1).toFixed(0);
-	};
+    ip: function () {
+        var randNum = function () {
+            return (Math.random() * 254 + 1).toFixed(0);
+        };
 
-	var result = [];
-	for (var i = 0; i < 4; i++) {
-		result[i] = randNum();
-	}
+        var result = [];
+        for (var i = 0; i < 4; i++) {
+            result[i] = randNum();
+        }
 
-	return result.join(".");
+        return result.join(".");
+    }
 };
+
+module.exports = internet;