ssickles-tire 0.4.2.7 → 0.4.3

data/test/unit/active_model_lint_test.rb

@@ -0,0 +1,17 @@
+require 'test_helper'
+
+module Tire
+  module Model
+
+    class ActiveModelLintTest < Test::Unit::TestCase
+
+      include ActiveModel::Lint::Tests
+
+      def setup
+        @model = PersistentArticle.new :title => 'Test'
+      end
+
+    end
+
+  end
+end

data/test/unit/index_alias_test.rb

@@ -12,15 +12,9 @@ module Tire
         should "have a name and defaults" do
           @alias = Alias.new :name => 'dummy'
           assert_equal 'dummy', @alias.name
-          assert_equal Configuration.url, @alias.url
           assert @alias.indices.empty?
         end
 
-        should "have a configurable base URL" do
-          @alias = Alias.new :url => 'http://example.com:9200'
-          assert_equal 'http://example.com:9200', @alias.url
-        end
-
         should "have indices" do
           @alias = Alias.new :indices => ['index_A', 'index_B']
           assert_equal ['index_A', 'index_B'], @alias.indices.to_a
@@ -100,7 +94,7 @@ module Tire
               a['add']['alias'] == 'alias_martha'
               end
           end.returns(mock_response('{}'), 200)
-
+          
           a = Alias.find('alias_martha')
           a.indices.push 'index_A'
           a.save
@@ -115,7 +109,7 @@ module Tire
                 a['remove']['alias'] == 'alias_martha'
               end
           end.returns(mock_response('{}'), 200)
-
+          
           a = Alias.find('alias_martha')
           a.indices.delete 'index_A'
           a.save
@@ -126,7 +120,7 @@ module Tire
             # puts json
             MultiJson.decode(json)['actions'].all? { |a| a['add']['routing'] == 'martha' }
           end.returns(mock_response('{}'), 200)
-
+          
           a = Alias.find('alias_martha')
           a.routing('martha')
           a.save
@@ -187,14 +181,6 @@ module Tire
           a = Alias.find('alias_martha')
           assert_instance_of Alias, a
           assert_equal ['index_B', 'index_C'], a.indices.to_a.sort
-
-          Configuration.client.expects(:get).with do |url, json|
-              url  == "http://example.com:9200/_aliases"
-            end.returns( mock_response(                               %q|{"index_A":{"aliases":{}},"index_B":{"aliases":{"alias_john":{"filter":{"term":{"user":"john"}}},"alias_martha":{"filter":{"term":{"user":"martha"}}}}},"index_C":{"aliases":{"alias_martha":{"filter":{"term":{"user":"martha"}}}}}}|), 200 )
-
-          a = Alias.find('http://example.com:9200', 'alias_martha')
-          assert_instance_of Alias, a
-          assert_equal ['index_B', 'index_C'], a.indices.to_a.sort
         end
 
         should "find an alias and configure it with a block" do

data/test/unit/index_test.rb

@@ -18,11 +18,6 @@ module Tire
         assert_equal "#{Configuration.url}/#{@index.name}", @index.url
       end
 
-      should "have a configurable URL endpoint" do
-        @index.url = 'http://example.com'
-        assert_equal "http://example.com/#{@index.name}", @index.url
-      end
-
       should "return HTTP response" do
         assert_respond_to @index, :response
 
@@ -216,21 +211,21 @@ module Tire
 
         should "set type from Hash :type property" do
           Configuration.client.expects(:post).with do |url,document|
-            url == "#{@index.url}/article"
+            url == "#{@index.url}/article/"
           end.returns(mock_response('{"ok":true,"_id":"test"}'))
           @index.store :type => 'article', :title => 'Test'
         end
 
         should "set type from Hash :_type property" do
           Configuration.client.expects(:post).with do |url,document|
-            url == "#{@index.url}/article"
+            url == "#{@index.url}/article/"
           end.returns(mock_response('{"ok":true,"_id":"test"}'))
           @index.store :_type => 'article', :title => 'Test'
         end
 
         should "set type from Object _type method" do
           Configuration.client.expects(:post).with do |url,document|
