* add ActiveRecord::Base.logger for specs

* Add ThinkingSphinx::Search#count and ActiveRecord::Base.search_count for
  returnings counts of Sphinx searches

lib/thinking_sphinx/active_record/search.rb

@@ -28,7 +28,14 @@ def search(*args)
               args << options
               ThinkingSphinx::Search.search(*args)
             end
-
+
+            def search_count(*args)
+              options = args.extract_options!
+              options[:class] = self
+              args << options
+              ThinkingSphinx::Search.count(*args)
+            end
+
             def search_for_id(*args)
               options = args.extract_options!
               options[:class] = self

lib/thinking_sphinx/search.rb

@@ -203,7 +203,12 @@ def search(*args)
           instances_from_results(results[:matches], options, klass)
         end
       end
-
+
+      def count(*args)
+        results, client = search_results(*args.clone)
+        results[:total] || 0
+      end
+
       # Checks if a document with the given id exists within a specific index.
       # Expected parameters:
       #

spec/spec_helper.rb

@@ -7,6 +7,8 @@
 require 'lib/thinking_sphinx'
 require 'spec/sphinx_helper'
 
+ActiveRecord::Base.logger = Logger.new(StringIO.new)
+
 Spec::Runner.configure do |config|
   # config.mock_with NotAMock::RspecMockFrameworkAdapter
 

spec/unit/thinking_sphinx/active_record/search_spec.rb

@@ -9,6 +9,10 @@
     ActiveRecord::Base.methods.should include("search")
   end
 
+  it "should add search_count to ActiveRecord::Base" do
+    ActiveRecord::Base.methods.should include("search_count")
+  end
+
   it "should add search_for_id to ActiveRecord::Base" do
     ActiveRecord::Base.methods.should include("search_for_id")
   end
@@ -78,4 +82,26 @@
       )
     end
   end
+
+  describe "search_count method" do
+    before :each do
+      ThinkingSphinx::Search.stub_method(:count => true)
+    end
+
+    it "should call ThinkingSphinx::Search#search with the class option set" do
+      Person.search_count("search")
+
+      ThinkingSphinx::Search.should have_received(:count).with(
+        "search", :class => Person
+      )
+    end
+
+    it "should override the class option" do
+      Person.search_count("search", :class => Friendship)
+
+      ThinkingSphinx::Search.should have_received(:count).with(
+        "search", :class => Person
+      )
+    end
+  end
 end

spec/unit/thinking_sphinx/search_spec.rb

@@ -160,4 +160,31 @@
       ).should == [@person_a, @person_b, @person_c]
     end
   end
+
+  describe "count method" do
+    before :each do
+      @client = Riddle::Client.stub_instance(
+        :filters    => [],
+        :filters=   => true,
+        :id_range=  => true,
+        :sort_mode  => :asc,
+        :limit      => 5,
+        :offset=    => 0,
+        :sort_mode= => true,
+        :query      => {
+          :matches  => [],
+          :total    => 50
+        }
+      )
+
+      ThinkingSphinx::Search.stub_methods(
+        :client_from_options => @client,
+        :search_conditions   => ["", []]
+      )
+    end
+
+    it "should return query total" do
+      ThinkingSphinx::Search.count(42, "an_index").should == 50
+    end
+  end
 end