brState() tests

2 files changed, 12 insertions, 6 deletions

diff --git a/lib/address.js b/lib/address.js
index d1892748..9b5a63c0 100644
--- a/lib/address.js
+++ b/lib/address.js
@@ -61,18 +61,18 @@ exports.secondaryAddress = function () {
     ));
 };
 
-exports.brState = function (abbr) {
-    return Helpers.randomize(definitions[abbr ? 'br_state_abbr' : 'br_state']());
+exports.brState = function (useAbbr) {
+    return useAbbr ? random.br_state_abbr() : random.br_state();
 };
 
 exports.ukCounty = function () {
-    return Helpers.randomize(definitions.uk_county());
+    return random.uk_county();
 };
 
 exports.ukCountry = function () {
-    return Helpers.randomize(definitions.uk_country());
+    return random.uk_country();
 };
 
-exports.usState = function (abbr) {
-    return Helpers.randomize(definitions[abbr ? 'us_state_abbr' : 'us_state']());
+exports.usState = function (useAbbr) {
+    return useAbbr ? random.us_state_abbr() : random.us_state();
 };
diff --git a/lib/random.js b/lib/random.js
index f1a4dfbd..e7a9b667 100644
--- a/lib/random.js
+++ b/lib/random.js
@@ -21,6 +21,12 @@ var method_names = [
     'city_prefix',
     'city_suffix',
     'street_suffix',
+    'br_state',
+    'br_state_abbr',
+    'us_state',
+    'us_state_abbr',
+    'uk_county',
+    'uk_country',
 
     // Name
     'first_name',