-            url == "#{@index.url}/article"
+            url == "#{@index.url}/article/"
           end.returns(mock_response('{"ok":true,"_id":"test"}'))
 
           article = Class.new do
@@ -242,7 +237,7 @@ module Tire
 
         should "set type from Object type method" do
           Configuration.client.expects(:post).with do |url,document|
-            url == "#{@index.url}/article"
+            url == "#{@index.url}/article/"
           end.returns(mock_response('{"ok":true,"_id":"test"}'))
 
           article = Class.new do
@@ -254,7 +249,7 @@ module Tire
 
         should "properly encode namespaced document types" do
           Configuration.client.expects(:post).with do |url,document|
-            url == "#{@index.url}/my_namespace%2Fmy_model"
+            url == "#{@index.url}/my_namespace%2Fmy_model/"
           end.returns(mock_response('{"ok":true,"_id":"123"}'))
 
           module MyNamespace
@@ -268,7 +263,7 @@ module Tire
         end
 
         should "set default type" do
-          Configuration.client.expects(:post).with("#{@index.url}/document", '{"title":"Test"}').returns(mock_response('{"ok":true,"_id":"test"}'))
+          Configuration.client.expects(:post).with("#{@index.url}/document/", '{"title":"Test"}').returns(mock_response('{"ok":true,"_id":"test"}'))
           @index.store :title => 'Test'
         end
 
@@ -315,7 +310,7 @@ module Tire
           Configuration.reset :wrapper
 
           Configuration.client.stubs(:post).with do |url, payload|
-                                              url     == "#{@index.url}/article" &&
+                                              url     == "#{@index.url}/article/" &&
                                               payload =~ /"title":"Test"/
                                             end.
                                             returns(mock_response('{"ok":true,"_id":"id-1"}'))
@@ -452,6 +447,23 @@ module Tire
           @index.bulk_store [ {:id => '1', :title => 'One'}, {:id => '2', :title => 'Two'} ]
         end
 
+        should "serialize ActiveModel instances" do
+          Configuration.client.expects(:post).with do |url, json|
+            url  == "#{ActiveModelArticle.index.url}/_bulk" &&
+            json =~ /"_index":"active_model_articles"/ &&
+            json =~ /"_type":"active_model_article"/ &&
+            json =~ /"_id":"1"/ &&
+            json =~ /"_id":"2"/ &&
+            json =~ /"title":"One"/ &&
+            json =~ /"title":"Two"/
+          end.returns(mock_response('{}', 200))
+
+          one = ActiveModelArticle.new 'title' => 'One'; one.id = '1'
+          two = ActiveModelArticle.new 'title' => 'Two'; two.id = '2'
+
+          ActiveModelArticle.index.bulk_store [ one, two ]
+        end
+
         context "namespaced models" do
           should "not URL-escape the document_type" do
             Configuration.client.expects(:post).with do |url, json|
@@ -502,13 +514,13 @@ module Tire
 
         should "display error message when collection item does not have ID" do
           Configuration.client.expects(:post).with do |url, json|
-            url  == "#{@index.url}/_bulk"
+            url  == "#{ActiveModelArticle.index.url}/_bulk"
           end.returns(mock_response('success', 200))
 
           STDERR.expects(:puts).once
 
           documents = [ { :title => 'Bogus' }, { :title => 'Real', :id => 1 } ]
-          @index.bulk_store documents
+          ActiveModelArticle.index.bulk_store documents
         end
 
       end
@@ -733,7 +745,7 @@ module Tire
           should "percolate document against all registered queries" do
             Configuration.client.expects(:post).
                                  with do |url, payload|
-                                   url     == "#{@index.url}/article?percolate=%2A" &&
+                                   url     == "#{@index.url}/article/?percolate=*" &&
                                    payload =~ /"title":"Test"/
                                  end.
                                  returns(mock_response('{"ok":true,"_id":"test","matches":["alerts"]}'))
