elasticsearch-model 6.0.0 → 6.1.0

data/test/integration/active_record_associations_parent_child_test.rb

@@ -1,188 +0,0 @@
-require 'test_helper'
-require 'active_record'
-
-# Needed for ActiveRecord 3.x ?
-ActiveRecord::Base.establish_connection( :adapter => 'sqlite3', :database => ":memory:" ) unless ActiveRecord::Base.connected?
-
-::ActiveRecord::Base.raise_in_transactional_callbacks = true if ::ActiveRecord::Base.respond_to?(:raise_in_transactional_callbacks) && ::ActiveRecord::VERSION::MAJOR.to_s < '5'
-
-class Question < ActiveRecord::Base
-  include Elasticsearch::Model
-
-  has_many :answers, dependent: :destroy
-
-  JOIN_TYPE = 'question'.freeze
-  JOIN_METADATA = { join_field: JOIN_TYPE}.freeze
-
-  index_name 'questions_and_answers'.freeze
-  document_type 'doc'.freeze
-
-  mapping do
-    indexes :title
-    indexes :text
-    indexes :author
-  end
-
-  def as_indexed_json(options={})
-    # This line is necessary for differences between ActiveModel::Serializers::JSON#as_json versions
-    json = as_json(options)[JOIN_TYPE] || as_json(options)
-    json.merge(JOIN_METADATA)
-  end
-
-  after_commit lambda { __elasticsearch__.index_document  },  on: :create
-  after_commit lambda { __elasticsearch__.update_document },  on: :update
-  after_commit lambda { __elasticsearch__.delete_document },  on: :destroy
-end
-
-class Answer < ActiveRecord::Base
-  include Elasticsearch::Model
-
-  belongs_to :question
-
-  JOIN_TYPE = 'answer'.freeze
-
-  index_name 'questions_and_answers'.freeze
-  document_type 'doc'.freeze
-
-  before_create :randomize_id
-
-  def randomize_id
-    begin
-      self.id = SecureRandom.random_number(1_000_000)
-    end while Answer.where(id: self.id).exists?
-  end
-
-  mapping do
-    indexes :text
-    indexes :author
-  end
-
-  def as_indexed_json(options={})
-    # This line is necessary for differences between ActiveModel::Serializers::JSON#as_json versions
-    json = as_json(options)[JOIN_TYPE] || as_json(options)
-    json.merge(join_field: { name: JOIN_TYPE, parent: question_id })
-  end
-
-  after_commit lambda { __elasticsearch__.index_document(routing: (question_id || 1))  },  on: :create
-  after_commit lambda { __elasticsearch__.update_document(routing: (question_id || 1)) },  on: :update
-  after_commit lambda {__elasticsearch__.delete_document(routing: (question_id || 1)) },  on: :destroy
-end
-
-module ParentChildSearchable
-  INDEX_NAME = 'questions_and_answers'.freeze
-  JOIN = 'join'.freeze
-
-  def create_index!(options={})
-    client = Question.__elasticsearch__.client
-    client.indices.delete index: INDEX_NAME rescue nil if options[:force]
-
-    settings = Question.settings.to_hash.merge Answer.settings.to_hash
-    mapping_properties = { join_field: { type: JOIN,
-                                         relations: { Question::JOIN_TYPE => Answer::JOIN_TYPE } } }
-
-    merged_properties = mapping_properties.merge(Question.mappings.to_hash[:doc][:properties]).merge(
-        Answer.mappings.to_hash[:doc][:properties])
-    mappings = { doc: { properties: merged_properties }}
-
-    client.indices.create index: INDEX_NAME,
-                          body: {
-                              settings: settings.to_hash,
-                              mappings: mappings }
-  end
-
-  extend self
-end
-
-module Elasticsearch
-  module Model
-    class ActiveRecordAssociationsParentChildIntegrationTest < Elasticsearch::Test::IntegrationTestCase
-
-      context "ActiveRecord associations with parent/child modelling" do
-        setup do
-          ActiveRecord::Schema.define(version: 1) do
-            create_table :questions do |t|
-              t.string     :title
-              t.text       :text
-              t.string     :author
-              t.timestamps null: false
-            end
-
-            create_table :answers do |t|
-              t.text       :text
-              t.string     :author
-              t.references :question
-              t.timestamps null: false
-            end
-
-            add_index(:answers, :question_id) unless index_exists?(:answers, :question_id)
-          end
-
-          Question.delete_all
-          ParentChildSearchable.create_index! force: true
-
-          q_1 = Question.create! title: 'First Question',  author: 'John'
-          q_2 = Question.create! title: 'Second Question', author: 'Jody'
-
-          q_1.answers.create! text: 'Lorem Ipsum', author: 'Adam'
-          q_1.answers.create! text: 'Dolor Sit',   author: 'Ryan'
-
-          q_2.answers.create! text: 'Amet Et', author: 'John'
-
-          Question.__elasticsearch__.refresh_index!
-        end
-
-        should "find questions by matching answers" do
-          response = Question.search(
-              { query: {
-                  has_child: {
-                      type: 'answer',
-                      query: {
-                          match: {
-                              author: 'john'
-                          }
-                      }
-                  }
-              }
-              })
-
-          assert_equal 'Second Question', response.records.first.title
-        end
-
-        should "find answers for matching questions" do
-          response = Answer.search(
-              { query: {
-                  has_parent: {
-                      parent_type: 'question',
-                      query: {
-                          match: {
-                              author: 'john'
-                          }
-                      }
-                  }
-              }
-              })
-
-          assert_same_elements ['Adam', 'Ryan'], response.records.map(&:author)
-        end
-
-        should "delete answers when the question is deleted" do
-          Question.where(title: 'First Question').each(&:destroy)
-          Question.__elasticsearch__.refresh_index!
-
-          response = Answer.search(
-              { query: {
-                  has_parent: {
-                      parent_type: 'question',
-                      query: {
-                          match_all: {}
-                      }
-                  }
-              }
-              })
-
-          assert_equal 1, response.results.total
-        end
-      end
-    end
-  end
-end

