* Added unit tests as per request in PR comment.

1 files changed, 31 insertions, 0 deletions

diff --git a/test/address.unit.js b/test/address.unit.js
index f6b299a1..03411d25 100644
--- a/test/address.unit.js
+++ b/test/address.unit.js
@@ -216,6 +216,37 @@ describe("address.js", function () {
         });
     });
 
+    describe("zipCode()", function () {
+        it("returns random zipCode", function () {
+            sinon.spy(faker.address, 'zipCode');
+            var zipCode = faker.address.zipCode();
+
+            assert.ok(zipCode);
+            assert.ok(faker.address.zipCode.called);
+
+            faker.address.zipCode.restore();
+        });
+
+        it("returns random zipCode - user specified format", function () {
+            var zipCode = faker.address.zipCode("?#? #?#");
+            assert.ok(zipCode);
+            assert.match(zipCode, /^[A-Za-z]\d[A-Za-z]\s\d[A-Za-z]\d$/, 'Expected zip code format does not match.');
+            // try another format
+            zipCode = faker.address.zipCode("###-###");
+            assert.ok(zipCode);
+            assert.match(zipCode, /^\d{3}-\d{3}$/, 'Expected zip code format does not match.');
+        });
+
+        it("returns zipCode with proper locale format", function () {
+            // we'll use the en_CA locale..
+            faker.locale = "en_CA";
+            var zipCode = faker.address.zipCode();
+
+            assert.ok(zipCode);
+            assert.match(zipCode, /^[A-Za-z]\d[A-Za-z]\s?\d[A-Za-z]\d$/, 'Expected zip code format for en_CA locale does not match.');
+        });
+    });
+
     describe("latitude()", function () {
         it("returns random latitude", function () {
             for (var i = 0; i < 100; i++) {