Add optional country code parameter for iban method

1 files changed, 11 insertions, 2 deletions

diff --git a/lib/finance.js b/lib/finance.js
index 19bc0c19..2939a659 100644
--- a/lib/finance.js
+++ b/lib/finance.js
@@ -248,10 +248,19 @@ self.litecoinAddress = function () {
   /**
    * iban
    *
+   * @param {boolean} [formatted=false] - Return a formatted version of the generated IBAN.
+   * @param {string} [countryCode] - The country code from which you want to generate an IBAN, if none is provided a random country will be used.
+   * @throws Will throw an error if the passed country code is not supported.
+   *
    * @method  faker.finance.iban
    */
-  self.iban = function (formatted) {
-      var ibanFormat = faker.random.arrayElement(ibanLib.formats);
+  self.iban = function (formatted, countryCode) {
+      var findFormat = function(currentFormat) { return currentFormat.country === countryCode; };
+      var ibanFormat = countryCode ? ibanLib.formats.find(findFormat) : faker.random.arrayElement(ibanLib.formats);
+      if (!ibanFormat) {
+          throw new Error('Country code ' + countryCode + ' not supported.');
+      }
+
       var s = "";
       var count = 0;
       for (var b = 0; b < ibanFormat.bban.length; b++) {