mirror of
https://github.com/thoughtbot/factory_bot.git
synced 2022-11-09 11:43:51 -05:00
18 lines
348 B
Ruby
18 lines
348 B
Ruby
module FactoryBot
|
|
class << self
|
|
attr_accessor :aliases
|
|
end
|
|
|
|
self.aliases = [
|
|
[/(.+)_id/, '\1'],
|
|
[/(.*)/, '\1_id']
|
|
]
|
|
|
|
def self.aliases_for(attribute)
|
|
aliases.map { |(pattern, replace)|
|
|
if pattern.match(attribute.to_s)
|
|
attribute.to_s.sub(pattern, replace).to_sym
|
|
end
|
|
}.compact << attribute
|
|
end
|
|
end
|