Adding test for zipcode by state, basic functionality

3 files changed, 26 insertions, 1 deletions

diff --git a/lib/address.js b/lib/address.js
index 21e14ff6..64742f8a 100644
--- a/lib/address.js
+++ b/lib/address.js
@@ -26,6 +26,15 @@ function Address (faker) {
     return Helpers.replaceSymbols(format);
   }
 
+
+  this.zipCodeByState = function (state) {
+    var zipFormat = faker.definitions.address.postcode_by_state[state];
+    if (zipFormat) {
+      return this.zipCode(zipFormat);
+    }
+    return undefined;
+  }
+
   /**
    * Generates a random localized city name. The format string can contain any
    * method provided by faker wrapped in `{{}}`, e.g. `{{name.firstName}}` in
diff --git a/lib/index.js b/lib/index.js
index cd1a6cda..30ea96bc 100644
--- a/lib/index.js
+++ b/lib/index.js
@@ -101,7 +101,7 @@ function Faker (opts) {
 
   var _definitions = {
     "name": ["first_name", "last_name", "prefix", "suffix", "gender", "title", "male_first_name", "female_first_name", "male_middle_name", "female_middle_name", "male_last_name", "female_last_name"],
-    "address": ["city_prefix", "city_suffix", "street_suffix", "county", "country", "country_code", "state", "state_abbr", "street_prefix", "postcode", "direction", "direction_abbr"],
+    "address": ["city_prefix", "city_suffix", "street_suffix", "county", "country", "country_code", "state", "state_abbr", "street_prefix", "postcode", "postcode_by_state", "direction", "direction_abbr"],
     "company": ["adjective", "noun", "descriptor", "bs_adjective", "bs_noun", "bs_verb", "suffix"],
     "lorem": ["words"],
     "hacker": ["abbreviation", "adjective", "noun", "verb", "ingverb", "phrase"],
diff --git a/test/address.unit.js b/test/address.unit.js
index 28775850..a7bd2e1a 100644
--- a/test/address.unit.js
+++ b/test/address.unit.js
@@ -252,6 +252,22 @@ describe("address.js", function () {
         });
     });
 
+    describe.only("zipCodeByState()", function () {
+        it("returns zipCode valid for specified State", function () {
+          faker.locale = "en_US";
+          var state = "IL";
+          var zipCode = faker.address.zipCodeByState(state);
+
+          assert.ok(zipCode >= 60000);
+          assert.ok(zipCode <= 60099);
+        });
+
+        it("throws error if locale is invalid", function () {
+        });
+        it("throws error if state is invalid", function () {
+        });
+    });
+
     describe("latitude()", function () {
         it("returns random latitude", function () {
             for (var i = 0; i < 100; i++) {