Merge pull request sunspot#628 from 4r2r/fix-data-accessor-for-stub-i…

…ssue

Fix: undefined method data_accessor_for for StubSessionProxy::Search

sunspot_rails/lib/sunspot/rails/stub_session_proxy.rb

@@ -71,8 +71,12 @@ def new_more_like_this(*args)
         Search.new
       end
 
+      class DataAccessorStub
+        attr_accessor :include, :select
+      end
+
       class Search
-        
+
         def build
           self
         end
@@ -102,28 +106,32 @@ def dynamic_facet(name)
           FacetStub.new
         end
 
+        def data_accessor_for(klass)
+          DataAccessorStub.new
+        end
+
         def execute
           self
         end
       end
-      
-      
+
+
       class PaginatedCollection < Array
-        
+
         def total_count
           0
         end
         alias :total_entries :total_count
-        
+
         def current_page
           1
         end
-        
+
         def per_page
           30
         end
         alias :limit_value :per_page
-        
+
         def total_pages
           1
         end
@@ -152,7 +160,7 @@ def out_of_bounds?
         def offset
           0
         end
-        
+
       end
 
       class FacetStub
@@ -162,7 +170,7 @@ def rows
         end
 
       end
-      
+
     end
   end
 end

sunspot_rails/spec/stub_session_proxy_spec.rb

@@ -140,5 +140,19 @@
     it 'should return empty array if listing facets' do
       @search.facets.should == []
     end
+
+    describe '#data_accessor_for' do
+      before do
+        @accessor = @search.data_accessor_for(Post)
+      end
+
+      it 'should provide accessor for select' do
+        @accessor.should respond_to(:select, :select=)
+      end
+
+      it 'should provide accessor for include' do
+        @accessor.should respond_to(:include, :include=)
+      end
+    end
   end
 end