@@ -743,7 +755,7 @@ module Tire
           should "percolate document against specific queries" do
             Configuration.client.expects(:post).
                                  with do |url, payload|
-                                   url     == "#{@index.url}/article?percolate=tag%3Aalerts" &&
+                                   url     == "#{@index.url}/article/?percolate=tag:alerts" &&
                                    payload =~ /"title":"Test"/
                                  end.
                                  returns(mock_response('{"ok":true,"_id":"test","matches":["alerts"]}'))
@@ -811,10 +823,10 @@ module Tire
           def index_something
             @tags = ['block', 'scope', 'revenge']
 
-            Index.any_instance.expects(:store).with(:title => 'Title From Outer Space', :tags => ['block', 'scope', 'revenge'])
+            Index.any_instance.expects(:store).with(title: 'Title From Outer Space', tags: ['block', 'scope', 'revenge'])
 
             Tire::Index.new 'outer-space' do |index|
-              index.store(:title => @my_title, :tags => @tags)
+              index.store title: @my_title, tags: @tags
             end
           end
 

data/test/unit/model_callbacks_test.rb

@@ -0,0 +1,116 @@
+require 'test_helper'
+
+class ModelOne
+  extend  ActiveModel::Naming
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  def save;       false; end
+  def destroy;    false; end
+end
+
+class ModelTwo
+  extend ActiveModel::Naming
+  extend ActiveModel::Callbacks
+  define_model_callbacks :save, :destroy
+
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  def save;    _run_save_callbacks {};                       end
+  def destroy; _run_destroy_callbacks { @destroyed = true }; end
+
+  def destroyed?; !!@destroyed; end
+end
+
+class ModelThree
+  extend ActiveModel::Naming
+  extend ActiveModel::Callbacks
+  define_model_callbacks :save, :destroy
+
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  def save;    _run_save_callbacks {};    end
+  def destroy; _run_destroy_callbacks {}; end
+end
+
+class ModelWithoutTireAutoCallbacks
+  extend ActiveModel::Naming
+  extend ActiveModel::Callbacks
+  define_model_callbacks :save, :destroy
+
+  include Tire::Model::Search
+  # DO NOT include Callbacks
+
+  def save;    _run_save_callbacks {};    end
+  def destroy; _run_destroy_callbacks {}; end
+end
+
+module Tire
+  module Model
+
+    class ModelCallbacksTest < Test::Unit::TestCase
+
+      context "Model without ActiveModel callbacks" do
+
+        should "not execute any callbacks" do
+          m = ModelOne.new
+          m.tire.expects(:update_index).never
+
+          m.save
+          m.destroy
+        end
+
+      end
+
+      context "Model with ActiveModel callbacks and implemented destroyed? method" do
+
+        should "execute the callbacks" do
+          m = ModelTwo.new
+          m.tire.expects(:update_index).twice
+
+          m.save
+          m.destroy
+        end
+
+      end
+
+      context "Model with ActiveModel callbacks without destroyed? method implemented" do
+
+        should "have the destroyed? method added" do
+          assert_respond_to ModelThree.new, :destroyed?
+        end
+
+        should "execute the callbacks" do
+          m = ModelThree.new
+          m.tire.expects(:update_index).twice
+
+          m.save
+          m.destroy
+        end
+
+      end
+
+      context "Model without Tire::Callbacks included" do
+
+        should "respond to Tire update_index callbacks" do
+          assert_respond_to ModelWithoutTireAutoCallbacks, :after_update_elasticsearch_index
+          assert_respond_to ModelWithoutTireAutoCallbacks, :before_update_elasticsearch_index
+        end
+
+        should "not execute the update_index hooks" do
+          m = ModelWithoutTireAutoCallbacks.new
+          m.tire.expects(:update_index).never
+
+          m.save
+          m.destroy
+        end
+      end
+
+      # ---------------------------------------------------------------------------
+
+    end
+
+  end
+end

