waistband 0.0.15

data/spec/lib/query_result_spec.rb

@@ -0,0 +1,23 @@
+require 'spec_helper'
+
+describe Waistband::QueryResult do
+
+  let(:hit)     { {"_index"=>"search", "_type"=>"search", "_id"=>"task_14969", "_score"=>5.810753, "_source"=>{"id"=>14969, "name"=>"shopping"}} }
+  let(:result)  { Waistband::QueryResult.new(hit) }
+
+  it "allows access to inner document" do
+    result.source.should  eql hit['_source']
+    result._id.should      eql 'task_14969'
+    result.score.should   eql 5.810753
+  end
+
+  it "provides methods for source attributes" do
+    result.id.should eql 14969
+    result.name.should eql 'shopping'
+  end
+
+  it "returns nil if the method missing misses" do
+    result.something_else.should be_nil
+  end
+
+end

data/spec/lib/query_spec.rb

@@ -0,0 +1,279 @@
+require 'spec_helper'
+require 'active_support/core_ext/kernel/reporting'
+
+Kernel.silence_stderr do
+  require 'kaminari'
+end
+
+describe Waistband::Query do
+
+  let(:index) { Waistband::Index.new('search') }
+  let(:query) { index.query('shopping ikea') }
+
+  describe '#execute!' do
+
+    it "gets results from elastic search" do
+      add_result!
+
+      json = query.send(:execute!)
+      json['hits'].should be_a Hash
+      json['hits']['total'].should > 0
+      json['hits']['hits'].size.should eql 1
+
+      json['hits']['hits'].each do |hit|
+        hit['_id'].should match(/^task_.*/)
+        hit['_source'].should be_a Hash
+        hit['_source']['id'].should eql 123123
+        hit['_source']['name'].should eql 'some shopping in ikea'
+        hit['_source']['user_id'].should eql 999
+        hit['_source']['description'].should eql 'i need you to pick up some stuff in ikea'
+      end
+    end
+
+  end
+
+  describe '#add_sort' do
+
+    it "adds sort field" do
+      query.add_sort('created_at', 'asc')
+      query.instance_variable_get('@sorts').should eql [{key: 'created_at', ord: 'asc'}]
+
+      query.add_sort('updated_at', 'desc')
+      query.instance_variable_get('@sorts').should eql [{key: 'created_at', ord: 'asc'}, {key: 'updated_at', ord: 'desc'}]
+    end
+
+  end
+
+  describe '#add_range' do
+
+    it "adds a range" do
+      query.add_range('task_id', 5, 10)
+      query.instance_variable_get('@ranges').should eql [{field: 'task_id', min: 5, max: 10}]
+    end
+
+  end
+
+  describe '#add_field' do
+
+    it "adds field" do
+      query.add_field('name')
+      query.instance_variable_get('@fields').should eql ['name']
+    end
+
+    it "adds multiple fields at once" do
+      query.add_field('name', 'description')
+      query.instance_variable_get('@fields').should eql ['name', 'description']
+    end
+
+  end
+
+  describe '#add_term' do
+
+    it "adds the term on the key" do
+      query.add_term('metro', 'boston')
+      query.instance_variable_get('@terms')['metro'][:keywords].should eql ['boston']
+    end
+
+    it "adds several terms on multiple words" do
+      query.add_term('metro', 'sf bay area')
+      query.instance_variable_get('@terms')['metro'][:keywords].should eql ['sf', 'bay', 'area']
+    end
+
+  end
+
+  describe '#terms' do
+
+    it "builds an array of all terms" do
+      query.add_term('metro', 'sf bay area')
+      query.send(:terms).should eql([
+        {
+          terms: {
+            metro: ['sf', 'bay', 'area']
+          }
+        }
+      ])
+    end
+
+    it "builds an array of single terms" do
+      query.add_term('metro', 'boston')
+      query.send(:terms).should eql([
+        {
+          terms: {
+            metro: ['boston']
+          }
+        }
+      ])
+    end
+
+    it "constructs correctly with multiple terms" do
+      query.add_term('metro', 'sf bay area')
+      query.add_term('geography', 'San Francisco')
+      query.send(:terms).should eql([
+        {
+          terms: {
+            metro: ["sf", "bay", "area"]}
+          },
+          {
+            terms: {
+              geography: ["San", "Francisco"]
+            }
+          }
+      ])
+    end
+
+  end
+
+  describe '#must_to_hash' do
+
+    it "creates an array of the must of the query" do
+      query.add_term('metro', 'sf bay area')
+      query.add_field('name')
+      query.send(:must_to_hash).should eql([
+        {
+          multi_match: {
+            query: "shopping ikea",
+            fields: ['name']
+          }
+        },
+        {
+          terms: {
+            metro: ["sf", "bay", "area"]
+          }
+        }
+      ])
+    end
+
+  end
+
+  describe '#from' do
+
+    it "returns 0 when page is 1" do
+      Waistband::Query.new('search', 'shopping ikea', page: 1, page_size: 20).send(:from).should eql 0
+    end
+
+    it "returns 20 when page is 2 and page_size is 20" do
+      Waistband::Query.new('search', 'shopping ikea', page: 2, page_size: 20).send(:from).should eql 20
+    end
+
+    it "returns 30 when page is 4 and page_size is 10" do
+      Waistband::Query.new('search', 'shopping ikea', page: 4, page_size: 10).send(:from).should eql 30
+    end
+
+    it "returns 10 when page is 2 and page_size is 10" do
+      Waistband::Query.new('search', 'shopping ikea', page: 2, page_size: 10).send(:from).should eql 10
+    end
+
+  end
+
+  describe '#to_hash' do
+
+    it "constructs the query's json" do
+      query.add_term('metro', 'sf bay area')
+      query.add_field('name')
+      query.add_sort('created_at', 'desc')
+      query.add_range('task_id', 1, 10)
+
+      query.send(:to_hash).should eql({
+        query: {
+          bool: {
+            must: [
+              {
+                multi_match: {
+                  query: "shopping ikea",
+                  fields: ['name']
+                },
+              },
+              {
+                range: {
+                  'task_id' => {
+                    lte: 10,
+                    gte: 1
+                  }
+                }
+              },
+              {
+                terms: {
+                  metro: ["sf","bay","area"]
+                }
+              }
+            ],
+            must_not: [],
+            should: []
+          }
+        },
+        from: 0,
+        size: 20,
+        sort: [
+          {'created_at' => 'desc'},
+          '_score'
+        ]
+      })
+    end
+
+    it 'constructs the query with several terms' do
+      query.add_term('metro', 'sf bay area')
+      query.add_term('geography', 'San Francisco')
+      query.add_field('name')
+      query.add_field('description')
+      
+      query.send(:to_hash).should eql({
+        query: {
+          bool: {
+            must: [
+              {
+                multi_match: {
+                  query: "shopping ikea",
+                  fields: ['name', 'description']
+                }
+              },
+              {
+                terms: {
+                  metro: ["sf", "bay", "area"]
+                }
+              },
+              {
+                terms: {
+                  geography: ["San", "Francisco"]
+                }
+              }
+            ],
+            must_not: [],
+            should: []
+          }
+        },
+        from: 0,
+        size: 20,
+        sort: ['_score']
+      })
+    end
+
+  end
+
+  describe '#results' do
+
+    it "returns a QueryResult array" do
+      add_result!
+
+      query.results.first.should be_a Waistband::QueryResult
+    end
+
+  end
+
+  describe '#paginated_results' do
+
+    it "returns a kaminari paginated array" do
+      add_result!
+
+      query.paginated_results.should be_an Array
+    end
+
+  end
+
+  def add_result!
+    index.store!("task_123123", {id: 123123, name: 'some shopping in ikea', user_id: 999, description: 'i need you to pick up some stuff in ikea'})
+    index.refresh
+
+    query.add_field('name')
+  end
+
+end

