19 lines
352 B
Ruby
19 lines
352 B
Ruby
module FactoryBot
|
|
class << self
|
|
attr_accessor :aliases
|
|
end
|
|
|
|
self.aliases = [
|
|
[/(.+)_id/, '\1'],
|
|
[/(.*)/, '\1_id'],
|
|
]
|
|
|
|
def self.aliases_for(attribute)
|
|
aliases.map do |(pattern, replace)|
|
|
if pattern.match(attribute.to_s)
|
|
attribute.to_s.sub(pattern, replace).to_sym
|
|
end
|
|
end.compact << attribute
|
|
end
|
|
end
|