[MODEL] Lazily wrap response in HashWrapper and provide #raw_response access

Author: estolfo

elasticsearch-model/lib/elasticsearch/model/response.rb

@@ -27,9 +27,7 @@ def initialize(klass, search, options={})
         # @return [Hash]
         #
         def response
-          @response ||= begin
-            HashWrapper.new(search.execute!)
-          end
+          @response ||= HashWrapper.new(search.execute!)
         end
 
         # Returns the collection of "hits" from Elasticsearch
@@ -63,7 +61,7 @@ def timed_out
         # Returns the statistics on shards
         #
         def shards
-          HashWrapper.new(response['_shards'])
+          response['_shards']
         end
 
         # Returns a Hashie::Mash of the aggregations
@@ -77,6 +75,10 @@ def aggregations
         def suggestions
           Suggestions.new(response['suggest'])
         end
+
+        def raw_response
+          @raw_response ||= search.execute!
+        end
       end
     end
   end

elasticsearch-model/lib/elasticsearch/model/response/base.rb

@@ -13,6 +13,7 @@ module Base
         def initialize(klass, response, options={})
           @klass     = klass
           @response  = response
+          @raw_response = response
         end
 
         # @abstract Implement this method in specific class

elasticsearch-model/test/unit/response_results_test.rb

@@ -27,5 +27,8 @@ def self.document_type; 'bar'; end
       assert_equal 'bar', @results.first.foo
     end
 
+    should "provide access to the raw response" do
+      assert_equal RESPONSE, @response.raw_response
+    end
   end
 end