Contents

module FactoryGirl
  class Registry
    include Enumerable

    def initialize
      @items = {}
    end

    def add(item)
      item.names.each { |name| add_as(name, item) }
      item
    end

    def find(name)
      @items[name.to_sym] or raise ArgumentError.new("Not registered: #{name.to_s}")
    end

    def each(&block)
      @items.values.each(&block)
    end

    def [](name)
      find(name)
    end

    def registered?(name)
      @items.key?(name.to_sym)
    end

    def clear
      @items.clear
    end

    private

    def add_as(name, item)
      if registered?(name)
        raise DuplicateDefinitionError, "Already defined: #{name}"
      else
        @items[name.to_sym] = item
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 2 rubygems

Version	Path
factory_girl-2.0.0.beta5	lib/factory_girl/registry.rb
factory_girl-2.0.0.beta4	lib/factory_girl/registry.rb
factory_girl-2.0.0.beta3	lib/factory_girl/registry.rb
factory_girl_kibiz0r-2.0.0.beta3	lib/factory_girl/registry.rb
factory_girl_kibiz0r-2.0.0.beta2	lib/factory_girl/registry.rb