elasticsearch-persistence-queryable 0.1.8

data/test/unit/model_find_test.rb

@@ -0,0 +1,148 @@
+require 'test_helper'
+
+require 'active_model'
+require 'virtus'
+
+require 'elasticsearch/persistence/model/errors'
+require 'elasticsearch/persistence/model/find'
+
+class Elasticsearch::Persistence::ModelFindTest < Test::Unit::TestCase
+  context "The model find module," do
+
+    class DummyFindModel
+      include ActiveModel::Naming
+      include ActiveModel::Conversion
+      include ActiveModel::Serialization
+      include ActiveModel::Serializers::JSON
+      include ActiveModel::Validations
+
+      include Virtus.model
+
+      extend  Elasticsearch::Persistence::Model::Find::ClassMethods
+
+      extend  ActiveModel::Callbacks
+      define_model_callbacks :create, :save, :update, :destroy
+      define_model_callbacks :find, :touch, only: :after
+
+      attribute :title, String
+      attribute :count, Integer, default: 0
+      attribute :created_at, DateTime, default: lambda { |o,a| Time.now.utc }
+      attribute :updated_at, DateTime, default: lambda { |o,a| Time.now.utc }
+    end
+
+    setup do
+      @gateway         = stub(client: stub(), index_name: 'foo', document_type: 'bar')
+      DummyFindModel.stubs(:gateway).returns(@gateway)
+
+      @response = MultiJson.load <<-JSON
+        {
+          "took": 14,
+          "timed_out": false,
+          "_shards": {
+            "total": 5,
+            "successful": 5,
+            "failed": 0
+          },
+          "hits": {
+            "total": 1,
+            "max_score": 1.0,
+            "hits": [
+              {
+                "_index": "dummy",
+                "_type": "dummy",
+                "_id": "abc123",
+                "_score": 1.0,
+                "_source": {
+                  "name": "TEST"
+                }
+              }
+            ]
+          }
+        }
+      JSON
+    end
+
+    should "find all records" do
+      @gateway
+        .expects(:search)
+        .with({ query: { match_all: {} }, size: 10_000 })
+        .returns(@response)
+
+      DummyFindModel.all
+    end
+
+    should "pass options when finding all records" do
+      @gateway
+        .expects(:search)
+        .with({ query: { match: { title: 'test' } }, size: 10_000, routing: 'abc123' })
+        .returns(@response)
+
+      DummyFindModel.all( { query: { match: { title: 'test' } }, routing: 'abc123' } )
+    end
+
+    context "finding via scan/scroll" do
+      setup do
+        @gateway
+        .expects(:deserialize)
+        .with('_source' => {'foo' => 'bar'})
+        .returns('_source' => {'foo' => 'bar'})
+
+        @gateway.client
+          .expects(:search)
+          .with do |arguments|
+            assert_equal 'scan', arguments[:search_type]
+            assert_equal 'foo',  arguments[:index]
+            assert_equal 'bar',  arguments[:type]
+            true
+          end
+          .returns(MultiJson.load('{"_scroll_id":"abc123==", "hits":{"hits":[]}}'))
+
+        @gateway.client
+          .expects(:scroll)
+          .twice
+          .returns(MultiJson.load('{"_scroll_id":"abc456==", "hits":{"hits":[{"_source":{"foo":"bar"}}]}}'))
+          .then
+          .returns(MultiJson.load('{"_scroll_id":"abc789==", "hits":{"hits":[]}}'))
+      end
+
+      should "find all records in batches" do
+        @doc = nil
+        result = DummyFindModel.find_in_batches { |batch| @doc = batch.first['_source']['foo'] }
+
+        assert_equal 'abc789==', result
+        assert_equal 'bar',      @doc
+      end
+
+      should "return an Enumerator for find in batches" do
+        @doc = nil
+        assert_nothing_raised do
+          e = DummyFindModel.find_in_batches
+          assert_instance_of Enumerator, e
+
+          e.each { |batch| @doc = batch.first['_source']['foo'] }
+          assert_equal 'bar',      @doc
+        end
+      end
+
+      should "find each" do
+        @doc = nil
+        result = DummyFindModel.find_each { |doc| @doc = doc['_source']['foo'] }
+
+        assert_equal 'abc789==', result
+        assert_equal 'bar',      @doc
+      end
+
+      should "return an Enumerator for find each" do
+        @doc = nil
+        assert_nothing_raised do
+          e = DummyFindModel.find_each
+          assert_instance_of Enumerator, e
+
+          e.each { |doc| @doc = doc['_source']['foo'] }
+          assert_equal 'bar',      @doc
+        end
+      end
+    end
+
+  end
+end

