elastic_adapter 0.0.5 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3ef3954288335470584fed955a44ab7f3cfd6440
-  data.tar.gz: 746286ade1824a240a0af5aa9a6372fde60ba916
+  metadata.gz: 3023280225ea7fdf0075512820f96c192d448e33
+  data.tar.gz: f492fc71afbd84718f09dc7fb2a176f9adee2de0
 SHA512:
-  metadata.gz: 4a698b921410af21e18157aa2b124766023cf60cf9a85a76c869c3a44540438322ded3481e8ecf854a5a110b5249b5b90da29382dda145fe28b998b38e06ed54
-  data.tar.gz: 32cc753dcd9fd5e32be19397c94588b75bfbda2d5d666adb2dc5fc4cb7bc8907289701dface6d82964cee984401218e9812e78212cdc9b3c6185f714bec85a1d
+  metadata.gz: 35fa163b104cb2965f2559ae8ea97a0438ededd741e339d7d9e163e0839360eef9004df5bd1140f93b5f487040a2f5a73ce498f43234efe80bfd03f82c7478d2
+  data.tar.gz: e5e2cf3126f2b0e58d98163d6716db4cec4da8f019c01606cfa975cd018b8f0dc89665fd4d02aba5efeaa4726194d6303a569f20630505edb28af06e94493ca1

data/lib/elastic_adapter/responses/search_response.rb

@@ -1,20 +1,37 @@
 module ElasticAdapter
   module Responses
     class SearchResponse < BaseResponse
-      attr_reader :hits
+      attr_reader :hits, :aggregations, :suggestions
 
       private
 
       def set_instance_variables
-        hits = []
-        object[:hits][:hits].each do |hit|
-          hits << {
-            id: hit[:id]
-          }.merge(hit[:source])
+        @hits = collect_hits
+        @aggregations = collect_aggregations
+        @suggestions = collect_suggestions
+      end
+
+      def collect_hits
+        object[:hits][:hits].map { |hit| { id: hit[:id] }.merge(hit[:source]) }
+      end
 
+      def collect_aggregations
+        aggs = {}
+        object[:aggregations].each do |agg_name, result|
+          aggs[agg_name] = []
+          result[:buckets].each do |agg|
+            aggs[agg_name] << { term: agg[:key], count: agg[:doc_count] }
+          end
         end
+        aggs
+      end
 
-        @hits = hits
+      def collect_suggestions
+        object[:suggest].map { |suggestion_name, content|
+          # In this context key is the named suggestion
+          # returned by elasticsearch
+          SuggestionResponse::Suggestion.new(suggestion_name, content)
+        }
       end
     end
   end

data/lib/elastic_adapter/version.rb

@@ -1,3 +1,3 @@
 module ElasticAdapter
-  VERSION = "0.0.5"
+  VERSION = "0.0.6"
 end

data/spec/responses/search_response_spec.rb

@@ -23,6 +23,36 @@ module ElasticAdapter
               }
             }
           ]
+        },
+        aggregations: {
+          products: {
+            doc_count_error_upper_bound: 46,
+            buckets: [
+              {
+                key: "Product A",
+                doc_count: 100
+              },
+              {
+                key: "Product Z",
+                doc_count: 52
+              }
+            ]
+          }
+        },
+        suggest: {
+          foo_suggest: [
+            {
+              text: "ba",
+              offset: 0,
+              length: 2,
+              options: [
+                {
+                  text: "bar",
+                  score: 1.0
+                }
+              ]
+            }
+          ]
         }
       }}
 
@@ -38,6 +68,71 @@ module ElasticAdapter
           })
         end
       end
+
+      describe "#aggregations" do
+        it "counts one" do
+          expect(subject.aggregations.count).to eq 1
+        end
+
+        it "contains products" do
+          expect(subject.aggregations).to have_key :products
+        end
+
+        describe "products" do
+          it "counts two" do
+            expect(subject.aggregations[:products].count).to eq 2
+          end
+
+          it "contains Product A with it's count" do
+            expect(subject.aggregations[:products]).to include({
+              term: "Product A",
+              count: 100
+            })
+          end
+
+          it "contains Product Z with it's count" do
+            expect(subject.aggregations[:products]).to include({
+              term: "Product Z",
+              count: 52
+            })
+          end
+        end
+      end
+
+      describe "suggestions" do
+        it "is an array" do
+          expect(subject.suggestions).to be_an Array
+        end
+
+        describe "first suggestion" do
+          describe "#terms" do
+            it "is an Array" do
+              expect(subject.suggestions.first.terms).to be_an Array
+            end
+
+            it "counts one" do
+              expect(subject.suggestions.first.terms.count).to eq 1
+            end
+
+            describe "first term" do
+              describe "#text" do
+                it "equals 'ba'" do
+                  expect(subject.suggestions.first.terms.first.text).to eq "ba"
+                end
+
+                describe "#options" do
+                  it "includes the options" do
+                    expect(subject.suggestions.first.terms.first.options).to include({
+                      text: "bar",
+                      score: 1.0
+                    })
+                  end
+                end
+              end
+            end
+          end
+        end
+      end
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: elastic_adapter
 version: !ruby/object:Gem::Version
-  version: 0.0.5
+  version: 0.0.6
 platform: ruby
 authors:
 - Kristopher Bredemeier
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-07-28 00:00:00.000000000 Z
+date: 2015-09-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: elasticsearch