drogus-mongo_mapper 0.6.10

data/test/functional/test_logger.rb

@@ -0,0 +1,20 @@
+require 'test_helper'
+
+class LoggerTest < Test::Unit::TestCase
+  context "with connection that has logger" do
+    setup do
+      @output = StringIO.new
+      @logger = Logger.new(@output)
+      MongoMapper.connection = Mongo::Connection.new('127.0.0.1', 27017, :logger => @logger)
+    end
+
+    should "be able to get access to that logger" do
+      MongoMapper.logger.should == @logger
+    end
+    
+    should "be able to log messages" do
+      MongoMapper.logger.debug 'testing'
+      @output.string.include?('testing').should be_true
+    end
+  end
+end

data/test/functional/test_modifiers.rb

@@ -0,0 +1,252 @@
+require 'test_helper'
+
+class ModifierTest < Test::Unit::TestCase
+  def setup
+    @page_class = Doc do
+      key :title, String
+      key :day_count, Integer, :default => 0
+      key :week_count, Integer, :default => 0
+      key :month_count, Integer, :default => 0
+      key :tags, Array
+    end
+  end
+  
+  def assert_page_counts(page, day_count, week_count, month_count)
+    page.reload
+    page.day_count.should == day_count
+    page.week_count.should == week_count
+    page.month_count.should == month_count
+  end
+  
+  should "be able to increment with criteria and modifier hashes" do
+    page = @page_class.create(:title => 'Home')
+    page2 = @page_class.create(:title => 'Home')
+    
+    @page_class.increment({:title => 'Home'}, {
+      :day_count => 1, :week_count => 2, :month_count => 3
+    })
+    
+    assert_page_counts page, 1, 2, 3
+    assert_page_counts page2, 1, 2, 3
+  end
+  
+  should "be able to increment with ids and modifier hash" do
+    page  = @page_class.create(:title => 'Home')
+    page2 = @page_class.create(:title => 'Home')
+    
+    @page_class.increment(page.id, page2.id, {
+      :day_count => 1, :week_count => 2, :month_count => 3
+    })
+    
+    assert_page_counts page, 1, 2, 3
+    assert_page_counts page2, 1, 2, 3
+  end
+  
+  should "be able to decrement with criteria and modifier hashes" do
+    page = @page_class.create(:title => 'Home', :day_count => 1, :week_count => 2, :month_count => 3)
+    page2 = @page_class.create(:title => 'Home', :day_count => 1, :week_count => 2, :month_count => 3)
+    
+    @page_class.decrement({:title => 'Home'}, {
+      :day_count => 1, :week_count => 2, :month_count => 3
+    })
+    
+    assert_page_counts page, 0, 0, 0
+    assert_page_counts page2, 0, 0, 0
+  end
+  
+  should "be able to decrement with ids and modifier hash" do
+    page = @page_class.create(:title => 'Home', :day_count => 1, :week_count => 2, :month_count => 3)
+    page2 = @page_class.create(:title => 'Home', :day_count => 1, :week_count => 2, :month_count => 3)
+    
+    @page_class.decrement(page.id, page2.id, {
+      :day_count => 1, :week_count => 2, :month_count => 3
+    })
+    
+    assert_page_counts page, 0, 0, 0
+    assert_page_counts page2, 0, 0, 0
+  end
+  
+  should "always decrement when decrement is called whether number is positive or negative" do
+    page = @page_class.create(:title => 'Home', :day_count => 1, :week_count => 2, :month_count => 3)
+    page2 = @page_class.create(:title => 'Home', :day_count => 1, :week_count => 2, :month_count => 3)
+    
+    @page_class.decrement(page.id, page2.id, {
+      :day_count => -1, :week_count => 2, :month_count => -3
+    })
+    
+    assert_page_counts page, 0, 0, 0
+    assert_page_counts page2, 0, 0, 0
+  end
+  
+  should "be able to set with criteria and modifier hashes" do
+    page  = @page_class.create(:title => 'Home')
+    page2 = @page_class.create(:title => 'Home')
+    
+    @page_class.set({:title => 'Home'}, :title => 'Home Revised')
+    
+    page.reload
+    page.title.should == 'Home Revised'
+    
+    page2.reload
+    page2.title.should == 'Home Revised'
+  end
+  
+  should "be able to set with ids and modifier hash" do
+    page  = @page_class.create(:title => 'Home')
+    page2 = @page_class.create(:title => 'Home')
+    
+    @page_class.set(page.id, page2.id, :title => 'Home Revised')
+    
+    page.reload
+    page.title.should == 'Home Revised'
+    
+    page2.reload
+    page2.title.should == 'Home Revised'
+  end
+  
+  should "be able to push with criteria and modifier hashes" do
+    page  = @page_class.create(:title => 'Home')
+    page2 = @page_class.create(:title => 'Home')
+    
+    @page_class.push({:title => 'Home'}, :tags => 'foo')
+    
+    page.reload
+    page.tags.should == %w(foo)
+    
+    page2.reload
+    page.tags.should == %w(foo)
+  end
+  
+  should "be able to push with ids and modifier hash" do
+    page  = @page_class.create(:title => 'Home')
+    page2 = @page_class.create(:title => 'Home')
+    
+    @page_class.push(page.id, page2.id, :tags => 'foo')
+    
+    page.reload
+    page.tags.should == %w(foo)
+    
+    page2.reload
+    page.tags.should == %w(foo)
+  end
+
+  should "be able to push all with criteria and modifier hashes" do
+    page  = @page_class.create(:title => 'Home')
+    page2 = @page_class.create(:title => 'Home')
+    tags = %w(foo bar)
+    
+    @page_class.push_all({:title => 'Home'}, :tags => tags)
+    
+    page.reload
+    page.tags.should == tags
+    
+    page2.reload
+    page.tags.should == tags
+  end
+  
+  should "be able to push all with ids and modifier hash" do
+    page  = @page_class.create(:title => 'Home')
+    page2 = @page_class.create(:title => 'Home')
+    tags = %w(foo bar)
+    
+    @page_class.push_all(page.id, page2.id, :tags => tags)
+    
+    page.reload
+    page.tags.should == tags
+    
+    page2.reload
+    page.tags.should == tags
+  end
+  
+  should "be able to pull with criteria and modifier hashes" do
+    page  = @page_class.create(:title => 'Home', :tags => %w(foo bar))
+    page2 = @page_class.create(:title => 'Home', :tags => %w(foo bar))
+    
+    @page_class.pull({:title => 'Home'}, :tags => 'foo')
+    
+    page.reload
+    page.tags.should == %w(bar)
+    
+    page2.reload
+    page.tags.should == %w(bar)
+  end
+  
+  should "be able to pull with ids and modifier hash" do
+    page  = @page_class.create(:title => 'Home', :tags => %w(foo bar))
+    page2 = @page_class.create(:title => 'Home', :tags => %w(foo bar))
+    
+    @page_class.pull(page.id, page2.id, :tags => 'foo')
+    
+    page.reload
+    page.tags.should == %w(bar)
+    
+    page2.reload
+    page.tags.should == %w(bar)
+  end
+
+  should "be able to pull all with criteria and modifier hashes" do
+    page  = @page_class.create(:title => 'Home', :tags => %w(foo bar baz))
+    page2 = @page_class.create(:title => 'Home', :tags => %w(foo bar baz))
+    
+    @page_class.pull_all({:title => 'Home'}, :tags => %w(foo bar))
+    
+    page.reload
+    page.tags.should == %w(baz)
+    
+    page2.reload
+    page.tags.should == %w(baz)
+  end
+  
+  should "be able to pull all with ids and modifier hash" do
+    page  = @page_class.create(:title => 'Home', :tags => %w(foo bar baz))
+    page2 = @page_class.create(:title => 'Home', :tags => %w(foo bar baz))
+    
+    @page_class.pull_all(page.id, page2.id, :tags => %w(foo bar))
+    
+    page.reload
+    page.tags.should == %w(baz)
+    
+    page2.reload
+    page.tags.should == %w(baz)
+  end
+
+  should "be able to push uniq with criteria and modifier hash" do
+    page  = @page_class.create(:title => 'Home', :tags => 'foo')
+    page2 = @page_class.create(:title => 'Home')
+    
+    @page_class.push_uniq({:title => 'Home'}, :tags => 'foo')
+    
+    page.reload
+    page.tags.should == %w(foo)
+    
+    page2.reload
+    page.tags.should == %w(foo)
+  end
+  
+  should "be able to push uniq with ids and modifier hash" do
+    page  = @page_class.create(:title => 'Home', :tags => 'foo')
+    page2 = @page_class.create(:title => 'Home')
+    
+    @page_class.push_uniq(page.id, page2.id, :tags => 'foo')
+    
+    page.reload
+    page.tags.should == %w(foo)
+    
+    page2.reload
+    page.tags.should == %w(foo)
+  end
+  
+  should "be able to remove the last element the array" do
+    page  = @page_class.create(:title => 'Home', :tags => %w(foo bar))
+    @page_class.pop(page.id, :tags => 1)
+    page.reload
+    page.tags.should == %w(foo)
+  end
+
+  should "be able to remove the first element of the array" do
+    page  = @page_class.create(:title => 'Home', :tags => %w(foo bar))
+    @page_class.pop(page.id, :tags => -1)
+    page.reload
+    page.tags.should == %w(bar)
+  end
+end

