elasticsearch-persistence 5.0.2 → 6.1.1

data/lib/rails/generators/elasticsearch/model/model_generator.rb

@@ -1,21 +0,0 @@
-require "rails/generators/elasticsearch_generator"
-
-module Elasticsearch
-  module Generators
-    class ModelGenerator < ::Rails::Generators::NamedBase
-      source_root File.expand_path('../templates', __FILE__)
-
-      desc "Creates an Elasticsearch::Persistence model"
-      argument :attributes, type: :array, default: [], banner: "attribute:type attribute:type"
-
-      check_class_collision
-
-      def create_model_file
-        @padding = attributes.map { |a| a.name.size }.max
-        template "model.rb.tt", File.join("app/models", class_path, "#{file_name}.rb")
-      end
-
-      hook_for :test_framework
-    end
-  end
-end

data/lib/rails/generators/elasticsearch/model/templates/model.rb.tt

@@ -1,9 +0,0 @@
-<% module_namespacing do -%>
-class <%= class_name %>
-  include Elasticsearch::Persistence::Model
-
-<% attributes.each do |attribute| -%>
-  <%= "attribute :#{attribute.name},".ljust(@padding+12) %> <%= attribute.type %>
-<% end -%>
-end
-<% end -%>

data/lib/rails/generators/elasticsearch_generator.rb

@@ -1,2 +0,0 @@
-require "rails/generators/named_base"
-require "rails/generators/active_model"

data/test/integration/model/model_basic_test.rb

