[MODEL] Added basic Mongoid integration test

elasticsearch-model/elasticsearch-model.gemspec

@@ -33,6 +33,7 @@ Gem::Specification.new do |s|
   s.add_development_dependency "activesupport", "> 3.0"
   s.add_development_dependency "activemodel",   "> 3.0"
   s.add_development_dependency "activerecord",  "> 3.0"
+  s.add_development_dependency "mongoid",       "> 2.0"
   s.add_development_dependency "oj"
 
   s.add_development_dependency "shoulda-context"

elasticsearch-model/test/integration/mongoid_basic_test.rb

@@ -0,0 +1,138 @@
+require 'test_helper'
+
+begin
+  require "mongoid"
+  session = Moped::Connection.new("localhost", 27017, 0.5)
+  session.connect
+  ENV["MONGODB_AVAILABLE"] = 'yes'
+rescue LoadError, Moped::Errors::ConnectionFailure => e
+end
+
+if ENV["MONGODB_AVAILABLE"]
+
+  logger = ::Logger.new(STDERR)
+  logger.formatter = lambda { |s, d, p, m| " #{m.ansi(:faint, :cyan)}\n" }
+  logger.level = ::Logger::DEBUG
+
+  Mongoid.logger = logger unless ENV['QUIET']
+  Moped.logger   = logger unless ENV['QUIET']
+
+  Mongoid.connect_to 'mongoid_articles'
+
+  module Elasticsearch
+    module Model
+      class MongoidBasicIntegrationTest < Elasticsearch::Test::IntegrationTestCase
+
+        class ::MongoidArticle
+          include Mongoid::Document
+          include Elasticsearch::Model
+          include Elasticsearch::Model::Callbacks
+
+          field :id, type: String
+          field :title, type: String
+          attr_accessible :title
+
+          mapping do
+            indexes :title,      type: 'string', analyzer: 'snowball'
+            indexes :created_at, type: 'date'
+          end
+        end
+
+        context "Mongoid integration" do
+          setup do
+            MongoidArticle.__elasticsearch__.create_index! force: true
+
+            MongoidArticle.delete_all
+
+            MongoidArticle.create! title: 'Test'
+            MongoidArticle.create! title: 'Testing Coding'
+            MongoidArticle.create! title: 'Coding'
+
+            MongoidArticle.__elasticsearch__.refresh_index!
+          end
+
+          should "index and find a document" do
+            response = MongoidArticle.search('title:test')
+
+            assert response.any?
+
+            assert_equal 2, response.results.size
+            assert_equal 2, response.records.size
+
+            assert_instance_of Elasticsearch::Model::Response::Result, response.results.first
+            assert_instance_of MongoidArticle, response.records.first
+
+            assert_equal 'Test', response.results.first.title
+            assert_equal 'Test', response.records.first.title
+          end
+
+          should "iterate over results" do
+            response = MongoidArticle.search('title:test')
+
+            assert_equal ['Test', 'Testing Coding'], response.results.map(&:title)
+            assert_equal ['Test', 'Testing Coding'], response.records.map(&:title)
+          end
+
+          should "access results from records" do
+            response = MongoidArticle.search('title:test')
+
+            response.records.each_with_hit do |r, h|
+              assert_not_nil h._score
+              assert_not_nil h._source.title
+            end
+          end
+
+          should "remove document from index on destroy" do
+            article = MongoidArticle.first
+
+            article.destroy
+            assert_equal 2, MongoidArticle.count
+
+            MongoidArticle.__elasticsearch__.refresh_index!
+
+            response = MongoidArticle.search 'title:test'
+
+            assert_equal 1, response.results.size
+            assert_equal 1, response.records.size
+          end
+
+          should "index updates to the document" do
+            article = MongoidArticle.first
+
+            article.title = 'Writing'
+            article.save
+
+            MongoidArticle.__elasticsearch__.refresh_index!
+
+            response = MongoidArticle.search 'title:write'
+
+            assert_equal 1, response.results.size
+            assert_equal 1, response.records.size
+          end
+
+          should "return results for a DSL search" do
+            response = MongoidArticle.search query: { match: { title: { query: 'test' } } }
+
+            assert_equal 2, response.results.size
+            assert_equal 2, response.records.size
+          end
+
+          should "return a paged collection" do
+            response = MongoidArticle.search query: { match: { title: { query: 'test' } } },
+                                      size: 2,
+                                      from: 1
+
+            assert_equal 1, response.results.size
+            assert_equal 1, response.records.size
+
+            assert_equal 'Testing Coding', response.results.first.title
+            assert_equal 'Testing Coding', response.records.first.title
+          end
+
+        end
+
+      end
+    end
+  end
+
+end