/
finance.rb
102 lines (91 loc) · 3.27 KB
/
finance.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
# frozen_string_literal: true
module Faker
class Finance < Base
CREDIT_CARD_TYPES = %i[visa mastercard discover american_express
diners_club jcb switch solo dankort
maestro forbrugsforeningen laser].freeze
MARKET_LIST = %i[nyse nasdaq].freeze
class << self
##
# Produces a random credit card number.
#
# @param types [String] Specific credit card type.
# @return [String]
#
# @example
# Faker::Finance.credit_card #=> "3018-348979-1853"
# Faker::Finance.credit_card(:mastercard) #=> "6771-8921-2291-6236"
# Faker::Finance.credit_card(:mastercard, :visa) #=> "4448-8934-1277-7195"
#
# @faker.version 1.2.0
def credit_card(*types)
types = CREDIT_CARD_TYPES if types.empty?
type = sample(types)
template = numerify(fetch("finance.credit_card.#{type}"))
# calculate the luhn checksum digit
multiplier = 1
luhn_sum = template.gsub(/[^0-9]/, '').split('').reverse.map(&:to_i).inject(0) do |sum, digit|
multiplier = (multiplier == 2 ? 1 : 2)
sum + (digit * multiplier).to_s.split('').map(&:to_i).inject(0) { |digit_sum, cur| digit_sum + cur }
end
# the sum plus whatever the last digit is must be a multiple of 10. So, the
# last digit must be 10 - the last digit of the sum.
luhn_digit = (10 - (luhn_sum % 10)) % 10
template.gsub('L', luhn_digit.to_s)
end
##
# Produces a random vat number.
#
# @param country [String] Two capital letter country code to use for the vat number.
# @return [String]
#
# @example
# Faker::Finance.vat_number #=> "BR38.395.329/2471-83"
# Faker::Finance.vat_number('DE') #=> "DE593306671"
# Faker::Finance.vat_number('ZA') #=> "ZA79494416181"
#
# @faker.version 1.9.2
def vat_number(legacy_country = NOT_GIVEN, country: 'BR')
warn_for_deprecated_arguments do |keywords|
keywords << :country if legacy_country != NOT_GIVEN
end
numerify(fetch("finance.vat_number.#{country}"))
rescue I18n::MissingTranslationData
raise ArgumentError, "Could not find vat number for #{country}"
end
def vat_number_keys
translate('faker.finance.vat_number').keys
end
##
# Returns a randomly-selected stock ticker from a specified market.
#
# @param markets [String] The name of the market to choose the ticker from (e.g. NYSE, NASDAQ)
# @return [String]
#
# @example
# Faker::Finance.ticker #=> 'AMZN'
# Faker::Finance.vat_number('NASDAQ') #=> 'GOOG'
#
# @faker.version next
def ticker(*markets)
markets = MARKET_LIST if markets.empty?
market = sample(markets)
fetch("finance.ticker.#{market}")
rescue I18n::MissingTranslationData
raise ArgumentError, "Could not find market named #{market}"
end
##
# Returns a randomly-selected stock market.
#
# @return [String]
#
# @example
# Faker::Finance.stock_market #=> 'NASDAQ'
#
# @faker.version next
def stock_market
fetch('finance.stock_market')
end
end
end
end