@@ -1,233 +0,0 @@
-require 'test_helper'
-
-require 'elasticsearch/persistence/model'
-require 'elasticsearch/persistence/model/rails'
-
-module Elasticsearch
-  module Persistence
-    class PersistenceModelBasicIntegrationTest < Elasticsearch::Test::IntegrationTestCase
-
-      class ::Person
-        include Elasticsearch::Persistence::Model
-        include Elasticsearch::Persistence::Model::Rails
-
-        settings index: { number_of_shards: 1 }
-        document_type 'human_being'
-
-        attribute :name, String,
-                  mapping: { fields: {
-                    name: { type: 'text', analyzer: 'snowball' },
-                    raw:  { type: 'keyword' }
-                  } }
-
-        attribute :birthday,   Date
-        attribute :department, String
-        attribute :salary,     Integer
-        attribute :admin,      Boolean, default: false
-
-        validates :name, presence: true
-      end
-
-      context "A basic persistence model" do
-        setup do
-          Person.create_index! force: true
-        end
-
-        should "save the object with custom ID" do
-          person = Person.new id: 1, name: 'Number One'
-          person.save
-
-          document = Person.find(1)
-          assert_not_nil document
-          assert_equal 'Number One', document.name
-        end
-
-        should "create the object with custom ID" do
-          person = Person.create id: 1, name: 'Number One'
-
-          document = Person.find(1)
-          assert_not_nil document
-          assert_equal 'Number One', document.name
-        end
-
-        should "save and find the object" do
-          person = Person.new name: 'John Smith', birthday: Date.parse('1970-01-01')
-          assert person.save
-
-          assert_not_nil person.id
-          document = Person.find(person.id)
-
-          assert_instance_of Person, document
-          assert_equal 'John Smith', document.name
-          assert_equal 'John Smith', Person.find(person.id).name
-
-          assert_not_nil Elasticsearch::Persistence.client.get index: 'people', type: 'human_being', id: person.id
-        end
-
-        should "not save an invalid object" do
-          person = Person.new name: nil
-          assert ! person.save
-        end
-
-        should "save an invalid object with the :validate option" do
-          person = Person.new name: nil, salary: 100
-          assert person.save validate: false
-
-          assert_not_nil person.id
-          document = Person.find(person.id)
-          assert_equal 100, document.salary
-        end
-
-        should "delete the object" do
-          person = Person.create name: 'John Smith', birthday: Date.parse('1970-01-01')
-
-          person.destroy
-          assert person.frozen?
-
-          assert_raise Elasticsearch::Transport::Transport::Errors::NotFound do
-            Elasticsearch::Persistence.client.get index: 'people', type: 'person', id: person.id
-          end
-        end
-
-        should "update an object attribute" do
-          person = Person.create name: 'John Smith'
-
-          person.update name: 'UPDATED'
-
-          assert_equal 'UPDATED', person.name
-          assert_equal 'UPDATED', Person.find(person.id).name
-        end
-
-        should "create the model with correct Date form Rails' form attributes" do
-          params = { "birthday(1i)"=>"2014",
-                     "birthday(2i)"=>"1",
-                     "birthday(3i)"=>"1"
-                   }
-          person = Person.create params.merge(name: 'TEST')
-
-          assert_equal Date.parse('2014-01-01'), person.birthday
-          assert_equal Date.parse('2014-01-01'), Person.find(person.id).birthday
-        end
-
-        should_eventually "update the model with correct Date form Rails' form attributes" do
-          params = { "birthday(1i)"=>"2014",
-                     "birthday(2i)"=>"1",
-                     "birthday(3i)"=>"1"
-                   }
-          person = Person.create params.merge(name: 'TEST')
-
-          person.update params.merge('birthday(1i)' => '2015')
-
-          assert_equal Date.parse('2015-01-01'), person.birthday
-          assert_equal Date.parse('2015-01-01'), Person.find(person.id).birthday
-        end
-
-        should "increment an object attribute" do
-          person = Person.create name: 'John Smith', salary: 1_000
-
-          person.increment :salary
-
-          assert_equal 1_001, person.salary
-          assert_equal 1_001, Person.find(person.id).salary
-        end
-
-        should "update the object timestamp" do
-          person = Person.create name: 'John Smith'
-          updated_at = person.updated_at
-
-          sleep 1
-          person.touch
-
-          assert person.updated_at > updated_at, [person.updated_at, updated_at].inspect
-
-          found = Person.find(person.id)
-          assert found.updated_at > updated_at, [found.updated_at, updated_at].inspect
-        end
-
-        should 'update the object timestamp on save' do
-          person = Person.create name: 'John Smith'
-          person.admin = true
-          sleep 1
-          person.save
-
-          Person.gateway.refresh_index!
-
-          found = Person.find(person.id)
-
-          # Compare without milliseconds
-          assert_equal person.updated_at.to_i, found.updated_at.to_i
-        end
-
-        should "respect the version" do
-          person = Person.create name: 'John Smith'
-
-          person.update( { name: 'UPDATE 1' })
-
-          assert_raise Elasticsearch::Transport::Transport::Errors::Conflict do
-            person.update( { name: 'UPDATE 2' }, { version: 1 } )
-          end
-        end
-
-        should "find all instances" do
-          Person.create name: 'John Smith'
-          Person.create name: 'Mary Smith'
-          Person.gateway.refresh_index!
-
-          people = Person.all
-
-          assert_equal 2, people.total
-          assert_equal 2, people.size
-        end
-
-        should "find instances by search" do
-          Person.create name: 'John Smith'
-          Person.create name: 'Mary Smith'
-          Person.gateway.refresh_index!
-
-          people = Person.search query: { match: { name: 'smith' } },
-                                 highlight: { fields: { name: {} } }
-
-          assert_equal 2, people.total
-          assert_equal 2, people.size
-
-          assert people.map_with_hit { |o,h| h._score }.all? { |s| s > 0 }
-
-          assert_not_nil people.first.hit
-          assert_match /smith/i, people.first.hit.highlight['name'].first
-        end
-
-        should "find instances in batches" do
-          50.times { |i| Person.create name: "John #{i+1}" }
-          Person.gateway.refresh_index!
-
-          @batches = 0
-          @results = []
-
-          Person.find_in_batches(_source_include: 'name') do |batch|
-            @batches += 1
-            @results += batch.map(&:name)
-          end
-
-          assert_equal  3, @batches
-          assert_equal 50, @results.size
-          assert_contains @results, 'John 1'
-        end
-
-        should "find each instance" do
-          50.times { |i| Person.create name: "John #{i+1}" }
-          Person.gateway.refresh_index!
-
-          @results = []
-
-          Person.find_each(_source_include: 'name') do |person|
-            @results << person.name
-          end
-
-          assert_equal 50, @results.size
-          assert_contains @results, 'John 1'
-        end
-      end
-
-    end
-  end
-end

data/test/integration/repository/custom_class_test.rb