data/spec/lib/stringify_all_spec.rb

@@ -0,0 +1,22 @@
+require 'spec_helper'
+
+describe Waistband::StringifyAll do
+
+  before do
+    Array.send(:include, Waistband::StringifyAll::Array)
+    Hash.send(:include, Waistband::StringifyAll::Hash)
+  end
+
+  it "stringifies everything in an array" do
+    [1, 2, 3].stringify_all.should eql %w(1 2 3)
+  end
+
+  it "stringifies everything in a hash" do
+    {'name' => :peter, 'description' => {'full' => 'ok'}}.stringify_all.should eql({'name' => 'peter', 'description' => "{\"full\"=>\"ok\"}"})
+  end
+
+  it "recurses" do
+    {'name' => :peter, 'description' => [1, 2, 3]}.stringify_all.should eql({'name' => 'peter', 'description' => "[1, 2, 3]"})
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,31 @@
+ENV['RACK_ENV'] ||= 'test'
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+APP_DIR ||= File.expand_path('../../', __FILE__)
+
+require 'waistband'
+require 'rspec'
+require 'timecop'
+require 'active_support/core_ext/integer/time'
+
+Dir["#{APP_DIR}/spec/support/**/*.rb"].each {|f| require f}
+
+RSpec.configure do |config|
+
+  config.before(:all) do
+    Waistband.configure do |c|
+      c.config_dir = "#{APP_DIR}/spec/config/waistband"
+    end
+  end
+
+  config.before(:each) do |c|
+    IndexHelper.destroy_all!
+    IndexHelper.create_all!
+  end
+
+  config.after(:each) do
+    Timecop.return
+  end
+
+end

