shared_examples_for 'is a database with an upsert trick' do
  describe :row do
    it "works for a single row (base case)" do
      upsert = Upsert.new connection, :pets
      assert_creates(Pet, [{:name => 'Jerry', :gender => 'male'}]) do
        upsert.row({:name => 'Jerry'}, {:gender => 'male'})
      end
    end
    it "works for complex selectors" do
      upsert = Upsert.new connection, :pets
      assert_creates(Pet, [{:name => 'Jerry', :gender => 'male', :tag_number => 4}]) do
        upsert.row({:name => 'Jerry', :gender => 'male'}, {:tag_number => 1})
        upsert.row({:name => 'Jerry', :gender => 'male'}, {:tag_number => 4})
      end
    end
    it "doesn't nullify columns that are not included in the selector or document" do
      assert_creates(Pet, [{:name => 'Jerry', :gender => 'male', :tag_number => 4}]) do
        one = Upsert.new connection, :pets
        one.row({:name => 'Jerry'}, {:gender => 'male'})
        two = Upsert.new connection, :pets
        two.row({:name => 'Jerry'}, {:tag_number => 4})
      end
    end
    it "works for a single row (not changing anything)" do
      upsert = Upsert.new connection, :pets
      assert_creates(Pet, [{:name => 'Jerry', :gender => 'male'}]) do
        upsert.row({:name => 'Jerry'}, {:gender => 'male'})
        upsert.row({:name => 'Jerry'}, {:gender => 'male'})
      end
    end
    it "works for a single row (changing something)" do
      upsert = Upsert.new connection, :pets
      assert_creates(Pet, [{:name => 'Jerry', :gender => 'neutered'}]) do
        upsert.row({:name => 'Jerry'}, {:gender => 'male'})
        upsert.row({:name => 'Jerry'}, {:gender => 'neutered'})
      end
      Pet.where(:gender => 'male').count.must_equal 0
    end
    it "works for a single row with implicit nulls" do
      upsert = Upsert.new connection, :pets
      assert_creates(Pet, [{:name => 'Inky', :gender => nil}]) do
        upsert.row({:name => 'Inky'}, {})
        upsert.row({:name => 'Inky'}, {})
      end
    end
    it "works for a single row with empty document" do
      upsert = Upsert.new connection, :pets
      assert_creates(Pet, [{:name => 'Inky', :gender => nil}]) do
        upsert.row(:name => 'Inky')
        upsert.row(:name => 'Inky')
      end
    end
    it "works for a single row with explicit nulls" do
      upsert = Upsert.new connection, :pets
      assert_creates(Pet, [{:name => 'Inky', :gender => nil}]) do
        upsert.row({:name => 'Inky'}, {:gender => nil})
        upsert.row({:name => 'Inky'}, {:gender => nil})
      end
    end
    it "works with ids" do
      jerry = Pet.create :name => 'Jerry', :lovability => 1.0
      upsert = Upsert.new connection, :pets
      assert_creates(Pet, [{:name => 'Jerry', :lovability => 2.0}]) do
        upsert.row({:id => jerry.id}, :lovability => 2.0)
      end
    end
  end
  describe :batch do
    it "works for multiple rows (base case)" do
      assert_creates(Pet, [{:name => 'Jerry', :gender => 'male'}]) do
        Upsert.batch(connection, :pets) do |upsert|
          upsert.row({:name => 'Jerry'}, :gender => 'male')
        end
      end
    end
    it "works for multiple rows (not changing anything)" do
      assert_creates(Pet, [{:name => 'Jerry', :gender => 'male'}]) do
        Upsert.batch(connection, :pets) do |upsert|
          upsert.row({:name => 'Jerry'}, :gender => 'male')
          upsert.row({:name => 'Jerry'}, :gender => 'male')
        end
      end
    end
    it "works for multiple rows (changing something)" do
      assert_creates(Pet, [{:name => 'Jerry', :gender => 'neutered'}]) do
        Upsert.batch(connection, :pets) do |upsert|
          upsert.row({:name => 'Jerry'}, :gender => 'male')
          upsert.row({:name => 'Jerry'}, :gender => 'neutered')
        end
      end
      Pet.where(:gender => 'male').count.must_equal 0
    end
  end
end