data/test/unit/model_gateway_test.rb

@@ -0,0 +1,99 @@
+require 'test_helper'
+
+require 'elasticsearch/persistence/model'
+require 'elasticsearch/persistence/model/rails'
+
+class Elasticsearch::Persistence::ModelGatewayTest < Test::Unit::TestCase
+  context "The model gateway" do
+    setup do
+      class DummyGatewayModel
+        include Elasticsearch::Persistence::Model
+      end
+    end
+
+    teardown do
+      Elasticsearch::Persistence::ModelGatewayTest.__send__ :remove_const, :DummyGatewayModel \
+      rescue NameError; nil
+    end
+
+    should "be accessible" do
+      assert_instance_of Elasticsearch::Persistence::Repository::Class, DummyGatewayModel.gateway
+
+      $a = 0
+      DummyGatewayModel.gateway { $a += 1 }
+      assert_equal 1, $a
+
+      @b = 0
+      def run!; DummyGatewayModel.gateway { |g| @b += 1 }; end
+      run!
+      assert_equal 1, @b
+
+      assert_equal DummyGatewayModel, DummyGatewayModel.gateway.klass
+    end
+
+    should "define common attributes" do
+      d = DummyGatewayModel.new
+
+      assert_respond_to d, :updated_at
+      assert_respond_to d, :created_at
+    end
+
+    should "allow to configure settings" do
+      DummyGatewayModel.settings(number_of_shards: 1)
+
+      assert_equal 1, DummyGatewayModel.settings.to_hash[:number_of_shards]
+    end
+
+    should "allow to configure mappings" do
+      DummyGatewayModel.mapping { indexes :name, analyzer: 'snowball' }
+
+      assert_equal 'snowball',
+                   DummyGatewayModel.mapping.to_hash[:dummy_gateway_model][:properties][:name][:analyzer]
+    end
+
+    should "configure the mapping via attribute" do
+      DummyGatewayModel.attribute :name, String, mapping: { analyzer: 'snowball' }
+
+      assert_respond_to DummyGatewayModel, :name
+      assert_equal 'snowball',
+                   DummyGatewayModel.mapping.to_hash[:dummy_gateway_model][:properties][:name][:analyzer]
+    end
+
+    should "configure the mapping via an attribute block" do
+      DummyGatewayModel.attribute :name, String do
+        indexes :name, analyzer: 'custom'
+      end
+
+      assert_respond_to DummyGatewayModel, :name
+      assert_equal 'custom',
+                   DummyGatewayModel.mapping.to_hash[:dummy_gateway_model][:properties][:name][:analyzer]
+    end
+
+    should "properly look up types for classes" do
+      assert_equal 'string',  Elasticsearch::Persistence::Model::Utils::lookup_type(String)
+      assert_equal 'integer', Elasticsearch::Persistence::Model::Utils::lookup_type(Integer)
+      assert_equal 'float',   Elasticsearch::Persistence::Model::Utils::lookup_type(Float)
+      assert_equal 'date',    Elasticsearch::Persistence::Model::Utils::lookup_type(Date)
+      assert_equal 'boolean', Elasticsearch::Persistence::Model::Utils::lookup_type(Virtus::Attribute::Boolean)
+    end
+
+    should "remove IDs from hash when serializing" do
+      assert_equal( {foo: 'bar'}, DummyGatewayModel.gateway.serialize(id: '123', foo: 'bar') )
+    end
+
+    should "set IDs from hash when deserializing" do
+      assert_equal 'abc123', DummyGatewayModel.gateway.deserialize('_id' => 'abc123', '_source' => {}).id
+    end
+
+    should "set @persisted variable from hash when deserializing" do
+      assert DummyGatewayModel.gateway.deserialize('_id' => 'abc123', '_source' => {}).instance_variable_get(:@persisted)
+    end
+
+    should "allow to access the raw hit from results as Hashie::Mash" do
+      assert_equal 0.42, DummyGatewayModel.gateway.deserialize('_score' => 0.42, '_source' => {}).hit._score
+    end
+
+
+
+  end
+end