data/test/integration/active_record_associations_test.rb

@@ -1,339 +0,0 @@
-require 'test_helper'
-require 'active_record'
-
-# Needed for ActiveRecord 3.x ?
-ActiveRecord::Base.establish_connection( :adapter => 'sqlite3', :database => ":memory:" ) unless ActiveRecord::Base.connected?
-
-::ActiveRecord::Base.raise_in_transactional_callbacks = true if ::ActiveRecord::Base.respond_to?(:raise_in_transactional_callbacks) && ::ActiveRecord::VERSION::MAJOR.to_s < '5'
-
-module Elasticsearch
-  module Model
-    class ActiveRecordAssociationsIntegrationTest < Elasticsearch::Test::IntegrationTestCase
-
-      # ----- Search integration via Concern module -----------------------------------------------------
-
-      module Searchable
-        extend ActiveSupport::Concern
-
-        included do
-          include Elasticsearch::Model
-          include Elasticsearch::Model::Callbacks
-
-          # Set up the mapping
-          #
-          settings index: { number_of_shards: 1, number_of_replicas: 0 } do
-            mapping do
-              indexes :title,      analyzer: 'snowball'
-              indexes :created_at, type: 'date'
-
-              indexes :authors do
-                indexes :first_name
-                indexes :last_name
-                indexes :full_name, type: 'text' do
-                  indexes :raw, type: 'keyword'
-                end
-              end
-
-              indexes :categories, type: 'keyword'
-
-              indexes :comments, type: 'nested' do
-                indexes :text
-                indexes :author
-              end
-            end
-          end
-
-          # Customize the JSON serialization for Elasticsearch
-          #
-          def as_indexed_json(options={})
-            {
-              title: title,
-              text:  text,
-              categories: categories.map(&:title),
-              authors:    authors.as_json(methods: [:full_name], only: [:full_name, :first_name, :last_name]),
-              comments:   comments.as_json(only: [:text, :author])
-            }
-          end
-
-          # Update document in the index after touch
-          #
-          after_touch() { __elasticsearch__.index_document }
-        end
-      end
-
-      context "ActiveRecord associations" do
-        setup do
-
-          # ----- Schema definition ---------------------------------------------------------------
-
-          ActiveRecord::Schema.define(version: 1) do
-            create_table :categories do |t|
-              t.string     :title
-              t.timestamps null: false
-            end
-
-            create_table :categories_posts, id: false do |t|
-              t.references :post, :category
-            end
-
-            create_table :authors do |t|
-              t.string     :first_name, :last_name
-              t.timestamps null: false
-            end
-
-            create_table :authorships do |t|
-              t.string     :first_name, :last_name
-              t.references :post
-              t.references :author
-              t.timestamps null: false
-            end
-
-            create_table :comments do |t|
-              t.string     :text
-              t.string     :author
-              t.references :post
-              t.timestamps null: false
-            end
-
-            add_index(:comments, :post_id) unless index_exists?(:comments, :post_id)
-
-            create_table :posts do |t|
-              t.string     :title
-              t.text       :text
-              t.boolean    :published
-              t.timestamps null: false
-            end
-          end
-
-          # ----- Models definition -------------------------------------------------------------------------
-
-          class Category < ActiveRecord::Base
-            has_and_belongs_to_many :posts
-          end
-
-          class Author < ActiveRecord::Base
-            has_many :authorships
-
-            after_update { self.authorships.each(&:touch) }
-
-            def full_name
-              [first_name, last_name].compact.join(' ')
-            end
-          end
-
-          class Authorship < ActiveRecord::Base
-            belongs_to :author
-            belongs_to :post, touch: true
-          end
-
-          class Comment < ActiveRecord::Base
-            belongs_to :post, touch: true
-          end
-
-          class Post < ActiveRecord::Base
-            has_and_belongs_to_many :categories, after_add:    [ lambda { |a,c| a.__elasticsearch__.index_document } ],
-                                                 after_remove: [ lambda { |a,c| a.__elasticsearch__.index_document } ]
-            has_many                :authorships
-            has_many                :authors, through: :authorships,
-                                              after_add: [ lambda { |a,c| a.__elasticsearch__.index_document } ],
-                                              after_remove: [ lambda { |a,c| a.__elasticsearch__.index_document } ]
-            has_many                :comments, after_add:    [ lambda { |a,c| a.__elasticsearch__.index_document } ],
-                                               after_remove: [ lambda { |a,c| a.__elasticsearch__.index_document } ]
-
-            after_touch() { __elasticsearch__.index_document }
-          end
-
-          # Include the search integration
-          #
-          Post.__send__ :include, Searchable
-          Comment.__send__ :include, Elasticsearch::Model
-          Comment.__send__ :include, Elasticsearch::Model::Callbacks
-
-          # ----- Reset the indices -----------------------------------------------------------------
-
-          Post.delete_all
-          Post.__elasticsearch__.create_index! force: true
-
-          Comment.delete_all
-          Comment.__elasticsearch__.create_index! force: true
-        end
-
-        should "index and find a document" do
-          Post.create! title: 'Test'
-          Post.create! title: 'Testing Coding'
-          Post.create! title: 'Coding'
-          Post.__elasticsearch__.refresh_index!
-
-          response = Post.search('title:test')
-
-          assert_equal 2, response.results.size
-          assert_equal 2, response.records.size
-
-          assert_equal 'Test', response.results.first.title
-          assert_equal 'Test', response.records.first.title
-        end
-
-        should "reindex a document after categories are changed" do
-          # Create categories
-          category_a = Category.where(title: "One").first_or_create!
-          category_b = Category.where(title: "Two").first_or_create!
-
-          # Create post
-          post = Post.create! title: "First Post", text: "This is the first post..."
-
-          # Assign categories
-          post.categories = [category_a,  category_b]
-
-          Post.__elasticsearch__.refresh_index!
-
-          query = { query: {
-                      bool: {
-                        must: {
-                          multi_match: {
-                            fields: ['title'],
-                            query: 'first'
-                          }
-                        },
-                        filter: {
-                          terms: {
-                            categories: ['One']
-                          }
-                        }
-                      }
-                    }
-                  }
-
-          response = Post.search query
-
-          assert_equal 1, response.results.size
-          assert_equal 1, response.records.size
-
-          # Remove category "One"
-          post.categories = [category_b]
-
-          Post.__elasticsearch__.refresh_index!
-          response = Post.search query
-
-          assert_equal 0, response.results.size
-          assert_equal 0, response.records.size
-        end
-
-        should "reindex a document after authors are changed" do
-          # Create authors
-          author_a = Author.where(first_name: "John", last_name: "Smith").first_or_create!
-          author_b = Author.where(first_name: "Mary", last_name: "Smith").first_or_create!
-          author_c = Author.where(first_name: "Kobe", last_name: "Griss").first_or_create!
-
-          # Create posts
-          post_1 = Post.create! title: "First Post", text: "This is the first post..."
-          post_2 = Post.create! title: "Second Post", text: "This is the second post..."
-          post_3 = Post.create! title: "Third Post", text: "This is the third post..."
-
-          # Assign authors
-          post_1.authors = [author_a,  author_b]
-          post_2.authors = [author_a]
-          post_3.authors = [author_c]
-
-          Post.__elasticsearch__.refresh_index!
-
-          response = Post.search 'authors.full_name:john'
-
-          assert_equal 2, response.results.size
-          assert_equal 2, response.records.size
-
-          post_3.authors << author_a
-
-          Post.__elasticsearch__.refresh_index!
-
-          response = Post.search 'authors.full_name:john'
-
-          assert_equal 3, response.results.size
-          assert_equal 3, response.records.size
-        end if defined?(::ActiveRecord) && ::ActiveRecord::VERSION::MAJOR >= 4
-
-        should "reindex a document after comments are added" do
-          # Create posts
-          post_1 = Post.create! title: "First Post", text: "This is the first post..."
-          post_2 = Post.create! title: "Second Post", text: "This is the second post..."
-
-          # Add comments
-          post_1.comments.create! author: 'John', text: 'Excellent'
-          post_1.comments.create! author: 'Abby', text: 'Good'
-
-          post_2.comments.create! author: 'John', text: 'Terrible'
-
-          Post.__elasticsearch__.refresh_index!
-
-          response = Post.search 'comments.author:john AND comments.text:good'
-          assert_equal 0, response.results.size
-
-          # Add comment
-          post_1.comments.create! author: 'John', text: 'Or rather just good...'
-
-          Post.__elasticsearch__.refresh_index!
-
-          response = Post.search 'comments.author:john AND comments.text:good'
-          assert_equal 0, response.results.size
-
-          response = Post.search \
-            query: {
-              nested: {
-                path: 'comments',
-                query: {
-                  bool: {
-                    must: [
-                      { match: { 'comments.author' => 'john' } },
-                      { match: { 'comments.text'   => 'good' } }
-                    ]
-                  }
-                }
-              }
-            }
-
-          assert_equal 1, response.results.size
-        end if defined?(::ActiveRecord) && ::ActiveRecord::VERSION::MAJOR >= 4
-
-        should "reindex a document after Post#touch" do
-          # Create categories
-          category_a = Category.where(title: "One").first_or_create!
-
-          # Create post
-          post = Post.create! title: "First Post", text: "This is the first post..."
-
-          # Assign category
-          post.categories << category_a
-
-          Post.__elasticsearch__.refresh_index!
-
-          assert_equal 1, Post.search('categories:One').size
-
-          # Update category
-          category_a.update_attribute :title, "Updated"
-
-          # Trigger touch on posts in category
-          category_a.posts.each { |p| p.touch }
-
-          Post.__elasticsearch__.refresh_index!
-
-          assert_equal 0, Post.search('categories:One').size
-          assert_equal 1, Post.search('categories:Updated').size
-        end
-
-        should "eagerly load associated records" do
-          post_1 = Post.create(title: 'One')
-          post_2 = Post.create(title: 'Two')
-          post_1.comments.create text: 'First comment'
-          post_1.comments.create text: 'Second comment'
-
-          Comment.__elasticsearch__.refresh_index!
-
-          records = Comment.search('first').records(includes: :post)
-
-          assert records.first.association(:post).loaded?, "The associated Post should be eagerly loaded"
-          assert_equal 'One', records.first.post.title
-        end
-      end
-
-    end
-  end
-end