load_balanced_tire 0.1

data/test/unit/logger_test.rb

@@ -0,0 +1,125 @@
+require 'test_helper'
+require 'time'
+
+module Tire
+
+  class LoggerTest < Test::Unit::TestCase
+    include Tire
+
+    context "Logger" do
+
+      context "initialized with an IO object" do
+
+        should "take STDOUT" do
+          assert_nothing_raised do
+            logger = Logger.new STDOUT
+          end
+        end
+
+        should "write to STDERR" do
+          STDERR.expects(:write).with('BOOM!')
+          logger = Logger.new STDERR
+          logger.write('BOOM!')
+        end
+
+      end
+
+      context "initialized with file" do
+        teardown { File.delete('myfile.log') }
+
+        should "create the file" do
+          assert_nothing_raised do
+            logger = Logger.new 'myfile.log'
+            assert File.exists?('myfile.log')
+          end
+        end
+
+        should "write to file" do
+          File.any_instance.expects(:write).with('BOOM!')
+          logger = Logger.new 'myfile.log'
+          logger.write('BOOM!')
+        end
+
+      end
+
+    end
+
+    context "levels" do
+
+      should "have the default level" do
+        logger = Logger.new STDERR
+        assert_equal 'info', logger.level
+      end
+
+      should "set the level" do
+        logger = Logger.new STDERR, :level => 'debug'
+        assert_equal 'debug', logger.level
+      end
+
+    end
+
+    context "tracing requests" do
+      setup do
+        Time.stubs(:now).returns(Time.parse('2011-03-19 11:00'))
+        @logger = Logger.new STDERR
+      end
+
+      should "log request in correct format" do
+        log = (<<-"log;").gsub(/^ +/, '')
+          # 2011-03-19 11:00:00:000 [_search] (["articles", "users"])
+          #
+          curl -X GET http://...
+
+        log;
+        @logger.expects(:write).with do |payload|
+          payload =~ Regexp.new( Regexp.escape('2011-03-19 11:00:00') )
+          payload =~ Regexp.new( Regexp.escape('_search') )
+          payload =~ Regexp.new( Regexp.escape('(["articles", "users"])') )
+        end
+        @logger.log_request('_search', ["articles", "users"], 'curl -X GET http://...')
+      end
+
+      should "log response in correct format" do
+        json = (<<-"json;").gsub(/^\s*/, '')
+        {
+          "took" : 4,
+          "hits" : {
+            "total" : 20,
+            "max_score" : 1.0,
+            "hits" : [ {
+              "_index" : "articles",
+              "_type" : "article",
+              "_id" : "Hmg0B0VSRKm2VAlsasdnqg",
+              "_score" : 1.0, "_source" : { "title" : "Article 1",  "published_on" : "2011-01-01" }
+            }, {
+              "_index" : "articles",
+              "_type" : "article",
+              "_id" : "booSWC8eRly2I06GTUilNA",
+              "_score" : 1.0, "_source" : { "title" : "Article 2", "published_on" : "2011-01-12" }
+            }
+            ]
+          }
+        }
+        json;
+        log  = (<<-"log;").gsub(/^\s*/, '')
+          # 2011-03-19 11:00:00:000 [200 OK] (4 msec)
+          #
+        log;
+        # log += json.split.map { |line| "# #{line}" }.join("\n")
+        json.each_line { |line| log += "# #{line}" }
+        log += "\n\n"
+        @logger.expects(:write).with do |payload|
+          payload =~ Regexp.new( Regexp.escape('2011-03-19 11:00:00') )
+          payload =~ Regexp.new( Regexp.escape('[200 OK]') )
+          payload =~ Regexp.new( Regexp.escape('(4 msec)') )
+          payload =~ Regexp.new( Regexp.escape('took') )
+          payload =~ Regexp.new( Regexp.escape('hits') )
+          payload =~ Regexp.new( Regexp.escape('_score') )
+        end
+        @logger.log_response('200 OK', 4, json)
+      end
+
+    end
+
+  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,528 @@
+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 initializing" do
+
+          should "be a new record" do
+            article = PersistentArticle.new :title => 'Test'
+
+            assert   article.new_record?, "#{article.inspect} should be `new_record?`"
+            assert ! article.persisted?,  "#{article.inspect} should NOT be `persisted?`"
+          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","_version":1}'))
+            article = PersistentArticle.create :title => 'Test', :tags => [:one, :two]
+
+            assert   article.persisted?,  "#{article.inspect} should be `persisted?`"
+            assert ! article.new_record?, "#{article.inspect} should NOT be `new_record?`"
+            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