data/test/unit/model_rails_test.rb

@@ -0,0 +1,88 @@
+require 'test_helper'
+
+require 'elasticsearch/persistence/model'
+require 'elasticsearch/persistence/model/rails'
+
+require 'rails'
+require 'action_controller/railtie'
+require 'action_view/railtie'
+
+class ::MyRailsModel
+  include Elasticsearch::Persistence::Model
+  include Elasticsearch::Persistence::Model::Rails
+
+  attribute :name, String, mapping: { analyzer: 'string' }
+  attribute :published_at, DateTime
+  attribute :published_on, Date
+end
+
+class Application < Rails::Application
+  config.eager_load = false
+  config.root       = File.dirname(File.expand_path('../../../tmp', __FILE__))
+  config.logger     = Logger.new($stderr)
+
+  routes.append do
+    resources :my_rails_models
+  end
+end
+
+class ApplicationController < ActionController::Base
+  include Application.routes.url_helpers
+  include ActionController::UrlFor
+end
+ApplicationController.default_url_options = { host: 'localhost' }
+ApplicationController._routes.append { resources :my_rails_models }
+
+class MyRailsModelController < ApplicationController; end
+
+Application.initialize!
+
+class Elasticsearch::Persistence::ModelRailsTest < Test::Unit::TestCase
+  context "The model in a Rails application" do
+
+    should "generate proper URLs and paths" do
+      model = MyRailsModel.new name: 'Test'
+      model.stubs(:id).returns(1)
+      model.stubs(:persisted?).returns(true)
+
+      controller = MyRailsModelController.new
+      controller.request = ActionDispatch::Request.new({})
+
+      assert_equal 'http://localhost/my_rails_models/1', controller.url_for(model)
+      assert_equal '/my_rails_models/1/edit',            controller.edit_my_rails_model_path(model)
+    end
+
+    should "generate a link" do
+      class MyView; include ActionView::Helpers::UrlHelper; end
+
+      model = MyRailsModel.new name: 'Test'
+      view  = MyView.new
+      view.expects(:url_for).with(model).returns('foo/bar')
+
+      assert_equal '<a href="foo/bar">Show</a>', view.link_to('Show', model)
+    end
+
+    should "parse DateTime from Rails forms" do
+      params = { "published_at(1i)"=>"2014",
+                 "published_at(2i)"=>"1",
+                 "published_at(3i)"=>"1",
+                 "published_at(4i)"=>"12",
+                 "published_at(5i)"=>"00"
+                }
+
+      m = MyRailsModel.new params
+      assert_equal "2014-01-01T12:00:00+00:00", m.published_at.iso8601
+    end
+
+    should "parse Date from Rails forms" do
+      params = { "published_on(1i)"=>"2014",
+                 "published_on(2i)"=>"1",
+                 "published_on(3i)"=>"1"
+                }
+
+      m = MyRailsModel.new params
+      assert_equal "2014-01-01", m.published_on.iso8601
+    end
+
+  end
+end