[STORE] Added the Store module

The `Store` module saves and deletes the documents in Elasticsearch
via the `save` and `delete` methods.

Author: karmi

elasticsearch-persistence/lib/elasticsearch/persistence.rb

@@ -7,6 +7,7 @@
 require 'elasticsearch/persistence/client'
 require 'elasticsearch/persistence/repository/naming'
 require 'elasticsearch/persistence/repository/serialize'
+require 'elasticsearch/persistence/repository/store'
 require 'elasticsearch/persistence/repository'
 
 require 'elasticsearch/persistence/repository/class'

elasticsearch-persistence/lib/elasticsearch/persistence/repository.rb

@@ -5,6 +5,7 @@ module Repository
       include Elasticsearch::Persistence::Client
       include Elasticsearch::Persistence::Repository::Naming
       include Elasticsearch::Persistence::Repository::Serialize
+      include Elasticsearch::Persistence::Repository::Store
 
       def new(options={}, &block)
         Elasticsearch::Persistence::Repository::Class.new options, &block

elasticsearch-persistence/lib/elasticsearch/persistence/repository/store.rb

@@ -0,0 +1,28 @@
+module Elasticsearch
+  module Persistence
+    module Repository
+
+      module Store
+        def save(document, options={})
+          serialized = serialize(document)
+          id   = __get_id_from_document(serialized)
+          type = klass || __get_type_from_class(document.class)
+          client.index( { index: 'test', type: type, id: id, body: serialized }.merge(options) )
+        end
+
+        def delete(document, options={})
+          if document.is_a?(String) || document.is_a?(Integer)
+            id   = document
+            type = klass
+          else
+            serialized = serialize(document)
+            id   = __get_id_from_document(serialized)
+            type = klass || __get_type_from_class(document.class)
+          end
+          client.delete( { index: 'test', type: type, id: id }.merge(options) )
+        end
+      end
+
+    end
+  end
+end

elasticsearch-persistence/test/unit/repository_store_test.rb

@@ -0,0 +1,120 @@
+require 'test_helper'
+
+class Elasticsearch::Persistence::RepositoryStoreTest < Test::Unit::TestCase
+  context "The repository store" do
+    class MyDocument; end
+
+    setup do
+      @shoulda_subject = Class.new() { include Elasticsearch::Persistence::Repository::Store }.new
+    end
+
+    context "save" do
+      should "serialize the document, get type from klass and index it" do
+        subject.expects(:serialize).returns({foo: 'bar'})
+        subject.expects(:klass).returns('foo_type')
+        subject.expects(:__get_id_from_document).returns('1')
+
+        client = mock
+        client.expects(:index).with do |arguments|
+          assert_equal 'foo_type', arguments[:type]
+          assert_equal '1', arguments[:id]
+          assert_equal({foo: 'bar'}, arguments[:body])
+        end
+        subject.expects(:client).returns(client)
+
+        subject.save({foo: 'bar'})
+      end
+
+      should "serialize the document, get type from document class and index it" do
+        subject.expects(:serialize).returns({foo: 'bar'})
+        subject.expects(:klass).returns(nil)
+        subject.expects(:__get_type_from_class).with(MyDocument).returns('my_document')
+        subject.expects(:__get_id_from_document).returns('1')
+
+        client = mock
+        client.expects(:index).with do |arguments|
+          assert_equal 'my_document', arguments[:type]
+          assert_equal '1', arguments[:id]
+          assert_equal({foo: 'bar'}, arguments[:body])
+        end
+        subject.expects(:client).returns(client)
+
+        subject.save(MyDocument.new)
+      end
+
+      should "pass the options to the client" do
+        subject.expects(:serialize).returns({foo: 'bar'})
+        subject.expects(:klass).returns('foo')
+        subject.expects(:__get_id_from_document).returns('1')
+
+        client = mock
+        client.expects(:index).with do |arguments|
+          assert_equal 'bambam', arguments[:routing]
+        end
+        subject.expects(:client).returns(client)
+
+        subject.save({foo: 'bar'}, routing: 'bambam')
+      end
+    end
+
+    context "delete" do
+      should "get type from klass when passed only ID" do
+        subject.expects(:serialize).never
+        subject.expects(:klass).returns('foo_type')
+        subject.expects(:__get_id_from_document).never
+
+        client = mock
+        client.expects(:delete).with do |arguments|
+          assert_equal 'foo_type', arguments[:type]
+          assert_equal '1', arguments[:id]
+        end
+        subject.expects(:client).returns(client)
+
+        subject.delete('1')
+      end
+
+      should "get ID from document and type from klass when passed a document" do
+        subject.expects(:serialize).returns({id: '1', foo: 'bar'})
+        subject.expects(:klass).returns('foo_type')
+        subject.expects(:__get_id_from_document).with({id: '1', foo: 'bar'}).returns('1')
+
+        client = mock
+        client.expects(:delete).with do |arguments|
+          assert_equal 'foo_type', arguments[:type]
+          assert_equal '1', arguments[:id]
+        end
+        subject.expects(:client).returns(client)
+
+        subject.delete({id: '1', foo: 'bar'})
+      end
+
+      should "get ID and type from document when passed a document" do
+        subject.expects(:serialize).returns({id: '1', foo: 'bar'})
+        subject.expects(:klass).returns(nil)
+        subject.expects(:__get_id_from_document).with({id: '1', foo: 'bar'}).returns('1')
+        subject.expects(:__get_type_from_class).with(MyDocument).returns('my_document')
+
+        client = mock
+        client.expects(:delete).with do |arguments|
+          assert_equal 'my_document', arguments[:type]
+          assert_equal '1', arguments[:id]
+        end
+        subject.expects(:client).returns(client)
+
+        subject.delete(MyDocument.new)
+      end
+
+      should "pass the options to the client" do
+        subject.expects(:klass).returns('foo')
+
+        client = mock
+        client.expects(:delete).with do |arguments|
+          assert_equal 'bambam', arguments[:routing]
+        end
+        subject.expects(:client).returns(client)
+
+        subject.delete('1', routing: 'bambam')
+      end
+    end
+  end
+end