41 lines
679 B
Ruby
41 lines
679 B
Ruby
require "active_support/core_ext/hash/indifferent_access"
|
|
|
|
module FactoryBot
|
|
class Registry
|
|
include Enumerable
|
|
|
|
attr_reader :name
|
|
|
|
def initialize(name)
|
|
@name = name
|
|
@items = ActiveSupport::HashWithIndifferentAccess.new
|
|
end
|
|
|
|
def clear
|
|
@items.clear
|
|
end
|
|
|
|
def each(&block)
|
|
@items.values.uniq.each(&block)
|
|
end
|
|
|
|
def find(name)
|
|
if registered?(name)
|
|
@items[name]
|
|
else
|
|
raise ArgumentError, "#{@name} not registered: #{name}"
|
|
end
|
|
end
|
|
|
|
alias :[] :find
|
|
|
|
def register(name, item)
|
|
@items[name] = item
|
|
end
|
|
|
|
def registered?(name)
|
|
@items.key?(name)
|
|
end
|
|
end
|
|
end
|