Contents

module FactoryBot
  class Registry
    include Enumerable

    attr_reader :name

    def initialize(name)
      @name  = name
      @items = Decorator::ClassKeyHash.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

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
factory_bot-4.11.1	lib/factory_bot/registry.rb
factory_bot-4.11.0	lib/factory_bot/registry.rb
factory_bot-4.10.0	lib/factory_bot/registry.rb
factory_bot-4.8.2	lib/factory_bot/registry.rb
factory_bot-1.0.1.alpha	lib/factory_bot/registry.rb
factory_bot-1.0.0.alpha	lib/factory_bot/registry.rb