@@ -1,85 +0,0 @@
-require 'test_helper'
-
-module Elasticsearch
-  module Persistence
-    class RepositoryCustomClassIntegrationTest < Elasticsearch::Test::IntegrationTestCase
-
-      class ::MyNote
-        attr_reader :attributes
-
-        def initialize(attributes={})
-          @attributes = Hashie::Mash.new(attributes)
-        end
-
-        def method_missing(method_name, *arguments, &block)
-          attributes.respond_to?(method_name) ? attributes.__send__(method_name, *arguments, &block) : super
-        end
-
-        def respond_to?(method_name, include_private=false)
-          attributes.respond_to?(method_name) || super
-        end
-
-        def to_hash
-          @attributes
-        end
-      end
-
-      context "A custom repository class" do
-        setup do
-          class ::MyNotesRepository
-            include Elasticsearch::Persistence::Repository
-
-            klass MyNote
-
-            settings number_of_shards: 1 do
-              mapping do
-                indexes :title, analyzer: 'snowball'
-              end
-            end
-
-            create_index!
-
-            def deserialize(document)
-              klass.new document.merge(document['_source'])
-            end
-          end
-
-          @repository = MyNotesRepository.new
-
-          @repository.client.cluster.health wait_for_status: 'yellow'
-        end
-
-        should "save the object under a correct index and type" do
-          @repository.save MyNote.new(id: '1', title: 'Test')
-          result = @repository.find(1)
-
-          assert_instance_of MyNote, result
-          assert_equal 'Test', result.title
-
-          assert_not_nil Elasticsearch::Persistence.client.get index: 'my_notes_repository',
-                                                               type: 'my_note',
-                                                               id: '1'
-        end
-
-        should "delete the object" do
-          note = MyNote.new id: 1, title: 'Test'
-          @repository.save note
-
-          assert_not_nil @repository.find(1)
-
-          @repository.delete(note)
-          assert_raise(Elasticsearch::Persistence::Repository::DocumentNotFound) { @repository.find(1) }
-        end
-
-        should "retrieve the object via a search query" do
-          note = MyNote.new title: 'Testing'
-          @repository.save note, refresh: true
-
-          results = @repository.search query: { match: { title: 'Test' } }
-          assert_equal 'Testing', results.first.title
-        end
-      end
-
-    end
-  end
-end

data/test/integration/repository/customized_class_test.rb

@@ -1,82 +0,0 @@
-require 'test_helper'
-
-module Elasticsearch
-  module Persistence
-    class RepositoryCustomizedClassIntegrationTest < Elasticsearch::Test::IntegrationTestCase
-
-      module ::My
-        class Note
-          attr_reader :attributes
-
-          def initialize(attributes={})
-            @attributes = attributes
-          end
-
-          def to_hash
-            @attributes
-          end
-        end
-      end
-
-      context "A custom repository class" do
-        setup do
-          @repository = Elasticsearch::Persistence::Repository.new do
-            index 'my_notes'
-            type  'my_note'
-            klass My::Note
-
-            settings number_of_shards: 1 do
-              mapping do
-                indexes :title, analyzer: 'snowball'
-              end
-            end
-
-            create_index!
-          end
-
-          @repository.client.cluster.health wait_for_status: 'yellow'
-        end
-
-        should "save the object under a correct index and type" do
-          @repository.save My::Note.new(id: '1', title: 'Test')
-
-          assert_instance_of My::Note, @repository.find(1)
-          assert_not_nil Elasticsearch::Persistence.client.get index: 'my_notes', type: 'my_note', id: '1'
-        end
-
-        should "update the document" do
-          @repository.save Note.new(id: 1, title: 'Test')
-
-          @repository.update 1, doc: { title: 'UPDATED' }
-          assert_equal 'UPDATED', @repository.find(1).attributes['title']
-        end
-
-        should "update the document with a script" do
-          @repository.save Note.new(id: 1, title: 'Test')
-
-          @repository.update 1, script: 'ctx._source.title = "UPDATED"'
-          assert_equal 'UPDATED', @repository.find(1).attributes['title']
-        end
-
-        should "delete the object" do
-          note = My::Note.new id: 1, title: 'Test'
-          @repository.save note
-
-          assert_not_nil @repository.find(1)
-
-          @repository.delete(note)
-          assert_raise(Elasticsearch::Persistence::Repository::DocumentNotFound) { @repository.find(1) }
-        end
-
-        should "create the index with correct mapping" do
-          note = My::Note.new title: 'Testing'
-          @repository.save note, refresh: true
-
-          results = @repository.search query: { match: { title: 'Test' } }
-          assert_equal 'Testing', results.first.attributes['title']
-        end
-      end
-
-    end
-  end
-end

