Sha256: 6ee24a476340c292dfb7f29f322c192999272abf8c7066b20795c34e7a7bce84
Contents?: true
Size: 608 Bytes
Versions: 16
Compression:
Stored size: 608 Bytes
Contents
module FactoryBot
# @api private
class Enum
def initialize(attribute_name, values = nil)
@attribute_name = attribute_name
@values = values
end
def build_traits(klass)
enum_values(klass).map do |trait_name, value|
build_trait(trait_name, @attribute_name, value || trait_name)
end
end
private
def enum_values(klass)
@values || klass.send(@attribute_name.to_s.pluralize)
end
def build_trait(trait_name, attribute_name, value)
Trait.new(trait_name) do
add_attribute(attribute_name) { value }
end
end
end
end
Version data entries
16 entries across 16 versions & 2 rubygems