data/spec/support/index_helper.rb

@@ -0,0 +1,21 @@
+class IndexHelper
+
+  INDEXES = %w(events search)
+
+  class << self
+
+    def destroy_all!
+      IndexHelper::INDEXES.each do |index|
+        Waistband::Index.new(index).destroy!
+      end
+    end
+
+    def create_all!
+      IndexHelper::INDEXES.each do |index|
+        Waistband::Index.new(index).create!
+      end
+    end
+
+  end
+
+end

data/waistband.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'waistband/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "waistband"
+  spec.version       = Waistband::VERSION
+  spec.authors       = ["David Jairala"]
+  spec.email         = ["davidjairala@gmail.com"]
+  spec.description   = %q{Ruby interface to Elastic Search}
+  spec.summary       = %q{Elastic Search all the things!}
+  spec.homepage      = "https://github.com/taskrabbit/waistband"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency 'activesupport',  '>= 3.0.0'
+  spec.add_dependency 'rest-client',    '~> 1.6.7'
+  spec.add_dependency 'json',           '~> 1.8.0'
+
+  spec.add_development_dependency "bundler", "~> 1.2.3"
+  spec.add_development_dependency "rake"
+end

metadata

@@ -0,0 +1,165 @@
+--- !ruby/object:Gem::Specification
+name: waistband
+version: !ruby/object:Gem::Version
+  version: 0.0.15
+  prerelease: 
+platform: ruby
+authors:
+- David Jairala
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-08-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: 3.0.0
+- !ruby/object:Gem::Dependency
+  name: rest-client
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.6.7
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.6.7
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.8.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.8.0
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.2.3
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.2.3
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Ruby interface to Elastic Search
+email:
+- davidjairala@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- .rvmrc
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/waistband.rb
+- lib/waistband/configuration.rb
+- lib/waistband/index.rb
+- lib/waistband/model.rb
+- lib/waistband/query.rb
+- lib/waistband/query_result.rb
+- lib/waistband/quick_error.rb
+- lib/waistband/stringify_all.rb
+- lib/waistband/version.rb
+- spec/config/waistband/waistband.yml
+- spec/config/waistband/waistband_events.yml
+- spec/config/waistband/waistband_search.yml
+- spec/lib/configuration_spec.rb
+- spec/lib/index_spec.rb
+- spec/lib/model_spec.rb
+- spec/lib/query_result_spec.rb
+- spec/lib/query_spec.rb
+- spec/lib/stringify_all_spec.rb
+- spec/spec_helper.rb
+- spec/support/index_helper.rb
+- waistband.gemspec
+homepage: https://github.com/taskrabbit/waistband
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.25
+signing_key: 
+specification_version: 3
+summary: Elastic Search all the things!
+test_files:
+- spec/config/waistband/waistband.yml
+- spec/config/waistband/waistband_events.yml
+- spec/config/waistband/waistband_search.yml
+- spec/lib/configuration_spec.rb
+- spec/lib/index_spec.rb
+- spec/lib/model_spec.rb
+- spec/lib/query_result_spec.rb
+- spec/lib/query_spec.rb
+- spec/lib/stringify_all_spec.rb
+- spec/spec_helper.rb
+- spec/support/index_helper.rb