data/test/integration/repository/default_class_test.rb

@@ -1,116 +0,0 @@
-require 'test_helper'
-
-module Elasticsearch
-  module Persistence
-    class RepositoryDefaultClassIntegrationTest < Elasticsearch::Test::IntegrationTestCase
-
-      class ::Note
-        attr_reader :attributes
-
-        def initialize(attributes={})
-          @attributes = attributes
-        end
-
-        def to_hash
-          @attributes
-        end
-      end
-
-      context "The default repository class" do
-        setup do
-          @repository = Elasticsearch::Persistence::Repository.new
-          @repository.client.cluster.health wait_for_status: 'yellow'
-        end
-
-        should "save the object with a custom ID and find it" do
-          @repository.save Note.new(id: '1', title: 'Test')
-
-          assert_equal 'Test', @repository.find(1).attributes['title']
-        end
-
-        should "save the object with an auto-generated ID and find it" do
-          response = @repository.save Note.new(title: 'Test')
-
-          assert_equal 'Test', @repository.find(response['_id']).attributes['title']
-        end
-
-        should "update the document" do
-          @repository.save Note.new(id: 1, title: 'Test')
-
-          @repository.update 1, type: 'note', doc: { title: 'UPDATED' }
-          assert_equal 'UPDATED', @repository.find(1).attributes['title']
-        end
-
-        should "update the document with a script" do
-          @repository.save Note.new(id: 1, title: 'Test')
-
-          @repository.update 1, type: 'note', script: 'ctx._source.title = "UPDATED"'
-          assert_equal 'UPDATED', @repository.find(1).attributes['title']
-        end
-
-        should "save the document with an upsert" do
-          @repository.update 1, type: 'note', script: 'ctx._source.clicks += 1', upsert: { clicks: 1 }
-          assert_equal 1, @repository.find(1).attributes['clicks']
-        end
-
-        should "delete an object" do
-          note = Note.new(id: '1', title: 'Test')
-
-          @repository.save(note)
-          assert_not_nil @repository.find(1)
-          @repository.delete(note)
-          assert_raise(Elasticsearch::Persistence::Repository::DocumentNotFound) { @repository.find(1) }
-        end
-
-        should "find multiple objects" do
-          (1..5).each { |i| @repository.save Note.new(id: i, title: "Test #{i}") }
-
-          assert_equal 5, @repository.find(1, 2, 3, 4, 5).size
-          assert_equal 5, @repository.find([1, 2, 3, 4, 5]).size
-        end
-
-        should "pass options to save and find" do
-          note = Note.new(id: '1', title: 'Test')
-          @repository.save note, routing: 'ABC'
-
-          @repository.client.cluster.health level: 'indices', wait_for_status: 'yellow'
-
-          assert_raise Elasticsearch::Persistence::Repository::DocumentNotFound do
-            @repository.find(1, routing: 'DEF')
-          end
-
-          assert_nothing_raised do
-            note = @repository.find(1, routing: 'ABC')
-            assert_instance_of Note, note
-          end
-        end
-
-        should "find notes with full text search" do
-          @repository.save Note.new(title: 'Test')
-          @repository.save Note.new(title: 'Test Test')
-          @repository.save Note.new(title: 'Crust')
-          @repository.client.indices.refresh index: @repository.index_name
-
-          results = @repository.search 'test'
-          assert_equal 2, results.size
-
-          results = @repository.search query: { match: { title: 'Test' } }
-          assert_equal 2, results.size
-        end
-
-        should "count notes" do
-          @repository.save Note.new(title: 'Test')
-          @repository.client.indices.refresh index: @repository.index_name
-          assert_equal 1, @repository.count
-        end
-
-        should "save and find a plain hash" do
-          @repository.save id: 1, title: 'Hash'
-          result = @repository.find(1)
-          assert_equal 'Hash', result['_source']['title']
-        end
-      end
-
-    end
-  end
-end