class FactoryBot::Enum

def build_trait(trait_name, attribute_name, value)
  Trait.new(trait_name) do
    add_attribute(attribute_name) { value }
  end
end

def build_traits(klass)
  enum_values(klass).map do |trait_name, value|
    build_trait(trait_name, @attribute_name, value || trait_name)
  end
end

def enum_values(klass)
  @values || klass.send(@attribute_name.to_s.pluralize)
end

def initialize(attribute_name, values = nil)
  @attribute_name = attribute_name
  @values = values
end