From 2abd905a6a35b16796224cfe2c1dd02be70890b3 Mon Sep 17 00:00:00 2001
From: Bryan Ricker <bricker88@gmail.com>
Date: Thu, 17 Oct 2013 23:46:53 -0700
Subject: [PATCH] Fix arity problem with rails 4.1.0

Fix specs for airty check

Don't check arity, just check the AR version
---
 .../models/active_record_relation_methods.rb  |  8 +++++--
 .../active_record_relation_methods_spec.rb    | 24 +++++++++++++++----
 2 files changed, 25 insertions(+), 7 deletions(-)

diff --git a/lib/kaminari/models/active_record_relation_methods.rb b/lib/kaminari/models/active_record_relation_methods.rb
index 972072d..3cbe842 100644
--- a/lib/kaminari/models/active_record_relation_methods.rb
+++ b/lib/kaminari/models/active_record_relation_methods.rb
@@ -16,12 +16,16 @@ module Kaminari
         # Remove includes only if they are irrelevant
         c = c.except(:includes) unless references_eager_loaded_tables?
 
+        # Rails 4.1 removes the `options` argument from AR::Relation#count
+        args = [column_name]
+        args << options if ActiveRecord::VERSION::STRING < '4.1.0'
+
         # .group returns an OrderdHash that responds to #count
-        c = c.count(column_name, options)
+        c = c.count(*args)
         if c.is_a?(Hash) || c.is_a?(ActiveSupport::OrderedHash)
           c.count
         else
-          c.respond_to?(:count) ? c.count(column_name, options) : c
+          c.respond_to?(:count) ? c.count(*args) : c
         end
       end
     end
diff --git a/spec/models/active_record/active_record_relation_methods_spec.rb b/spec/models/active_record/active_record_relation_methods_spec.rb
index f75dd6f..548d7c9 100644
--- a/spec/models/active_record/active_record_relation_methods_spec.rb
+++ b/spec/models/active_record/active_record_relation_methods_spec.rb
@@ -23,7 +23,11 @@ if defined? ActiveRecord
       context "when the scope use conditions on includes" do
         it "should keep includes and successfully count the results" do
           # Only @author and @author2 have books titled with the title00x partern
-          User.includes(:books_authored).where("books.title LIKE 'title00%'").page(1).total_count.should == 2
+          if ActiveRecord::VERSION::STRING >= "4.1.0"
+            User.includes(:books_authored).references(:books).where("books.title LIKE 'title00%'").page(1).total_count.should == 2
+          else
+            User.includes(:books_authored).where("books.title LIKE 'title00%'").page(1).total_count.should == 2
+          end
         end
       end
 
@@ -34,14 +38,24 @@ if defined? ActiveRecord
       end
 
       context "when total_count receives options" do
-        it "should return a distinct total count" do
-          User.page(1).total_count(:name, :distinct => true).should == 4
+        it "should return a distinct total count for rails ~> 4.0.0" do
+          if ActiveRecord::VERSION::STRING < "4.1.0"
+            User.page(1).total_count(:name, :distinct => true).should == 4
+          end
+        end
+
+        it "should ignore the options for rails 4.1+" do
+          if ActiveRecord::VERSION::STRING >= "4.1.0"
+            User.page(1).total_count(:name, :distinct => true).should == 7
+          end
         end
       end
 
       context "when count receives options" do
-        it "should return a distinct set by column" do
-          User.page(1).count(:name, :distinct => true).should == 4
+        it "should return a distinct set by column for rails ~> 4.0.0" do
+          if ActiveRecord::VERSION::STRING < "4.1.0"
+            User.page(1).count(:name, :distinct => true).should == 4
+          end
         end
       end