require 'spec_helper'
describe Upsert do
  describe "can be speeded up with upserting" do
    describe 'compared to native ActiveRecord' do
      it "is faster than new/set/save" do
        assert_faster_than 'find + new/set/save', lotsa_records do |records|
          records.each do |selector, setter|
            if pet = Pet.where(selector).first
              pet.update_attributes setter, :without_protection => true
            else
              pet = Pet.new
              selector.each do |k, v|
                pet.send "#{k}=", v
              end
              setter.each do |k, v|
                pet.send "#{k}=", v
              end
              pet.save!
            end
          end
        end
      end
      it "is faster than find_or_create + update_attributes" do
        assert_faster_than 'find_or_create + update_attributes', lotsa_records do |records|
          dynamic_method = nil
          records.each do |selector, setter|
            dynamic_method ||= "find_or_create_by_#{selector.keys.join('_or_')}"
            pet = Pet.send(dynamic_method, *selector.values)
            pet.update_attributes setter, :without_protection => true
          end
        end
      end
      it "is faster than create + rescue/find/update" do
        assert_faster_than 'create + rescue/find/update', lotsa_records do |records|
          dynamic_method = nil
          records.each do |selector, setter|
            dynamic_method ||= "find_or_create_by_#{selector.keys.join('_or_')}"
            begin
              Pet.create selector.merge(setter), :without_protection => true
            rescue
              pet = Pet.send(dynamic_method, *selector.values)
              pet.update_attributes setter, :without_protection => true
            end
          end
        end
      end
    end

    # FIXME apparently no longer faster?
    # if ENV['DB'] == 'mysql' && RUBY_VERSION >= '1.9'
    #   describe 'compared to activerecord-import' do
    #     it "is faster than faking upserts with activerecord-import" do
    #       assert_faster_than 'faking upserts with activerecord-import', lotsa_records do |records|
    #         columns = nil
    #         all_values = []
    #         records.each do |selector, setter|
    #           columns ||= (selector.keys + setter.keys).uniq
    #           all_values << columns.map do |k|
    #             if setter.has_key?(k)
    #               # prefer the setter so that you can change rows
    #               setter[k]
    #             else
    #               selector[k]
    #             end
    #           end
    #         end
    #         Pet.import columns, all_values, :timestamps => false, :on_duplicate_key_update => columns
    #       end
    #     end
    #   end
    # end

  end
end