module ActiveMocker
module Mock

  class HasMany < Association

    include Queries

    def self.new(collection, foreign_key=nil, foreign_id=nil, relation_class=nil)
      return Relation.new(collection) if relation_class.nil?
      super
    end

    def initialize(collection, foreign_key, foreign_id, relation_class)
      @relation_class = relation_class
      @foreign_key    = foreign_key
      @foreign_id     = foreign_id
      self.class.include "#{relation_class.name}::Scopes".constantize
      super(collection)
      set_foreign_key if ActiveMocker::Mock.config.experimental
    end

    def set_foreign_key
      collection.each do |item|
        item.send(:write_attribute, foreign_key, foreign_id) if item.respond_to?("#{foreign_key}=")
      end
    end

    # @api private
    attr_reader :relation_class, :foreign_key, :foreign_id

    def build(options={}, &block)
      new_record = relation_class.new(init_options.merge!(options), &block)

      def new_record.belongs_to(collection)
        @belongs_to_collection = collection
      end

      new_record.belongs_to(self)

      def new_record.save
        @belongs_to_collection << self
        super
      end

      new_record
    end

    def create(options={}, &block)
      created_record = relation_class.create(init_options.merge!(options), &block)
      collection << created_record
      created_record
    end

    alias_method :create!, :create

    # @api private
    def init_options
      {foreign_key => foreign_id}
    end

  end

end
end