mongo_mapper 0.6.10 → 0.7.0

data/specs.watchr

@@ -1,6 +1,27 @@
+def growl(title, msg, img)
+  %x{growlnotify -m #{ msg.inspect} -t #{title.inspect} --image ~/.watchr/#{img}.png}
+end
+
+def form_growl_message(str)
+  results = str.split("\n").last
+  if results =~ /[1-9]\s(failure|error)s?/
+    growl "Test Results", "#{results}", "fail"
+  elsif results != ""
+    growl "Test Results", "#{results}", "pass"
+  end
+end
+
 def run(cmd)
   puts(cmd)
-  system(cmd)
+  output = ""
+  IO.popen(cmd) do |com|
+    com.each_char do |c|
+      print c
+      output << c
+      $stdout.flush
+    end
+  end
+  form_growl_message output
 end
 
 def run_test_file(file)
@@ -17,7 +38,7 @@ end
 
 watch('test/test_helper\.rb') { system('clear'); run_all_tests }
 watch('test/.*/test_.*\.rb') { |m| system('clear'); run_test_file(m[0]) }
-watch('lib/.*') { |m| related_test_files(m[0]).each { |file| system('clear'); run_test_file(file) } }
+watch('lib/.*') { |m| related_test_files(m[0]).each { |file| run_test_file(file) } }
 
 # Ctrl-\
 Signal.trap('QUIT') do

data/test/functional/associations/test_belongs_to_proxy.rb

@@ -3,24 +3,26 @@ require 'models'
 
 class BelongsToProxyTest < Test::Unit::TestCase
   def setup    
-    @post_class = Class.new do
-      include MongoMapper::Document
-    end
-    
-    @comment_class = Class.new do
-      include MongoMapper::Document
+    @post_class = Doc()
+    @comment_class = Doc do
       key :post_id, String
     end
-    @comment_class.belongs_to :post, :class => @post_class
     
-    @post_class.collection.remove
-    @comment_class.collection.remove
+    @comment_class.belongs_to :post, :class => @post_class
   end
   
   should "default to nil" do
     @comment_class.new.post.nil?.should be_true
   end
   
+  should "send object id to target" do
+    post = @post_class.new(:name => 'mongomapper')
+    comment = @comment_class.new(:name => 'Foo!', :post => post)
+    comment.save
+    
+    comment.post.object_id.should == comment.post.target.object_id
+  end
+  
   should "have boolean presence method" do
     comment = @comment_class.new(:name => 'Foo!')
     comment.post?.should be_false
@@ -78,7 +80,7 @@ class BelongsToProxyTest < Test::Unit::TestCase
       setup do
         Property.key :thing_id, ObjectId
         Property.belongs_to :thing, :dependent => :destroy
-        Thing.has_many :properties
+        Thing.many :properties
 
         @thing = Thing.create(:name => "Tree")
         @property1 = Property.create

data/test/functional/associations/test_many_documents_as_proxy.rb

@@ -131,23 +131,6 @@ class ManyDocumentsAsProxyTest < Test::Unit::TestCase
       @post2.save
     end
 
-    context "with :all" do
-      should "work" do
-        @post.comments.find(:all).should include(@comment1)
-        @post.comments.find(:all).should include(@comment2)
-      end
-
-      should "work with conditions" do
-        comments = @post.comments.find(:all, :body => 'comment1')
-        comments.should == [@comment1]
-      end
-
-      should "work with order" do
-        comments = @post.comments.find(:all, :order => 'body desc')
-        comments.should == [@comment2, @comment1]
-      end
-    end
-
     context "with #all" do
       should "work" do
         @post.comments.all.should include(@comment1)
@@ -171,9 +154,9 @@ class ManyDocumentsAsProxyTest < Test::Unit::TestCase
       end
 
       should "not work for id not in association" do
-        lambda {
+        assert_raises(MongoMapper::DocumentNotFound) do
           @post.comments.find!(@comment5._id)
-        }.should raise_error(MongoMapper::DocumentNotFound)
+        end
       end
     end
 
@@ -184,9 +167,9 @@ class ManyDocumentsAsProxyTest < Test::Unit::TestCase
       end
 
       should "not work for ids not in association" do
-        lambda {
+        assert_raises(MongoMapper::DocumentNotFound) do
           @post.comments.find!(@comment1._id, @comment2._id, @comment4._id)
-        }.should raise_error(MongoMapper::DocumentNotFound)
+        end
       end
     end
     

data/test/functional/associations/test_many_documents_proxy.rb

@@ -242,36 +242,30 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
     
     context "all" do
       should "work" do
-        @project1.statuses.find(:all, :order => "position asc").should == [@brand_new, @complete]
         @project1.statuses.all(:order => "position asc").should == [@brand_new, @complete]
       end
       
       should "work with conditions" do
-        @project1.statuses.find(:all, :name => 'Complete').should == [@complete]
         @project1.statuses.all(:name => 'Complete').should == [@complete]
       end
     end
     
     context "first" do
       should "work" do
-        @project1.statuses.find(:first, :order => 'name').should == @complete
         @project1.statuses.first(:order => 'name').should == @complete
       end
       
       should "work with conditions" do
-        @project1.statuses.find(:first, :name => 'Complete').should == @complete
         @project1.statuses.first(:name => 'Complete').should == @complete
       end
     end
     
     context "last" do
       should "work" do
-        @project1.statuses.find(:last, :order => "position asc").should == @complete
         @project1.statuses.last(:order => "position asc").should == @complete
       end
       
       should "work with conditions" do
-        @project1.statuses.find(:last, :order => 'position', :name => 'New').should == @brand_new
         @project1.statuses.last(:order => 'position', :name => 'New').should == @brand_new
       end
     end
@@ -295,9 +289,9 @@ class ManyDocumentsProxyTest < Test::Unit::TestCase
       end
       
       should "not work for ids not in association" do
-        lambda {
+        assert_raises(MongoMapper::DocumentNotFound) do
           @project1.statuses.find!(@brand_new.id, @complete.id, @archived.id)
-        }.should raise_error(MongoMapper::DocumentNotFound)
+        end
       end
     end
     

data/test/functional/associations/test_many_embedded_polymorphic_proxy.rb

@@ -21,30 +21,50 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
 
   should "be able to replace the association" do
     catalog = Catalog.new
-    catalog.medias = [Video.new("file" => "video.mpg", "length" => 3600)]
+    catalog.medias = [Video.new('file' => 'video.mpg', 'length' => 3600)]
     catalog.save.should be_true
 
     catalog = catalog.reload
     catalog.medias.size.should == 1
-    catalog.medias[0].file.should == "video.mpg"
+    catalog.medias[0].file.should == 'video.mpg'
+    catalog.medias[0].new?.should == false
   end
-
+  
+  context "count" do
+    should "default to 0" do
+      Catalog.new.medias.count.should == 0
+    end
+    
+    should 'return correct count if any are embedded' do
+      catalog = Catalog.new
+      catalog.medias = [
+        Video.new('file' => 'video.mpg', 'length' => 3600),
+        Music.new('file' => 'music.mp3', 'bitrate' => '128kbps'),
+        Image.new('file' => 'image.png', 'width' => 800, 'height' => 600)
+      ]
+      catalog.medias.count.should == 3
+      catalog.save.should be_true
+      catalog.reload
+      catalog.medias.count.should == 3
+    end
+  end
+  
   should "store different associations" do      
     catalog = Catalog.new
     catalog.medias = [
-      Video.new("file" => "video.mpg", "length" => 3600),
-      Music.new("file" => "music.mp3", "bitrate" => "128kbps"),
-      Image.new("file" => "image.png", "width" => 800, "height" => 600)
+      Video.new('file' => 'video.mpg', 'length' => 3600),
+      Music.new('file' => 'music.mp3', 'bitrate' => '128kbps'),
+      Image.new('file' => 'image.png', 'width' => 800, 'height' => 600)
     ]
     catalog.save.should be_true
     
     catalog = catalog.reload
     catalog.medias.size.should == 3
-    catalog.medias[0].file.should == "video.mpg"
+    catalog.medias[0].file.should == 'video.mpg'
     catalog.medias[0].length.should == 3600
-    catalog.medias[1].file.should == "music.mp3"
+    catalog.medias[1].file.should == 'music.mp3'
     catalog.medias[1].bitrate.should == "128kbps"
-    catalog.medias[2].file.should == "image.png"
+    catalog.medias[2].file.should == 'image.png'
     catalog.medias[2].width.should == 800
     catalog.medias[2].height.should == 600
   end
@@ -52,38 +72,38 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
   context "With modularized models" do
     should "set associations correctly" do
       fleet_attributes = { 
-        "name" => "My Fleet", 
-        "transports" => [
-          {"_type" => "TrModels::Ambulance", "license_plate" => "GGG123", "icu" => true},
-          {"_type" => "TrModels::Car", "license_plate" => "ABC123", "model" => "VW Golf", "year" => 2001}, 
-          {"_type" => "TrModels::Car", "license_plate" => "DEF123", "model" => "Honda Accord", "year" => 2008},
+        'name' => 'My Fleet', 
+        'transports' => [
+          {'_type' => 'TrModels::Ambulance', 'license_plate' => 'GGG123', 'icu' => true},
+          {'_type' => 'TrModels::Car', 'license_plate' => 'ABC123', 'model' => 'VW Golf', 'year' => 2001}, 
+          {'_type' => 'TrModels::Car', 'license_plate' => 'DEF123', 'model' => 'Honda Accord', 'year' => 2008},
         ] 
       }
       
       fleet = TrModels::Fleet.new(fleet_attributes)
       fleet.transports.size.should == 3
       fleet.transports[0].class.should == TrModels::Ambulance
-      fleet.transports[0].license_plate.should == "GGG123"
+      fleet.transports[0].license_plate.should == 'GGG123'
       fleet.transports[0].icu.should be_true
       fleet.transports[1].class.should == TrModels::Car
-      fleet.transports[1].license_plate.should == "ABC123"
-      fleet.transports[1].model.should == "VW Golf"
+      fleet.transports[1].license_plate.should == 'ABC123'
+      fleet.transports[1].model.should == 'VW Golf'
       fleet.transports[1].year.should == 2001
       fleet.transports[2].class.should == TrModels::Car
-      fleet.transports[2].license_plate.should == "DEF123"
-      fleet.transports[2].model.should == "Honda Accord"
+      fleet.transports[2].license_plate.should == 'DEF123'
+      fleet.transports[2].model.should == 'Honda Accord'
       fleet.transports[2].year.should == 2008      
       fleet.save.should be_true
       
       fleet = fleet.reload
       fleet.transports.size.should == 3
-      fleet.transports[0].license_plate.should == "GGG123"
+      fleet.transports[0].license_plate.should == 'GGG123'
       fleet.transports[0].icu.should be_true
-      fleet.transports[1].license_plate.should == "ABC123"
-      fleet.transports[1].model.should == "VW Golf"
+      fleet.transports[1].license_plate.should == 'ABC123'
+      fleet.transports[1].model.should == 'VW Golf'
       fleet.transports[1].year.should == 2001
-      fleet.transports[2].license_plate.should == "DEF123"
-      fleet.transports[2].model.should == "Honda Accord"
+      fleet.transports[2].license_plate.should == 'DEF123'
+      fleet.transports[2].model.should == 'Honda Accord'
       fleet.transports[2].year.should == 2008      
     end
     
@@ -101,31 +121,31 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
     
     should "be able to replace the association" do
       fleet = TrModels::Fleet.new
-      fleet.transports = [TrModels::Car.new("license_plate" => "DCU2013", "model" => "Honda Civic")]
+      fleet.transports = [TrModels::Car.new('license_plate' => 'DCU2013', 'model' => 'Honda Civic')]
       fleet.save.should be_true
     
       fleet = fleet.reload
       fleet.transports.size.should == 1
-      fleet.transports[0].license_plate.should == "DCU2013"
+      fleet.transports[0].license_plate.should == 'DCU2013'
     end
     
     should "store different associations" do
       fleet = TrModels::Fleet.new
       fleet.transports = [
-        TrModels::Car.new("license_plate" => "ABC1223", "model" => "Honda Civic", "year" => 2003),
-        TrModels::Bus.new("license_plate" => "XYZ9090", "max_passengers" => 51),
-        TrModels::Ambulance.new("license_plate" => "HDD3030", "icu" => true)
+        TrModels::Car.new('license_plate' => 'ABC1223', 'model' => 'Honda Civic', 'year' => 2003),
+        TrModels::Bus.new('license_plate' => 'XYZ9090', 'max_passengers' => 51),
+        TrModels::Ambulance.new('license_plate' => 'HDD3030', 'icu' => true)
       ]
       fleet.save.should be_true
     
       fleet = fleet.reload
       fleet.transports.size.should == 3
-      fleet.transports[0].license_plate.should == "ABC1223"
-      fleet.transports[0].model.should == "Honda Civic"
+      fleet.transports[0].license_plate.should == 'ABC1223'
+      fleet.transports[0].model.should == 'Honda Civic'
       fleet.transports[0].year.should == 2003
-      fleet.transports[1].license_plate.should == "XYZ9090"
+      fleet.transports[1].license_plate.should == 'XYZ9090'
       fleet.transports[1].max_passengers.should == 51
-      fleet.transports[2].license_plate.should == "HDD3030"
+      fleet.transports[2].license_plate.should == 'HDD3030'
       fleet.transports[2].icu.should == true
     end
   end
@@ -134,9 +154,9 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
     should "work using a block passed to many" do
       catalog = Catalog.new
       medias = catalog.medias = [
-        Video.new("file" => "video.mpg", "length" => 3600, :visible => true),
-        Music.new("file" => "music.mp3", "bitrate" => "128kbps", :visible => true),
-        Image.new("file" => "image.png", "width" => 800, "height" => 600, :visible => false)
+        Video.new('file' => 'video.mpg', 'length' => 3600, :visible => true),
+        Music.new('file' => 'music.mp3', 'bitrate' => '128kbps', :visible => true),
+        Image.new('file' => 'image.png', 'width' => 800, 'height' => 600, :visible => false)
       ]
       catalog.save
       catalog.medias.visible.should == [medias[0], medias[1]]
@@ -145,9 +165,9 @@ class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
     should "work using many's :extend option" do
       fleet = TrModels::Fleet.new
       transports = fleet.transports = [
-        TrModels::Car.new("license_plate" => "ABC1223", "model" => "Honda Civic", "year" => 2003, :purchased_on => 2.years.ago.to_date),
-        TrModels::Bus.new("license_plate" => "XYZ9090", "max_passengers" => 51, :purchased_on => 3.years.ago.to_date),
-        TrModels::Ambulance.new("license_plate" => "HDD3030", "icu" => true, :purchased_on => 1.year.ago.to_date)
+        TrModels::Car.new('license_plate' => 'ABC1223', 'model' => 'Honda Civic', 'year' => 2003, :purchased_on => 2.years.ago.to_date),
+        TrModels::Bus.new('license_plate' => 'XYZ9090', 'max_passengers' => 51, :purchased_on => 3.years.ago.to_date),
+        TrModels::Ambulance.new('license_plate' => 'HDD3030', 'icu' => true, :purchased_on => 1.year.ago.to_date)
       ]
       fleet.save
       fleet.transports.to_be_replaced.should == [transports[1]]

data/test/functional/associations/test_many_embedded_proxy.rb

@@ -3,126 +3,153 @@ require 'models'
 
 class ManyEmbeddedProxyTest < Test::Unit::TestCase
   def setup
-    Project.collection.remove
-    RealPerson.collection.remove
+    @comment_class = EDoc do
+      key :name, String
+      key :body, String
+    end
+    @post_class = Doc do
+      key :title, String
+    end
+    @post_class.many :comments, :class => @comment_class
+    
+    @pet_class = EDoc do
+      key :name, String
+    end
+    @pet_class.embedded_in :person
+    @person_class = EDoc do
+      key :name, String
+    end
+    @person_class.key :child, @person_class
+    @person_class.many :pets, :class => @pet_class
+    
+    @owner_class = Doc do
+      key :name, String
+    end
+    @owner_class.many :pets, :class => @pet_class
   end
     
   should "default reader to empty array" do
-    Project.new.addresses.should == []
+    @post_class.new.comments.should == []
   end
   
   should "allow adding to association like it was an array" do
-    project = Project.new
-    project.addresses << Address.new
-    project.addresses.push Address.new
-    project.addresses.size.should == 2
+    post = @post_class.new
+    post.comments << @comment_class.new
+    post.comments.push @comment_class.new
+    post.comments.size.should == 2
   end
 
   should "be embedded in document on save" do
-    sb = Address.new(:city => 'South Bend', :state => 'IN')
-    chi = Address.new(:city => 'Chicago', :state => 'IL')
-    project = Project.new
-    project.addresses << sb
-    project.addresses << chi
-    project.save
+    frank = @comment_class.new(:name => 'Frank', :body => 'Hi!')
+    bill = @comment_class.new(:name => 'Bill', :body => 'Hi!')
+    post = @post_class.new
+    post.comments << frank
+    post.comments << bill
+    post.save
 
-    project.reload
-    project.addresses.size.should == 2
-    project.addresses[0].should == sb
-    project.addresses[1].should == chi
+    post.reload
+    post.comments.size.should == 2
+    post.comments[0].should == frank
+    post.comments[0].new?.should == false
+    post.comments[1].should == bill
+    post.comments[1].new?.should == false
   end
   
   should "allow embedding arbitrarily deep" do
-    @document = Class.new do
-      include MongoMapper::Document
-      set_collection_name 'test'
-      key :person, Person
-    end
-    @document.collection.remove
+    @klass = Doc()
+    @klass.key :person, @person_class
     
-    meg = Person.new(:name => "Meg")
-    meg.child = Person.new(:name => "Steve")
-    meg.child.child = Person.new(:name => "Linda")
+    meg             = @person_class.new(:name => 'Meg')
+    meg.child       = @person_class.new(:name => 'Steve')
+    meg.child.child = @person_class.new(:name => 'Linda')
     
-    doc = @document.new(:person => meg)
+    doc = @klass.new(:person => meg)
     doc.save
-    
     doc.reload
+    
     doc.person.name.should == 'Meg'
     doc.person.child.name.should == 'Steve'
     doc.person.child.child.name.should == 'Linda'
   end
   
-  should "allow assignment of 'many' embedded documents using a hash" do
+  should "allow assignment of many embedded documents using a hash" do
     person_attributes = { 
-      "name" => "Mr. Pet Lover", 
-      "pets" => [
-        {"name" => "Jimmy", "species" => "Cocker Spainel"},
-        {"name" => "Sasha", "species" => "Siberian Husky"}, 
+      'name' => 'Mr. Pet Lover', 
+      'pets' => [
+        {'name' => 'Jimmy', 'species' => 'Cocker Spainel'},
+        {'name' => 'Sasha', 'species' => 'Siberian Husky'}, 
       ] 
     }
     
-    pet_lover = RealPerson.new(person_attributes)
-    pet_lover.name.should == "Mr. Pet Lover"
-    pet_lover.pets[0].name.should == "Jimmy"
-    pet_lover.pets[0].species.should == "Cocker Spainel"
-    pet_lover.pets[1].name.should == "Sasha"
-    pet_lover.pets[1].species.should == "Siberian Husky"
-    pet_lover.save.should be_true
-    
-    pet_lover.reload
-    pet_lover.name.should == "Mr. Pet Lover"
-    pet_lover.pets[0].name.should == "Jimmy"
-    pet_lover.pets[0].species.should == "Cocker Spainel"
-    pet_lover.pets[1].name.should == "Sasha"
-    pet_lover.pets[1].species.should == "Siberian Husky"
+    owner = @owner_class.new(person_attributes)
+    owner.name.should == 'Mr. Pet Lover'
+    owner.pets[0].name.should == 'Jimmy'
+    owner.pets[0].species.should == 'Cocker Spainel'
+    owner.pets[1].name.should == 'Sasha'
+    owner.pets[1].species.should == 'Siberian Husky'
+
+    owner.save.should be_true
+    owner.reload
+
+    owner.name.should == 'Mr. Pet Lover'
+    owner.pets[0].name.should == 'Jimmy'
+    owner.pets[0].species.should == 'Cocker Spainel'
+    owner.pets[1].name.should == 'Sasha'
+    owner.pets[1].species.should == 'Siberian Husky'
   end
 
   context "embedding many embedded documents" do
     setup do
-      @document = Class.new do
-        include MongoMapper::Document
-        set_collection_name 'test'
-        many :people
-      end
-      @document.collection.remove
+      @klass = Doc()
+      @klass.many :people, :class => @person_class
     end
 
     should "persist all embedded documents" do
-      meg    = Person.new(:name => "Meg")
-      sparky = Pet.new(:name => "Sparky", :species => "Dog")
-      koda   = Pet.new(:name => "Koda", :species => "Dog")
-
-      doc = @document.new
-      meg.pets << sparky
-      meg.pets << koda
+      meg = @person_class.new(:name => 'Meg', :pets => [
+        @pet_class.new(:name => 'Sparky', :species => 'Dog'),
+        @pet_class.new(:name => 'Koda', :species => 'Dog')
+      ])
+      
+      doc = @klass.new
       doc.people << meg
       doc.save
-
       doc.reload
-      doc.people.first.name.should == "Meg"
+      
+      doc.people.first.name.should == 'Meg'
       doc.people.first.pets.should_not == []
-      doc.people.first.pets.first.name.should == "Sparky"
-      doc.people.first.pets.first.species.should == "Dog"
-      doc.people.first.pets[1].name.should == "Koda"
-      doc.people.first.pets[1].species.should == "Dog"
+      doc.people.first.pets.first.name.should == 'Sparky'
+      doc.people.first.pets.first.species.should == 'Dog'
+      doc.people.first.pets[1].name.should == 'Koda'
+      doc.people.first.pets[1].species.should == 'Dog'
     end
 
     should "create a reference to the root document for all embedded documents before save" do
-      meg    = Person.new(:name => "Meg")
-      sparky = Pet.new(:name => "Sparky", :species => "Dog")
-      doc    = @document.new
+      doc = @klass.new
+      meg = @person_class.new(:name => 'Meg')
+      pet = @pet_class.new(:name => 'Sparky', :species => 'Dog')
+      
       doc.people << meg
-      meg.pets << sparky
+      meg.pets << pet
 
       doc.people.first._root_document.should == doc
       doc.people.first.pets.first._root_document.should == doc
     end
+    should "create a reference to the owning document for all embedded documents before save" do
+      doc = @klass.new
+      meg = @person_class.new(:name => 'Meg')
+      pet = @pet_class.new(:name => 'Sparky', :species => 'Dog')
+      
+      doc.people << meg
+      meg.pets << pet
+
+      doc.people.first._parent_document.should == doc
+      doc.people.first.pets.first._parent_document.should == doc.people.first
+    end
 
     should "create a reference to the root document for all embedded documents" do
-      sparky = Pet.new(:name => "Sparky", :species => "Dog")
-      meg    = Person.new(:name => "Meg", :pets => [sparky])
-      doc    = @document.new
+      sparky = @pet_class.new(:name => 'Sparky', :species => 'Dog')
+      meg    = @person_class.new(:name => 'Meg', :pets => [sparky])
+      doc    = @klass.new
       doc.people << meg
       doc.save
 
@@ -130,32 +157,69 @@ class ManyEmbeddedProxyTest < Test::Unit::TestCase
       doc.people.first._root_document.should == doc
       doc.people.first.pets.first._root_document.should == doc
     end
+    should "create a reference to the owning document for all embedded documents" do
+      doc = @klass.new
+      meg = @person_class.new(:name => 'Meg')
+      pet = @pet_class.new(:name => 'Sparky', :species => 'Dog')
+      
+      doc.people << meg
+      meg.pets << pet
+      doc.save
+
+      doc.reload
+      doc.people.first._parent_document.should == doc
+      doc.people.first.pets.first._parent_document.should == doc.people.first
+    end
+
+    should "create embedded_in relationship for embedded docs" do
+      doc = @klass.new
+      meg = @person_class.new(:name => 'Meg')
+      pet = @pet_class.new(:name => 'Sparky', :species => 'Dog')
+      
+      doc.people << meg
+      meg.pets << pet
+      doc.save
+
+      doc.reload
+      doc.people.first.pets.first.person.should == doc.people.first
+    end
   end
   
   should "allow finding by id" do
-    sparky = Pet.new(:name => "Sparky", :species => "Dog")
-    meg    = Person.new(:name => "Meg", :pets => [sparky])
+    sparky = @pet_class.new(:name => 'Sparky', :species => 'Dog')
+    meg    = @owner_class.create(:name => 'Meg', :pets => [sparky])
     
     meg.pets.find(sparky._id).should     == sparky  # oid
     meg.pets.find(sparky.id.to_s).should == sparky  # string
   end
   
+  context "count" do
+    should "default to 0" do
+      @owner_class.new.pets.count.should == 0
+    end
+    
+    should "return correct count if any are embedded" do
+      owner = @owner_class.new(:name => 'Meg')
+      owner.pets = [@pet_class.new, @pet_class.new]
+      owner.pets.count.should == 2
+      owner.save
+      owner.reload
+      owner.pets.count.should == 2
+    end
+  end
+  
   context "extending the association" do
     setup do
-      @address_class = Class.new do
-        include MongoMapper::EmbeddedDocument
+      @address_class = EDoc do
         key :address, String
         key :city, String
         key :state, String
         key :zip, Integer
       end
       
-      @project_class = Class.new do
-        include MongoMapper::Document
+      @project_class = Doc do
         key :name, String
       end
-      
-      @project_class.collection.remove
     end
     
     should "work using a block passed to many" do