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

Author: estolfo

elasticsearch-persistence/lib/elasticsearch/persistence/repository/response/results.rb

@@ -11,6 +11,7 @@ class Results
           include Enumerable
 
           attr_reader :repository
+          attr_reader :raw_response
 
           # The key for accessing the results in an Elasticsearch query response.
           #
@@ -30,8 +31,8 @@ class Results
           #
           def initialize(repository, response, options={})
             @repository = repository
-            @response   = Elasticsearch::Model::HashWrapper.new(response)
-            @options    = options
+            @raw_response = response
+            @options = options
           end
 
           def method_missing(method_name, *arguments, &block)
@@ -45,13 +46,13 @@ def respond_to?(method_name, include_private = false)
           # The number of total hits for a query
           #
           def total
-            response[HITS][TOTAL]
+            raw_response[HITS][TOTAL]
           end
 
           # The maximum score for a query
           #
           def max_score
-            response[HITS][MAX_SCORE]
+            raw_response[HITS][MAX_SCORE]
           end
 
           # Yields [object, hit] pairs to the block
@@ -76,7 +77,7 @@ def map_with_hit(&block)
           # @return [Array]
           #
           def results
-            @results ||= response[HITS][HITS].map do |document|
+            @results ||= raw_response[HITS][HITS].map do |document|
               repository.deserialize(document.to_hash)
             end
           end
@@ -93,7 +94,7 @@ def results
           # @return [Elasticsearch::Model::HashWrapper]
           #
           def response
-            @response
+            @response ||= Elasticsearch::Model::HashWrapper.new(raw_response)
           end
         end
       end

elasticsearch-persistence/test/unit/repository_response_results_test.rb

@@ -93,6 +93,10 @@ class MyDocument; end
 
       assert_equal ['FOO---bar'], subject.map_with_hit { |object, hit| "#{object}---#{hit.foo}" }
     end
+
+    should "provide access to the raw response" do
+      assert_equal 5, subject.raw_response['_shards']['total']
+    end
   end
 
 end