data/test/unit/model_import_test.rb

@@ -0,0 +1,71 @@
+require 'test_helper'
+
+class ImportModel
+  extend  ActiveModel::Naming
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  DATA = (1..4).to_a
+
+  def self.paginate(options={})
+    options = {:page => 1, :per_page => 1000}.update options
+    DATA.slice( (options[:page]-1)*options[:per_page]...options[:page]*options[:per_page] )
+  end
+
+  def self.all(options={})
+    DATA
+  end
+
+  def self.count
+    DATA.size
+  end
+end
+
+module Tire
+  module Model
+
+    class ImportTest < Test::Unit::TestCase
+
+      context "Model::Import" do
+
+        should "have the import method" do
+          assert_respond_to ImportModel, :import
+        end
+
+        should "paginate the results by default when importing" do
+          Tire::Index.any_instance.expects(:bulk_store).returns(true).times(2)
+
+          ImportModel.import :per_page => 2
+        end
+
+        should "call the passed block on every batch, and NOT manipulate the documents array" do
+          Tire::Index.any_instance.expects(:bulk_store).with { |c,o| c == [1, 2] }
+          Tire::Index.any_instance.expects(:bulk_store).with { |c,o| c == [3, 4] }
+
+          runs = 0
+          ImportModel.import :per_page => 2 do |documents|
+            runs += 1
+            # Don't forget to return the documents at the end of the block
+            documents
+          end
+
+          assert_equal 2, runs
+        end
+
+        should "manipulate the documents in passed block" do
+          Tire::Index.any_instance.expects(:bulk_store).with { |c,o| c == [2, 3] }
+          Tire::Index.any_instance.expects(:bulk_store).with { |c,o| c == [4, 5] }
+
+          ImportModel.import :per_page => 2 do |documents|
+            # Add 1 to every "document" and return them
+            documents.map { |d| d + 1 }
+          end
+
+        end
+
+      end
+
+    end
+
+  end
+end

data/test/unit/model_persistence_test.rb

