mongo_mapper 0.7.0 → 0.7.1

data/test/functional/test_embedded_document.rb

@@ -3,51 +3,56 @@ require 'models'
 
 class EmbeddedDocumentTest < Test::Unit::TestCase
   def setup
-    @klass = Doc do
-      key :first_name, String
-      key :last_name, String
+    @klass = Doc('Person') do
+      key :name, String
     end
-    
-    @pet_klass = EDoc do
+
+    @pet_klass = EDoc('Pet') do
       key :name, String
     end
-    
+
     @klass.many :pets, :class => @pet_klass
-    
-    @address_class = EDoc do
+
+    @address_class = EDoc('Address') do
       key :city, String
       key :state, String
     end
   end
-    
-  context "Saving a document with an embedded document" do
+
+  context "Saving a document with a key that is an embedded document" do
     setup do
       @klass.key :foo, @address_class
-      
-      @address = @address_class.new(:city => 'South Bend', :state => 'IN')
-      @doc = @klass.new(:foo => @address)
     end
-    
+
     should "embed embedded document" do
-      @doc.save
-      @doc.foo.city.should == 'South Bend'
-      @doc.foo.state.should == 'IN'
+      address = @address_class.new(:city => 'South Bend', :state => 'IN')
+      doc = @klass.create(:foo => address)
+      doc.foo.city.should == 'South Bend'
+      doc.foo.state.should == 'IN'
 
-      doc = @doc.reload
+      doc = doc.reload
       doc.foo.city.should == 'South Bend'
       doc.foo.state.should == 'IN'
     end
+
+    should "assign _parent_document and _root_document" do
+      address = @address_class.new(:city => 'South Bend', :state => 'IN')
+      address._parent_document.should be_nil
+      doc = @klass.create(:foo => address)
+      address._parent_document.should be(doc)
+      address._root_document.should be(doc)
+    end
   end
-  
+
   should "correctly instantiate single collection inherited embedded documents" do
     document = Doc('Foo') do
       key :message, Message
     end
-    
+
     doc1 = document.create(:message => Enter.new)
     doc1.reload.message.class.should be(Enter)
   end
-  
+
   context "new?" do
     setup do
       @klass.key :foo, @address_class
@@ -58,65 +63,65 @@ class EmbeddedDocumentTest < Test::Unit::TestCase
       doc = @klass.new(:foo => address)
       address.new?.should == true
     end
-    
+
     should "not be new after document is saved" do
       address = @address_class.new(:city => 'South Bend', :state => 'IN')
       doc = @klass.new(:foo => address)
       doc.save
       doc.foo.new?.should == false
     end
-    
+
     should "not be new when document is read back" do
       address = @address_class.new(:city => 'South Bend', :state => 'IN')
       doc = @klass.new(:foo => address)
       doc.save
-      
+
       doc = doc.reload
       doc.foo.new?.should == false
     end
   end
-  
+
   should "be able to save" do
     person = @klass.create
-    
+
     pet = @pet_klass.new(:name => 'sparky')
     person.pets << pet
     pet.should be_new
     pet.save
     pet.should_not be_new
-    
+
     person.reload
     person.pets.first.should == pet
   end
-  
+
   should "be able to dynamically add new keys and save" do
     person = @klass.create
-    
+
     pet = @pet_klass.new(:name => 'sparky', :crazy_key => 'crazy')
     person.pets << pet
     pet.save
-    
+
     person.reload
     person.pets.first.crazy_key.should == 'crazy'
   end
-  
+
   should "be able to update_attributes" do
     pet = @pet_klass.new(:name => 'sparky')
     person = @klass.create(:pets => [pet])
     person.reload
     pet = person.pets.first
-    
+
     pet.update_attributes(:name => 'koda').should be_true
     person.reload
     person.pets.first._id.should == pet._id
     person.pets.first.name.should == 'koda'
   end
-  
+
   should "be able to update_attributes!" do
     person = @klass.create(:pets => [@pet_klass.new(:name => 'sparky')])
     person.reload
     pet = person.pets.first
