# frozen_string_literal: true

require_relative '../../test_helper'

class TestFakerIdNumber < Test::Unit::TestCase
  def setup
    @tester = Faker::IdNumber
  end

  def test_valid_ssn
    sample = @tester.valid

    assert_equal(11, sample.length)
    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_equal(11, sample.length)
    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 Faker::IdNumber::CHECKS[mod], sample[9]
  end

  def test_spanish_nie
    sample = @tester.spanish_foreign_citizen_number

    assert_equal 11, sample.length
    assert_includes 'XYZ', 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 Faker::IdNumber::CHECKS[mod], sample[10]
  end

  def test_south_african_id_number
    assert_valid_south_african_id_number(@tester.south_african_id_number)
  end

  def test_valid_south_african_id_number
    assert_valid_south_african_id_number(@tester.valid_south_african_id_number)
  end

  def test_invalid_south_african_id_number
    sample = @tester.invalid_south_african_id_number

    assert_raises Date::Error do
      Date.parse(south_african_id_number_to_date_of_birth_string(sample))
    end
  end

  def test_brazilian_citizen_number
    sample = @tester.brazilian_citizen_number

    assert_match(/^\d{11}$/, sample)
    assert_match(/(\d)((?!\1)\d)+/, sample)
  end

  def test_brazilian_citizen_number_formatted
    sample = @tester.brazilian_citizen_number(formatted: true)

    assert_match(/^\d{3}\.\d{3}\.\d{3}-\d{2}$/, sample)
  end

  def test_brazilian_id
    sample = @tester.brazilian_id

    assert_match(/^\d{8}(X|\d)?$/, sample)
    assert_match(/(\d)((?!\1)\d)+/, sample)
  end

  def test_brazilian_id_formatted
    sample = @tester.brazilian_id(formatted: true)

    assert_match(/^\d{1,2}.\d{3}.\d{3}-[\dX]$/, sample)
  end

  def test_brazilian_citizen_number_checksum_digit
    digits = '128991760'
    checksum_digit = Faker::IdNumber.send(:brazilian_citizen_number_checksum_digit, digits)

    assert_equal('4', checksum_digit)
    digits = '1289917604'
    checksum_digit = Faker::IdNumber.send(:brazilian_citizen_number_checksum_digit, digits)

    assert_equal('8', checksum_digit)
  end

  def test_brazilian_id_checksum_digit
    digits = '41987080'
    checksum_digit = Faker::IdNumber.send(:brazilian_id_checksum_digit, digits)

    assert_equal('5', checksum_digit)
  end

  def test_brazilian_document_checksum
    digits = '123456789'
    checksum = Faker::IdNumber.send(:brazilian_document_checksum, digits)

    assert_equal(2100, checksum)
  end

  def test_brazilian_document_digit
    citizen_number_digit10 = Faker::IdNumber.send(:brazilian_document_digit, 10)
    citizen_number_digit_other = Faker::IdNumber.send(:brazilian_document_digit, 9)
    id_digit10 = Faker::IdNumber.send(:brazilian_document_digit, 1, id: true)
    id_digit11 = Faker::IdNumber.send(:brazilian_document_digit, 0, id: true)
    id_digit_other = Faker::IdNumber.send(:brazilian_document_digit, 2, id: true)

    assert_equal('0', citizen_number_digit10)
    assert_equal('9', citizen_number_digit_other)
    assert_equal('X', id_digit10)
    assert_equal('0', id_digit11)
    assert_equal('9', id_digit_other)
  end

  def test_brazilian_citizen_number_digit
    digit10 = Faker::IdNumber.send(:brazilian_citizen_number_digit, 10)
    digit_other = Faker::IdNumber.send(:brazilian_citizen_number_digit, 9)

    assert_equal('0', digit10)
    assert_equal('9', digit_other)
  end

  def test_brazilian_id_digit
    digit10 = Faker::IdNumber.send(:brazilian_id_digit, 1)
    digit11 = Faker::IdNumber.send(:brazilian_id_digit, 0)
    digit_other = Faker::IdNumber.send(:brazilian_id_digit, 2)

    assert_equal('X', digit10)
    assert_equal('0', digit11)
    assert_equal('9', digit_other)
  end

  def test_chilean_id
    sample = @tester.chilean_id

    assert_match(/^\d{8}-[K\d]$/, sample)
  end

  def test_chilean_verification_code_k
    verification_code = Faker::IdNumber.send(:chilean_verification_code, 20_680_873)

    assert_equal('K', verification_code)
  end

  def test_chilean_verification_code_0
    verification_code = Faker::IdNumber.send(:chilean_verification_code, 13_196_022)

    assert_equal(0, verification_code)
  end

  def test_croatian_id
    sample = @tester.croatian_id

    assert_match(/^\d{11}$/, sample)
  end

  def test_croatian_id_international
    sample = @tester.croatian_id(international: true)

    assert_match(/^HR\d{11}$/, sample)
  end

  def test_croatian_id_checksum_digit
    digits = '8764670153'
    checksum_digit = Faker::IdNumber.send(:croatian_id_checksum_digit, digits)

    assert_equal(5, checksum_digit)
  end

  def test_danish_id_number
    sample = @tester.danish_id_number

    assert_match(/^\d{10}$/, sample)
  end

  def test_french_insee_number
    sample = @tester.french_insee_number

    assert_match(/^(?<gnd>\d{1})(?<year>\d{2})(?<month>\d{2})(?<department1>\d{1})(?<department2>[0-9AB]{1})(?<place>\d{3})(?<indv>\d{3})(?<ctrl>\d{2})$/, sample)
  end

  def test_danish_id_number_formatted
    sample = @tester.danish_id_number(formatted: true)

    assert_match(/^\d{6}-\d{4}$/, sample)
  end

  def test_danish_id_number_birthday
    sample = @tester.danish_id_number(birthday: Date.new(1995, 1, 2))

    assert_match(/^020195\d{4}$/, sample)
  end

  def test_danish_id_number_birthday_early_1800
    assert_raises ArgumentError do
      @tester.danish_id_number(birthday: Date.new(1815, 1, 2))
    end
  end

  def test_danish_id_number_birthday_late_1800
    sample = @tester.danish_id_number(birthday: Date.new(1895, 1, 2))

    assert_match(/^020195[5678]\d{3}$/, sample)
  end

  def test_danish_id_number_birthday_early_1900
    sample = @tester.danish_id_number(birthday: Date.new(1915, 1, 2))

    assert_match(/^020115[0123]\d{3}$/, sample)
  end

  def test_danish_id_number_birthday_late_1900
    sample = @tester.danish_id_number(birthday: Date.new(1995, 1, 2))

    assert_match(/^020195[012349]\d{3}$/, sample)
  end

  def test_danish_id_number_birthday_early_2000
    sample = @tester.danish_id_number(birthday: Date.new(2015, 1, 2))

    assert_match(/^020115[456789]\d{3}$/, sample)
  end

  def test_danish_id_number_birthday_mid_2000
    sample = @tester.danish_id_number(birthday: Date.new(2055, 1, 2))

    assert_match(/^020155[5678]\d{3}$/, sample)
  end

  def test_danish_id_number_birthday_late_2000
    assert_raises ArgumentError do
      @tester.danish_id_number(birthday: Date.new(2095, 1, 2))
    end
  end

  def test_danish_id_number_gender_female
    sample = @tester.danish_id_number(gender: :female)

    assert_predicate sample.chars.last.to_i, :even?
  end

  def test_danish_id_number_gender_male
    sample = @tester.danish_id_number(gender: :male)

    assert_predicate sample.chars.last.to_i, :odd?
  end

  def test_danish_id_number_invalid_gender
    assert_raises ArgumentError do
      @tester.danish_id_number(gender: :invalid)
    end
  end

  private

  def south_african_id_number_to_date_of_birth_string(sample)
    "19#{sample[0..1]}/#{sample[2..3]}/#{sample[4..5]}}"
  end

  def assert_valid_south_african_id_number(sample)
    assert_equal 13, sample.length
    assert_match(/^\d{13}$/, sample)
    assert_include Faker::IdNumber::ZA_CITIZENSHIP_DIGITS, sample[10]
    assert_equal Faker::IdNumber::ZA_RACE_DIGIT, sample[11]
    assert Date.parse(south_african_id_number_to_date_of_birth_string(sample))
  end