@@ -0,0 +1,516 @@
+require 'test_helper'
+
+module Tire
+  module Model
+
+    class PersistenceTest < Test::Unit::TestCase
+
+      context "Model" do
+
+        should "have default index name" do
+          assert_equal 'persistent_articles', PersistentArticle.index_name
+          assert_equal 'persistent_articles', PersistentArticle.new(:title => 'Test').index_name
+        end
+
+        should "allow to set custom index name" do
+          assert_equal 'custom-index-name', PersistentArticleWithCustomIndexName.index_name
+
+          PersistentArticleWithCustomIndexName.index_name "another-index-name"
+          assert_equal 'another-index-name', PersistentArticleWithCustomIndexName.index_name
+          assert_equal 'another-index-name', PersistentArticleWithCustomIndexName.index.name
+        end
+
+        context "with index prefix" do
+          setup do
+            Model::Search.index_prefix 'prefix'
+          end
+          
+          teardown do
+            Model::Search.index_prefix nil
+          end
+          
+          should "have configured prefix in index_name" do
+            assert_equal 'prefix_persistent_articles', PersistentArticle.index_name
+            assert_equal 'prefix_persistent_articles', PersistentArticle.new(:title => 'Test').index_name
+          end
+          
+        end
+
+        should "have document_type" do
+          assert_equal 'persistent_article', PersistentArticle.document_type
+          assert_equal 'persistent_article', PersistentArticle.new(:title => 'Test').document_type
+        end
+
+        should "allow to define property" do
+          assert_nothing_raised do
+            a = PersistentArticle.new
+            class << a
+              property :status
+            end
+          end
+        end
+
+      end
+
+      context "Finders" do
+
+        setup do
+          @first  = { '_id' => 1, '_version' => 1, '_index' => 'persistent_articles', '_type' => 'persistent_article', '_source' => { :title => 'First'  } }
+          @second = { '_id' => 2, '_index' => 'persistent_articles', '_type' => 'persistent_article', '_source' => { :title => 'Second' } }
+          @third  = { '_id' => 3, '_index' => 'persistent_articles', '_type' => 'persistent_article', '_source' => { :title => 'Third'  } }
+          @find_all = { 'hits' => { 'hits' => [
+            @first,
+            @second,
+            @third
+          ] } }
+          @find_first = { 'hits' => { 'hits' => [ @first ] } }
+          @find_last_two = { 'hits' => { 'hits' => [ @second, @third ] } }
+          @find_twenty_ids = { 'hits' => { 'hits' => 20.times.map { @first }   } }
+        end
+
+        should "find document by numeric ID" do
+          Configuration.client.expects(:get).returns(mock_response(@first.to_json))
+          document = PersistentArticle.find 1
+
+          assert_instance_of PersistentArticle, document
+          assert_equal 1, document.id
+          assert_equal 1, document.attributes['id']
+          assert_equal 'First', document.attributes['title']
+          assert_equal 'First', document.title
+        end
+
+        should "have _type, _index, _id, _version attributes" do
+          Configuration.client.expects(:get).returns(mock_response(@first.to_json))
+          document = PersistentArticle.find 1
+
+          assert_instance_of PersistentArticle, document
+          assert_equal 1, document.id
+          assert_equal 1, document.attributes['id']
+          assert_equal 'persistent_articles', document._index
+          assert_equal 'persistent_article', document._type
+          assert_equal 1, document._version
+        end
+
+        should "find document by string ID" do
+          Configuration.client.expects(:get).returns(mock_response(@first.to_json))
+          document = PersistentArticle.find '1'
+
+          assert_instance_of PersistentArticle, document
+          assert_equal 1, document.id
+          assert_equal 1, document.attributes['id']
+          assert_equal 'First', document.attributes['title']
+          assert_equal 'First', document.title
+        end
+
+        should "find document by list of IDs" do
+          Configuration.client.expects(:get).returns(mock_response(@find_last_two.to_json))
+          documents = PersistentArticle.find 2, 3
+
+          assert_equal 2, documents.count
+        end
+
+        should "find document by array of IDs" do
+          Configuration.client.expects(:get).returns(mock_response(@find_last_two.to_json))
+          documents = PersistentArticle.find [2, 3]
+
+          assert_equal 2, documents.count
+        end
+
+        should "find all documents listed in IDs array" do
+          ids = (1..20).to_a
+          Configuration.client.expects(:get).returns(mock_response(@find_twenty_ids.to_json))
+          Tire::Search::Search.any_instance.expects(:size).with(ids.size)
+
+          documents = PersistentArticle.find ids
+          assert_equal ids.size, documents.count
+        end
+
+        should "find all documents" do
+          Configuration.client.stubs(:get).returns(mock_response(@find_all.to_json))
+          documents = PersistentArticle.all
+
+          assert_equal 3, documents.count
+          assert_equal 'First', documents.first.attributes['title']
+          assert_equal PersistentArticle.find(:all).map { |e| e.id }, PersistentArticle.all.map { |e| e.id }
+        end
+
+        should "find first document" do
+          Configuration.client.expects(:get).returns(mock_response(@find_first.to_json))
+          document = PersistentArticle.first
+
+          assert_equal 'First', document.attributes['title']
+        end
+
+        should "raise error when passing incorrect argument" do
+          assert_raise(ArgumentError) do
+             PersistentArticle.find :name => 'Test'
+          end
+        end
+
+        should_eventually "raise error when document is not found" do
+          assert_raise(DocumentNotFound) do
+             PersistentArticle.find 'xyz001'
+          end
+        end
+
+      end
+
+      context "Persistent model" do
+
+        setup { @article = PersistentArticle.new :title => 'Test', :tags => [:one, :two] }
+
+        context "attribute methods" do
+
+          should "allow to set attributes on initialization" do
+            assert_not_nil @article.attributes
+            assert_equal 'Test', @article.attributes['title']
+          end
+
+          should "allow to leave attributes blank on initialization" do
+            assert_nothing_raised { PersistentArticle.new }
+          end
+
+          should "have getter methods for attributes" do
+            assert_not_nil @article.title
+            assert_equal 'Test', @article.title
+            assert_equal [:one, :two], @article.tags
+          end
+
+          should "have getter methods for attribute passed as a String" do
+            article = PersistentArticle.new 'title' => 'Tony Montana'
+            assert_not_nil article.title
+            assert_equal   'Tony Montana', article.title
+          end
+
+          should "raise error when getting unknown attribute" do
+            assert_raise(NoMethodError) do
+              @article.krapulitz
+            end
+          end
+
+          should "not raise error when getting unset attribute" do
+            article = PersistentArticle.new :title => 'Test'
+
+            assert_nothing_raised { article.published_on }
+            assert_nil article.published_on
+          end
+
+          should "return default value for attribute" do
+            article = PersistentArticleWithDefaults.new :title => 'Test'
+            assert_equal [],    article.tags
+            assert_equal false, article.hidden
+          end
+
+          should "not affect default value" do
+            article = PersistentArticleWithDefaults.new :title => 'Test'
+            article.tags << "ruby"
+
+            article.options[:switches] << "switch_1"
+
+            assert_equal [], PersistentArticleWithDefaults.new.tags
+            assert_equal [], PersistentArticleWithDefaults.new.options[:switches]
+           end
+
+          should "have query method for attribute" do
+            assert_equal true, @article.title?
+          end
+
+          should "raise error when querying for unknown attribute" do
+            assert_raise(NoMethodError) do
+              @article.krapulitz?
+            end
+          end
+
+          should "not raise error when querying for unset attribute" do
+            article = PersistentArticle.new :title => 'Test'
+
+            assert_nothing_raised { article.published_on? }
+            assert ! article.published_on?
+          end
+
+          should "return true for respond_to? calls for set attributes" do
+            article = PersistentArticle.new :title => 'Test'
+            assert article.respond_to?(:title)
+          end
+
+          should "return false for respond_to? calls for unknown attributes" do
+            article = PersistentArticle.new :title => 'Test'
+            assert ! article.respond_to?(:krapulitz)
+          end
+
+          should "return true for respond_to? calls for defined but unset attributes" do
+            article = PersistentArticle.new :title => 'Test'
+
+            assert article.respond_to?(:published_on)
+          end
+
+          should "have attribute names" do
+            article = PersistentArticle.new :title => 'Test', :tags => ['one', 'two']
+            assert_equal ['published_on', 'tags', 'title'], article.attribute_names
+          end
+
+          should "have setter method for attribute" do
+            @article.title = 'Updated'
+            assert_equal 'Updated', @article.title
+            assert_equal 'Updated', @article.attributes['title']
+          end
+
+          should_eventually "allow to set deeply nested attributes on initialization" do
+            article = PersistentArticle.new :title => 'Test', :author => { :first_name => 'John', :last_name => 'Smith' }
+
+            assert_equal 'John',  article.author.first_name
+            assert_equal 'Smith', article.author.last_name
+            assert_equal({ :first_name => 'John', :last_name => 'Smith' }, article.attributes['author'])
+          end
+
+          should_eventually "allow to set deeply nested attributes on update" do
+            article = PersistentArticle.new :title => 'Test', :author => { :first_name => 'John', :last_name => 'Smith' }
+
+            article.author.first_name = 'Robert'
+            article.author.last_name  = 'Carpenter'
+
+            assert_equal 'Robert',    article.author.first_name
+            assert_equal 'Carpenter', article.author.last_name
+            assert_equal({ :first_name => 'Robert', :last_name => 'Carpenter' }, article.attributes['author'])
+          end
+
+        end
+
+        context "with casting" do
+
+          should "cast the value as custom class" do
+            article = PersistentArticleWithCastedItem.new :title => 'Test',
+                                                          :author => { :first_name => 'John', :last_name => 'Smith' }
+            assert_instance_of Author, article.author
+            assert_equal 'John', article.author.first_name
+          end
+
+          should "cast the value as collection of custom classes" do
+            article = PersistentArticleWithCastedCollection.new :title => 'Test',
+                                                                :comments => [{:nick => '4chan', :body => 'WHY U NO?'}]
+            assert_instance_of Array,   article.comments
+            assert_instance_of Comment, article.comments.first
+            assert_equal '4chan',       article.comments.first.nick
+          end
+
+          should "automatically format strings in UTC format as Time" do
+            article = PersistentArticle.new :published_on => '2011-11-01T23:00:00Z'
+            assert_instance_of Time, article.published_on
+            assert_equal 2011, article.published_on.year
+          end
+
+          should "cast anonymous Hashes as Hashr instances" do
+            article = PersistentArticleWithCastedItem.new :stats => { :views => 100, :meta => { :tags => 'A' }  }
+            assert_equal 100, article.stats.views
+            assert_equal 'A', article.stats.meta.tags
+          end
+
+        end
+
+        context "when creating" do
+
+          should "save the document with generated ID in the database" do
+            Configuration.client.expects(:post).
+                                 with do |url, payload|
+                                  doc = MultiJson.decode(payload)
+                                  url == "#{Configuration.url}/persistent_articles/persistent_article/" &&
+                                  doc['title'] == 'Test' &&
+                                  doc['tags']  == ['one', 'two']
+                                  doc['published_on'] == nil
+                                end.
+                                returns(mock_response('{"ok":true,"_id":"abc123"}'))
+            article = PersistentArticle.create :title => 'Test', :tags => [:one, :two]
+
+            assert article.persisted?, "#{article.inspect} should be `persisted?`"
+            assert_equal 'abc123', article.id
+          end
+
+          should "save the document with custom ID in the database" do
+            Configuration.client.expects(:post).
+                                 with do |url, payload|
+                                   doc = MultiJson.decode(payload)
+                                   url == "#{Configuration.url}/persistent_articles/persistent_article/r2d2" &&
+                                   doc['title'] == 'Test' &&
+                                   doc['published_on'] == nil
+                                 end.
+                                 returns(mock_response('{"ok":true,"_id":"r2d2"}'))
+            article = PersistentArticle.create :id => 'r2d2', :title => 'Test'
+
+            assert_equal 'r2d2', article.id
+          end
+
+          should "perform model validations" do
+            Configuration.client.expects(:post).never
+
+            assert ! ValidatedModel.create(:name => nil)
+          end
+
+        end
+
+        context "when creating" do
+
+          should "set the id property" do
+            Configuration.client.expects(:post).
+                                 with do |url, payload|
+                                   doc = MultiJson.decode(payload)
+                                   url == "#{Configuration.url}/persistent_articles/persistent_article/" &&
+                                   doc['title'] == 'Test'
+                                 end.
+                                 returns(mock_response('{"ok":true,"_id":"1"}'))
+
+            article = PersistentArticle.create :title => 'Test'
+            assert_equal '1', article.id
+          end
+
+          should "not set the id property if already set" do
+            Configuration.client.expects(:post).
+                                  with do |url, payload|
+                                    doc = MultiJson.decode(payload)
+                                    url == "#{Configuration.url}/persistent_articles/persistent_article/123" &&
+                                    doc['title'] == 'Test' &&
+                                    doc['published_on'] == nil
+                                  end.
+                                  returns(mock_response('{"ok":true, "_id":"XXX"}'))
+
+            article = PersistentArticle.create :id => '123', :title => 'Test'
+            assert_equal '123', article.id
+          end
+
+        end
+
+        context "when saving" do
+
+          should "save the document with updated attribute" do
+            article = PersistentArticle.new :id => '1', :title => 'Test'
+
+            Configuration.client.expects(:post).
+                                 with do |url, payload|
+                                   doc = MultiJson.decode(payload)
+                                   url == "#{Configuration.url}/persistent_articles/persistent_article/1" &&
+                                   doc['title'] == 'Test' &&
+                                   doc['published_on'] == nil
+                                 end.
+                                 returns(mock_response('{"ok":true,"_id":"1"}'))
+            assert article.save
+
+            article.title = 'Updated'
+
+            Configuration.client.expects(:post).
+                                 with do |url, payload|
+                                   doc = MultiJson.decode(payload)
+                                   url == "#{Configuration.url}/persistent_articles/persistent_article/1" &&
+                                   doc['title'] == 'Updated'
+                                 end.
+                                 returns(mock_response('{"ok":true,"_id":"1"}'))
+            assert article.save
+          end
+
+          should "perform validations" do
+            article = ValidatedModel.new :name => nil
+            assert ! article.save
+          end
+
+          should "set the id property itself" do
+            article = PersistentArticle.new
+            article.title = 'Test'
+
+            Configuration.client.expects(:post).with("#{Configuration.url}/persistent_articles/persistent_article/",
+                                                     article.to_indexed_json).
+                                                returns(mock_response('{"ok":true,"_id":"1"}'))
+             assert article.save
+             assert_equal '1', article.id
+          end
+
+          should "not set the id property if already set" do
+            article = PersistentArticle.new
+            article.id    = '456'
+            article.title = 'Test'
+
+            Configuration.client.expects(:post).
+                                 with do |url, payload|
+                                   doc = MultiJson.decode(payload)
+                                   url == "#{Configuration.url}/persistent_articles/persistent_article/456" &&
+                                   doc['title'] == 'Test'
+                                 end.
+                                 returns(mock_response('{"ok":true,"_id":"XXX"}'))
+             assert article.save
+             assert_equal '456', article.id
+          end
+
+        end
+
+        context "when destroying" do
+
+          should "delete the document from the database" do
+            Configuration.client.expects(:post).
+                                 with do |url, payload|
+                                   doc = MultiJson.decode(payload)
+                                   url == "#{Configuration.url}/persistent_articles/persistent_article/123" &&
+                                   doc['title'] == 'Test'
+                                 end.returns(mock_response('{"ok":true,"_id":"123"}'))
+
+            Configuration.client.expects(:delete).
+                                 with("#{Configuration.url}/persistent_articles/persistent_article/123").
+                                 returns(mock_response('{"ok":true,"acknowledged":true}', 200))
+
+            article = PersistentArticle.new :id => '123', :title => 'Test'
+            article.save
+            article.destroy
+          end
+
+        end
+
+        context "when updating attributes" do
+
+          should "update single attribute" do
+            @article.expects(:save).returns(true)
+
+            @article.update_attribute :title, 'Updated'
+            assert_equal 'Updated', @article.title
+          end
+
+          should "update all attributes" do
+            @article.expects(:save).returns(true)
+
+            @article.update_attributes :title => 'Updated', :tags => ['three']
+            assert_equal 'Updated', @article.title
+            assert_equal ['three'], @article.tags
+          end
+
+        end
+
+      end
+
+      context "Persistent model with mapping definition" do
+
+        should "create the index with mapping" do
+          expected = {
+            :settings => {},
+            :mappings => { :persistent_article_with_mapping => {
+              :properties => { :title => { :type => 'string', :analyzer => 'snowball', :boost => 10 } }
+            }}
+          }
+
+          Tire::Index.any_instance.stubs(:exists?).returns(false)
+          Tire::Index.any_instance.expects(:create).with(expected)
+
+          class ::PersistentArticleWithMapping
+
+            include Tire::Model::Persistence
+            include Tire::Model::Search
+            include Tire::Model::Callbacks
+
+            mapping do
+              property :title, :type => 'string', :analyzer => 'snowball', :boost => 10
+            end
+
+          end
+
+          assert_equal 'snowball', PersistentArticleWithMapping.mapping[:title][:analyzer]
+        end
+
+      end
+
+    end
+  end
+end