Add Spanish citizen id and docs

README.md

@@ -69,6 +69,7 @@ Contents
  - [Faker::Hobbit](doc/hobbit.md)
  - [Faker::HowIMetYourMother](doc/how_i_met_your_mother.md)
  - [Faker::HitchhikersGuideToTheGalaxy](doc/hitchhikers_guide_to_the_galaxy.md)
+ - [Faker::IDNumber](doc/id_number.md)
  - [Faker::Internet](doc/internet.md)
  - [Faker::Job](doc/job.md)
  - [Faker::Kpop](doc/kpop.md)

doc/id_number.md

@@ -0,0 +1,15 @@
+# Faker::IDNumber
+
+```ruby
+# Generate a valid US Social Security number
+Faker::IDNumber.valid #=> "552-56-3593"
+
+# Generate an invalid US Social Security number
+Faker::IDNumber.invalid #=> "311-72-0000"
+
+# Generate a Spanish citizen identifier (DNI)
+Faker::IDNumber.spanish_citizen_number
+
+# Generate a Spanish foreign born citizen identifier (NIE)
+Faker::IDNumber.spanish_foreign_citizen_number
+```

lib/faker/id_number.rb

@@ -25,6 +25,25 @@ def ssn_valid
  INVALID_SSN.any? { |regex| regex =~ ssn } ? ssn_valid : ssn
  end
 
+ def spanish_citizen_number
+ checks = "TRWAGMYFPDXBNJZSQVHLCKE"
+ num = Faker::Number.number(8)
+ mod = num.to_i % 23
+ check = checks[mod]
+ "#{num}-#{check}"
+ end
+
+ def spanish_foreign_citizen_number
+ checks = "TRWAGMYFPDXBNJZSQVHLCKE"
+ code = "XYZ"
+ digits = Faker::Number.number(7)
+ prefix = code[rand(code.length)]
+ prefix_val = "XYZ".index(prefix).to_s
+ mod = ("#{prefix_val}#{digits.to_s}").to_i % 23
+ check = checks[mod]
+ "#{prefix}-#{digits}-#{check}"
+ end
+
  private
 
  def _translate(key)

test/test_faker_id_number.rb

@@ -0,0 +1,50 @@
+require File.expand_path(File.dirname(__FILE__) + '/test_helper.rb')
+
+class TestFakerIdNumber < Test::Unit::TestCase
+
+ def setup
+ @tester = Faker::IDNumber
+ end
+
+ def test_valid_ssn
+ sample = @tester.valid
+ assert sample.length == 11
+ assert_equal "-", sample[3]
+ assert_equal "-", sample[6]
+ assert sample[0..2].split().map{:to_i}.all?{:is_digit?}
+ assert sample[4..5].split().map{:to_i}.all?{:is_digit?}
+ assert sample[7..9].split().map{:to_i}.all?{:is_digit?}
+ end
+
+ def test_invalid_ssn
+ sample = @tester.invalid
+ assert sample.length == 11
+ assert_equal "-", sample[3]
+ assert_equal "-", sample[6]
+ assert sample[0..2].split().map{:to_i}.all?{:is_digit?}
+ assert sample[4..5].split().map{:to_i}.all?{:is_digit?}
+ assert sample[7..9].split().map{:to_i}.all?{:is_digit?}
+ end
+
+ def test_spanish_DNI
+ sample = @tester.spanish_citizen_number
+ assert_equal 10, sample.length
+ assert sample[0..7].split().map{:to_i}.all?{:is_digit?}
+ assert_equal sample[8], "-"
+ mod = sample[0..7].to_i % 23
+ assert_equal "TRWAGMYFPDXBNJZSQVHLCKE"[mod], sample[9]
+ end
+
+ def test_spanish_NIE
+ sample = @tester.spanish_foreign_citizen_number
+ assert_equal 11, sample.length
+ assert "XYZ".include?(sample[0])
+ assert_equal "-", sample[1]
+ assert sample[2..8].split().map{:to_i}.all?{:is_digit?}
+ assert_equal "-", sample[9]
+ prefix = "XYZ".index(sample[0]).to_s
+ mod = ("#{prefix}#{sample[2..8]}").to_i % 23
+ assert_equal "TRWAGMYFPDXBNJZSQVHLCKE"[mod], sample[10]
+ end
+
+end