-    
+
     attributes = {:name => 'koda'}
     pet.expects(:attributes=).with(attributes)
     pet.expects(:save!)

data/test/functional/test_indexing.rb

@@ -0,0 +1,44 @@
+require 'test_helper'
+
+class IndexingTest < Test::Unit::TestCase
+  context "Indexing" do
+    setup do
+      @document = Doc do
+        set_collection_name 'users'
+
+        key :first_name, String
+        key :last_name, String
+        key :age, Integer
+        key :date, Date
+      end
+      drop_indexes(@document)
+    end
+
+    should "allow creating index for a key" do
+      @document.ensure_index :first_name
+      @document.should have_index('first_name_1')
+    end
+
+    should "allow creating unique index for a key" do
+      @document.ensure_index :first_name, :unique => true
+      @document.should have_index('first_name_1')
+    end
+
+    should "allow creating index on multiple keys" do
+      @document.ensure_index [[:first_name, 1], [:last_name, -1]]
+
+      # order is different for different versions of ruby so instead of
+      # just checking have_index('first_name_1_last_name_-1') I'm checking
+      # the values of the indexes to make sure the index creation was successful
+      @document.collection.index_information.detect do |index|
+        keys = index[1]
+        keys.include?(['first_name', 1]) && keys.include?(['last_name', -1])
+      end.should_not be_nil
+    end
+
+    should "work with :index shortcut when defining key" do
+      @document.key :father, String, :index => true
+      @document.should have_index('father_1')
+    end
+  end
+end

data/test/functional/test_modifiers.rb

@@ -10,243 +10,313 @@ class ModifierTest < Test::Unit::TestCase
       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
+  context "using class methods" do
+    should "be able to increment with criteria and modifier hashes" do
+      page = @page_class.create(:title => 'Home')
+      page2 = @page_class.create(:title => 'Home')
 
-  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
+      @page_class.increment({:title => 'Home'}, {
+        :day_count => 1, :week_count => 2, :month_count => 3
+      })
 
-  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)
+      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
 
-  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)
+  context "using instance methods" do
+    should "be able to increment with modifier hashes" do
+      page = @page_class.create
+
+      page.increment({:day_count => 1, :week_count => 2, :month_count => 3})
+
+      assert_page_counts page, 1, 2, 3
+    end
+
+    should "be able to decrement with modifier hashes" do
+      page = @page_class.create(:day_count => 1, :week_count => 2, :month_count => 3)
+
+      page.decrement({:day_count => 1, :week_count => 2, :month_count => 3})
+
+      assert_page_counts page, 0, 0, 0
+    end
+
+    should "always decrement when decrement is called whether number is positive or negative" do
+      page = @page_class.create(:day_count => 1, :week_count => 2, :month_count => 3)
+
+      page.decrement({:day_count => -1, :week_count => 2, :month_count => -3})
+
+      assert_page_counts page, 0, 0, 0
+    end
+
+    should "be able to set with modifier hashes" do
+      page  = @page_class.create(:title => 'Home')
+
+      page.set(:title => 'Home Revised')
+
+      page.reload
+      page.title.should == 'Home Revised'
+    end
+
+    should "be able to push with modifier hashes" do
+      page = @page_class.create
+
+      page.push(:tags => 'foo')
+
+      page.reload
+      page.tags.should == %w(foo)
+    end
+
+    should "be able to pull with criteria and modifier hashes" do
+      page  = @page_class.create(:tags => %w(foo bar))
+
+      page.pull(:tags => 'foo')
+
+      page.reload
+      page.tags.should == %w(bar)
+    end
+
+    should "be able to push uniq with criteria and modifier hash" do
+      page  = @page_class.create(:tags => 'foo')
+      page2 = @page_class.create
+
+      page.push_uniq(:tags => 'foo')
+      page.push_uniq(:tags => 'foo')
+
+      page.reload
+      page.tags.should == %w(foo)
+
+      page2.reload
+      page.tags.should == %w(foo)
+    end
   end
+
 end