end

class TestFakerIDNumber < Test::Unit::TestCase
  def setup
    Faker::Deprecator.skip_warning do
      @tester = Faker::IDNumber
    end
  end

  def test_valid_ssn
    sample = @tester.valid

    assert_equal(11, sample.length)
    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_equal(11, sample.length)
    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 Faker::IdNumber::CHECKS[mod], sample[9]
  end

  def test_spanish_nie
    sample = @tester.spanish_foreign_citizen_number

    assert_equal 11, sample.length
    assert_includes 'XYZ', 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 Faker::IdNumber::CHECKS[mod], sample[10]
  end

  def test_south_african_id_number
    assert_valid_south_african_id_number(@tester.south_african_id_number)
  end

  def test_valid_south_african_id_number
    assert_valid_south_african_id_number(@tester.valid_south_african_id_number)
  end

  def test_invalid_south_african_id_number
    sample = @tester.invalid_south_african_id_number

    assert_raises Date::Error do
      Date.parse(south_african_id_number_to_date_of_birth_string(sample))
    end
  end

  def test_brazilian_citizen_number
    sample = @tester.brazilian_citizen_number

    assert_match(/^\d{11}$/, sample)
    assert_match(/(\d)((?!\1)\d)+/, sample)
  end

  def test_brazilian_citizen_number_formatted
    sample = @tester.brazilian_citizen_number(formatted: true)

    assert_match(/^\d{3}\.\d{3}\.\d{3}-\d{2}$/, sample)
  end

  def test_brazilian_id
    sample = @tester.brazilian_id

    assert_match(/^\d{8}(X|\d)?$/, sample)
    assert_match(/(\d)((?!\1)\d)+/, sample)
  end

  def test_brazilian_id_formatted
    sample = @tester.brazilian_id(formatted: true)

    assert_match(/^\d{1,2}.\d{3}.\d{3}-[\dX]$/, sample)
  end

  def test_brazilian_citizen_number_checksum_digit
    digits = '128991760'
    checksum_digit = Faker::IdNumber.send(:brazilian_citizen_number_checksum_digit, digits)

    assert_equal('4', checksum_digit)
    digits = '1289917604'
    checksum_digit = Faker::IdNumber.send(:brazilian_citizen_number_checksum_digit, digits)

    assert_equal('8', checksum_digit)
  end

  def test_brazilian_id_checksum_digit
    digits = '41987080'
    checksum_digit = Faker::IdNumber.send(:brazilian_id_checksum_digit, digits)

    assert_equal('5', checksum_digit)
  end

  def test_brazilian_document_checksum
    digits = '123456789'
    checksum = Faker::IdNumber.send(:brazilian_document_checksum, digits)

    assert_equal(2100, checksum)
  end

  def test_brazilian_document_digit
    citizen_number_digit10 = Faker::IdNumber.send(:brazilian_document_digit, 10)
    citizen_number_digit_other = Faker::IdNumber.send(:brazilian_document_digit, 9)
    id_digit10 = Faker::IdNumber.send(:brazilian_document_digit, 1, id: true)
    id_digit11 = Faker::IdNumber.send(:brazilian_document_digit, 0, id: true)
    id_digit_other = Faker::IdNumber.send(:brazilian_document_digit, 2, id: true)

    assert_equal('0', citizen_number_digit10)
    assert_equal('9', citizen_number_digit_other)
    assert_equal('X', id_digit10)
    assert_equal('0', id_digit11)
    assert_equal('9', id_digit_other)
  end

  def test_brazilian_citizen_number_digit
    digit10 = Faker::IdNumber.send(:brazilian_citizen_number_digit, 10)
    digit_other = Faker::IdNumber.send(:brazilian_citizen_number_digit, 9)

    assert_equal('0', digit10)
    assert_equal('9', digit_other)
  end

  def test_brazilian_id_digit
    digit10 = Faker::IdNumber.send(:brazilian_id_digit, 1)
    digit11 = Faker::IdNumber.send(:brazilian_id_digit, 0)
    digit_other = Faker::IdNumber.send(:brazilian_id_digit, 2)

    assert_equal('X', digit10)
    assert_equal('0', digit11)
    assert_equal('9', digit_other)
  end

  def test_chilean_id
    sample = @tester.chilean_id

    assert_match(/^\d{8}-[K\d]$/, sample)
  end

  def test_chilean_verification_code_k
    verification_code = Faker::IdNumber.send(:chilean_verification_code, 20_680_873)

    assert_equal('K', verification_code)
  end

  def test_chilean_verification_code_0
    verification_code = Faker::IdNumber.send(:chilean_verification_code, 13_196_022)

    assert_equal(0, verification_code)
  end

  def test_croatian_id
    sample = @tester.croatian_id

    assert_match(/^\d{11}$/, sample)
  end

  def test_croatian_id_international
    sample = @tester.croatian_id(international: true)

    assert_match(/^HR\d{11}$/, sample)
  end

  def test_croatian_id_checksum_digit
    digits = '8764670153'
    checksum_digit = Faker::IdNumber.send(:croatian_id_checksum_digit, digits)

    assert_equal(5, checksum_digit)
  end

  def test_danish_id_number
    sample = @tester.danish_id_number

    assert_match(/^\d{10}$/, sample)
  end

  def test_french_insee_number
    sample = @tester.french_insee_number

    assert_match(/^(?<gnd>\d{1})(?<year>\d{2})(?<month>\d{2})(?<department1>\d{1})(?<department2>[0-9AB]{1})(?<place>\d{3})(?<indv>\d{3})(?<ctrl>\d{2})$/, sample)
  end

  def test_danish_id_number_formatted
    sample = @tester.danish_id_number(formatted: true)

    assert_match(/^\d{6}-\d{4}$/, sample)
  end

  def test_danish_id_number_birthday
    sample = @tester.danish_id_number(birthday: Date.new(1995, 1, 2))

    assert_match(/^020195\d{4}$/, sample)
  end

  def test_danish_id_number_birthday_early_1800
    assert_raises ArgumentError do
      @tester.danish_id_number(birthday: Date.new(1815, 1, 2))
    end
  end

  def test_danish_id_number_birthday_late_1800
    sample = @tester.danish_id_number(birthday: Date.new(1895, 1, 2))

    assert_match(/^020195[5678]\d{3}$/, sample)
  end

  def test_danish_id_number_birthday_early_1900
    sample = @tester.danish_id_number(birthday: Date.new(1915, 1, 2))

    assert_match(/^020115[0123]\d{3}$/, sample)
  end

  def test_danish_id_number_birthday_late_1900
    sample = @tester.danish_id_number(birthday: Date.new(1995, 1, 2))

    assert_match(/^020195[012349]\d{3}$/, sample)
  end

  def test_danish_id_number_birthday_early_2000
    sample = @tester.danish_id_number(birthday: Date.new(2015, 1, 2))

    assert_match(/^020115[456789]\d{3}$/, sample)
  end

  def test_danish_id_number_birthday_mid_2000
    sample = @tester.danish_id_number(birthday: Date.new(2055, 1, 2))

    assert_match(/^020155[5678]\d{3}$/, sample)
  end

  def test_danish_id_number_birthday_late_2000
    assert_raises ArgumentError do
      @tester.danish_id_number(birthday: Date.new(2095, 1, 2))
    end
  end

  def test_danish_id_number_gender_female
    sample = @tester.danish_id_number(gender: :female)

    assert_predicate sample.chars.last.to_i, :even?
  end

  def test_danish_id_number_gender_male
    sample = @tester.danish_id_number(gender: :male)

    assert_predicate sample.chars.last.to_i, :odd?
  end

  def test_danish_id_number_invalid_gender
    assert_raises ArgumentError do
      @tester.danish_id_number(gender: :invalid)
    end
  end

  private

  def south_african_id_number_to_date_of_birth_string(sample)
    "19#{sample[0..1]}/#{sample[2..3]}/#{sample[4..5]}}"
  end

  def assert_valid_south_african_id_number(sample)
    assert_equal 13, sample.length
    assert_match(/^\d{13}$/, sample)
    assert_include Faker::IdNumber::ZA_CITIZENSHIP_DIGITS, sample[10]
    assert_equal Faker::IdNumber::ZA_RACE_DIGIT, sample[11]
    assert Date.parse(south_african_id_number_to_date_of_birth_string(sample))
  end
end
