thoughtbot--factory_bot/lib/factory_girl/factory.rb

require "active_support/core_ext/hash/keys"
require "active_support/inflector"

module FactoryGirl
  class Factory
    attr_reader :name, :definition #:nodoc:

    def initialize(name, options = {}) #:nodoc:
      assert_valid_options(options)
      @name             = name.is_a?(Symbol) ? name : name.to_s.underscore.to_sym
      @parent           = options[:parent]
      @aliases          = options[:aliases] || []
      @class_name       = options[:class]
      @default_strategy = options[:default_strategy]
      @definition       = Definition.new(@name)

      inherit_traits(options[:traits] || [])
    end

    delegate :add_callback, :declare_attribute, :to_create, :define_trait,
             :defined_traits, :traits, :inherit_traits, :to => :@definition

    def factory_name
      $stderr.puts "DEPRECATION WARNING: factory.factory_name is deprecated; use factory.name instead."
      name
    end

    def build_class #:nodoc:
      class_name.to_s.camelize.constantize
    end

    def default_strategy #:nodoc:
      @default_strategy || parent.default_strategy
    end

    def run(proxy_class, overrides, &block) #:nodoc:
      block ||= lambda {|result| result }

      runner_options = {
        :attributes  => attributes,
        :callbacks   => callbacks,
        :to_create   => to_create,
        :build_class => build_class,
        :proxy_class => proxy_class,
        :overrides   => overrides.dup
      }

      block[Runner.new(runner_options).run]
    end

    def human_names
      names.map {|name| name.to_s.humanize.downcase }
    end

    def associations
      attributes.select {|attribute| attribute.association? }
    end

    # Names for this factory, including aliases.
    #
    # Example:
    #
    #   factory :user, :aliases => [:author] do
    #     # ...
    #   end
    #
    #   FactoryGirl.create(:author).class
    #   # => User
    #
    # Because an attribute defined without a value or block will build an
    # association with the same name, this allows associations to be defined
    # without factories, such as:
    #
    #   factory :user, :aliases => [:author] do
    #     # ...
    #   end
    #
    #   factory :post do
    #     author
    #   end
    #
    #   FactoryGirl.create(:post).author.class
    #   # => User
    def names
      [name] + @aliases
    end

    def compile
      parent.defined_traits.each {|trait| define_trait(trait) }
      parent.compile
      @definition.compile
    end

    def with_traits(traits)
      self.clone.tap do |factory_with_traits|
        factory_with_traits.inherit_traits traits
      end
    end

    protected

    def class_name #:nodoc:
      @class_name || parent.class_name || name

    end

    def attributes
      compile
      AttributeList.new(@name).tap do |list|
        traits.each do |trait|
          list.apply_attributes(trait.attributes)
        end

        list.apply_attributes(@definition.attributes)
        list.apply_attributes(parent.attributes)
      end
    end

    def callbacks
      [parent.callbacks, traits.map(&:callbacks).reverse, @definition.callbacks].flatten
    end

    private

    def assert_valid_options(options)
      options.assert_valid_keys(:class, :parent, :default_strategy, :aliases, :traits)

      if options[:default_strategy]
        Proxy.ensure_strategy_exists!(options[:default_strategy])
        $stderr.puts "DEPRECATION WARNING: default_strategy is deprecated."
        $stderr.puts "Override to_create if you need to prevent a call to #save!."
      end
    end

    def parent
      if @parent
        FactoryGirl.factory_by_name(@parent)
      else
        NullFactory.new
      end
    end

    def initialize_copy(source)
      super
      @definition = @definition.clone
    end

    class Runner
      def initialize(options = {})
        @attributes  = options[:attributes]
        @callbacks   = options[:callbacks]
        @to_create   = options[:to_create]
        @build_class = options[:build_class]
        @proxy_class = options[:proxy_class]
        @overrides   = options[:overrides]
      end

      def run
        apply_attributes
        apply_remaining_overrides

        proxy.result(@to_create)
      end

      private

      def apply_attributes
        @attributes.each do |attribute|
          if overrides_for_attribute(attribute).any?
            handle_attribute_with_overrides(attribute)
          else
            handle_attribute_without_overrides(attribute)
          end
        end
      end

      def apply_remaining_overrides
        @overrides.each { |attr, val| add_static_attribute(attr, val) }
      end

      def overrides_for_attribute(attribute)
        @overrides.select { |attr, val| attribute.alias_for?(attr) }
      end

      def handle_attribute_with_overrides(attribute)
        overrides_for_attribute(attribute).each do |attr, val|
          add_static_attribute(attr, val, attribute.ignored)
          @overrides.delete(attr)
        end
      end

      def add_static_attribute(attr, val, ignored = false)
        proxy.set(Attribute::Static.new(attr, val, ignored))
      end

      def handle_attribute_without_overrides(attribute)
        proxy.set(attribute)
      end

      def proxy
        @proxy ||= @proxy_class.new(@build_class, @callbacks)
      end
    end
  end
end