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,17 @@ def ssn_valid
  INVALID_SSN.any? { |regex| regex =~ ssn } ? ssn_valid : ssn
  end
 
+ def spanish_citizen_number
+ checks = "TRWAGMYFPDXBNJZSQVHLCKE"
+ "#{Faker::Number.number(8)}-#{checks[rand(checks.length)]}"
+ end
+
+ def spanish_foreign_citizen_number
+ checks = "TRWAGMYFPDXBNJZSQVHLCKE"
+ code = "XY"
+ "#{code[rand(code.length)]}-#{Faker::Number.number(7)}-#{checks[rand(checks.length)]}"
+ end
+
  private
 
  def _translate(key)

test/test_faker_id_number.rb

@@ -0,0 +1,47 @@
+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 sample[3] == "-"
+ assert 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 sample[3] == "-"
+ assert 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 sample.length == 10
+ assert sample[0..7].split().map{:to_i}.all?{:is_digit?}
+ assert sample[8] == "-"
+ assert "TRWAGMYFPDXBNJZSQVHLCKE".include?(sample[9])
+ end
+
+ def test_spanish_NIE
+ sample = @tester.spanish_foreign_citizen_number
+ assert sample.length == 11
+ assert "XY".include?(sample[0])
+ assert sample[1] == "-"
+ assert sample[2..8].split().map{:to_i}.all?{:is_digit?}
+ assert sample[9] == "-"
+ assert "TRWAGMYFPDXBNJZSQVHLCKE".include?(sample[10])
+ end
+
+end