data/test/functional/test_pagination.rb

@@ -0,0 +1,93 @@
+require 'test_helper'
+
+class PaginationTest < Test::Unit::TestCase
+  context "Paginating" do
+    setup do
+      @document = Doc do
+        set_collection_name 'users'
+
+        key :first_name, String
+        key :last_name, String
+        key :age, Integer
+        
+        def self.per_page; 1 end
+      end
+      
+      @doc1 = @document.create({:first_name => 'John', :last_name => 'Nunemaker', :age => '27'})
+      @doc2 = @document.create({:first_name => 'Steve', :last_name => 'Smith', :age => '28'})
+      @doc3 = @document.create({:first_name => 'Steph', :last_name => 'Nunemaker', :age => '26'})
+    end
+
+    should "return the total pages" do
+      result = @document.paginate(:per_page => 2, :page => 1)
+      result.total_pages.should == 2
+    end
+    
+    should "return the total pages when defaulting to the document class per_page" do
+      result = @document.paginate(:page => 1)
+      result.total_pages.should == 3
+    end
+    
+    should "return the total of records" do
+      result = @document.paginate(:per_page => 2, :page => 1)
+      result.total_entries.should == 3
+    end
+
+    should "return the items" do
+      result = @document.paginate(:per_page => 2, :page => 1, :order => 'first_name')
+      result.size.should == 2
+      result.should == [@doc1, @doc3]
+    end
+
+    should "accept conditions" do
+      result = @document.paginate({
+        :last_name => 'Nunemaker',
+        :order     => "age DESC",
+        :per_page  => 2, 
+        :page      => 1,
+      })
+      result.should == [@doc1, @doc3]
+      result.first.age.should == 27
+
+      result = @document.paginate({
+        :conditions => {:last_name => 'Nunemaker'},
+        :order      => "age DESC",
+        :per_page   => 2,
+        :page       => 1} )
+      result.should == [@doc1, @doc3]
+      result.first.age.should == 27
+    end
+
+    should "withstand rigor" do      
+      result = @document.paginate({
+        :per_page  => 1, 
+        :page      => 1,
+        :order     => 'age desc', 
+        :last_name => 'Nunemaker'
+      })
+      result.should == [@doc1]
+      result.total_entries.should == 2
+      result.total_pages.should == 2
+      
+      result = @document.paginate({
+        :per_page  => 1,
+        :page      => 2,
+        :order     => 'age desc',
+        :last_name => 'Nunemaker'
+      })
+      result.should == [@doc3]
+      result.total_entries.should == 2
+      result.total_pages.should == 2
+      
+      result = @document.paginate({
+        :per_page  => 2,
+        :page      => 1,
+        :order     => 'age desc',
+        :last_name => 'Nunemaker'
+      })
+      result.should == [@doc1, @doc3]
+      result.total_entries.should == 2
+      result.total_pages.should == 1
+    end
+  end
+end

data/test/functional/test_protected.rb

@@ -0,0 +1,155 @@
+require 'test_helper'
+
+class ProtectedTest < Test::Unit::TestCase
+  context 'A document with protected attributes' do
+    setup do
+      @doc_class = Doc do
+        key :name, String
+        key :admin, Boolean, :default => false
+
+        attr_protected :admin
+      end
+
+      @doc = @doc_class.create(:name => 'Steve Sloan')
+    end
+
+    should 'have protected attributes class method' do
+      @doc_class.protected_attributes.should == [:admin].to_set
+    end
+    
+    should "default protected attributes to nil" do
+      Doc().protected_attributes.should be_nil
+    end
+
+    should "have protected attributes instance method" do
+      @doc.protected_attributes.should equal(@doc_class.protected_attributes)
+    end
+
+    should "work with :protected shortcut when defining key" do
+      Doc() do
+        key :user_id, ObjectId, :protected => true
+      end.protected_attributes.should == [:user_id].to_set
+    end
+
+    should 'assign protected attribute through accessor' do
+      @doc.admin = true
+      @doc.admin.should be_true
+    end
+
+    should "ignore protected attribute on #initialize" do
+      doc = @doc_class.new(:name => 'John', :admin => true)
+      doc.admin.should be_false
+      doc.name.should == 'John'
+    end
+
+    should "not ignore protected attributes on #initialize from the database" do
+      doc = @doc_class.new(:name => 'John')
+      doc.admin = true
+      doc.save!
+      
+      doc = @doc_class.first(:name => 'John')
+      doc.admin.should be_true
+      doc.name.should == 'John'
+    end
+
+    should 'ignore protected attribute on #update_attributes' do
+      @doc.update_attributes(:name => 'Ren Hoek', :admin => true)
+      @doc.name.should == 'Ren Hoek'
+      @doc.admin.should be_false
+    end
+
+    should 'ignore protected attribute on #update_attributes!' do
+      @doc.update_attributes!(:name => 'Stimpson J. Cat', :admin => true)
+      @doc.name.should == 'Stimpson J. Cat'
+      @doc.admin.should be_false
+    end
+  end
+  
+  context "Single collection inherited protected attributes" do
+    setup do
+      class ::GrandParent
+        include MongoMapper::Document
+
+        key :_type, String
+        key :site_id, ObjectId
+
+        attr_protected :site_id
+      end
+      GrandParent.collection.remove
+
+      class ::Child < ::GrandParent
+        key :position, Integer
+
+        attr_protected :position
+      end
+
+      class ::GrandChild < ::Child; end
+
+      class ::OtherChild < ::GrandParent
+        key :blog_id, ObjectId
+
+        attr_protected :blog_id
+      end
+    end
+
+    teardown do
+      Object.send :remove_const, 'GrandParent' if defined?(::GrandParent)
+      Object.send :remove_const, 'Child'       if defined?(::Child)
+      Object.send :remove_const, 'GrandChild'  if defined?(::GrandChild)
+      Object.send :remove_const, 'OtherChild'  if defined?(::OtherChild)
+    end
+
+    should "share keys down the inheritance trail" do
+      GrandParent.protected_attributes.should == [:site_id].to_set
+      Child.protected_attributes.should == [:site_id, :position].to_set
+      GrandChild.protected_attributes.should == [:site_id, :position].to_set
+      OtherChild.protected_attributes.should == [:site_id, :blog_id].to_set
+    end
+  end
+
+  context 'An embedded document with protected attributes' do
+    setup do
+      @doc_class = Doc('Project')
+      @edoc_class = EDoc('Person') do
+        key :name, String
+        key :admin, Boolean, :default => false
+
+        attr_protected :admin
+      end
+      @doc_class.many :people, :class => @edoc_class
+
+      @doc = @doc_class.create(:title => 'MongoMapper')
+      @edoc = @edoc_class.new(:name => 'Steve Sloan')
+      @doc.people << @edoc
+    end
+
+    should 'have protected attributes class method' do
+      @edoc_class.protected_attributes.should == [:admin].to_set
+    end
+    
+    should "default protected attributes to nil" do
+      EDoc().protected_attributes.should be_nil
+    end
+
+    should "have protected attributes instance method" do
+      @edoc.protected_attributes.should equal(@edoc_class.protected_attributes)
+    end
+
+    should 'assign protected attribute through accessor' do
+      @edoc.admin = true
+      @edoc.admin.should be_true
+    end
+
+    should 'ignore protected attribute on #update_attributes' do
+      @edoc.update_attributes(:name => 'Ren Hoek', :admin => true)
+      @edoc.name.should == 'Ren Hoek'
+      @edoc.admin.should be_false
+    end
+
+    should 'ignore protected attribute on #update_attributes!' do
+      @edoc.update_attributes!(:name => 'Stimpson J. Cat', :admin => true)
+      @edoc.name.should == 'Stimpson J. Cat'
+      @edoc.admin.should be_false
+    end
+  end
+end