jamieorc-mongo_mapper 0.11.1.1

data/test/functional/associations/test_many_embedded_polymorphic_proxy.rb

@@ -0,0 +1,239 @@
+require 'test_helper'
+require 'models'
+
+class ManyEmbeddedPolymorphicProxyTest < Test::Unit::TestCase
+  def setup
+    Catalog.collection.remove
+    TrModels::Fleet.collection.remove
+  end
+
+  should "default reader to empty array" do
+    catalog = Catalog.new
+    catalog.medias.should == []
+  end
+
+  should "allow adding to association like it was an array" do
+    catalog = Catalog.new
+    catalog.medias << Video.new
+    catalog.medias.push Video.new
+    catalog.medias.size.should == 2
+  end
+
+  should "be able to replace the association" do
+    catalog = Catalog.new
+    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].new?.should == false
+  end
+
+  context "associating objects of non-SCI class" do
+    should "work on replacement" do
+      catalog = Catalog.new
+      catalog.medias = [Human.new(:name => 'Frank'), Robot.new(:serial_number => '1B')]
+
+      catalog.medias.size.should == 2
+      catalog.medias[0].name.should == 'Frank'
+      catalog.medias[0].class.should == Human
+      catalog.medias[1].serial_number.should == '1B'
+      catalog.medias[1].class.should == Robot
+
+      catalog.save.should be_true
+      catalog.reload
+
+      catalog.medias.size.should == 2
+      catalog.medias[0].name.should == 'Frank'
+      catalog.medias[0].class.should == Human
+      catalog.medias[1].serial_number.should == '1B'
+      catalog.medias[1].class.should == Robot
+    end
+
+    should "work on replacement with hashes" do
+      catalog = Catalog.new
+      catalog.medias = [{:name => 'Frank', '_type' => 'Human'}, {:serial_number => '1B', '_type' => 'Robot'}]
+
+      catalog.medias.size.should == 2
+      catalog.medias[0].name.should == 'Frank'
+      catalog.medias[0].class.should == Human
+      catalog.medias[1].serial_number.should == '1B'
+      catalog.medias[1].class.should == Robot
+
+      catalog.save.should be_true
+      catalog.reload
+
+      catalog.medias.size.should == 2
+      catalog.medias[0].name.should == 'Frank'
+      catalog.medias[0].class.should == Human
+      catalog.medias[1].serial_number.should == '1B'
+      catalog.medias[1].class.should == Robot
+    end
+
+    should "work with concatination" do
+      catalog = Catalog.new
+      catalog.medias << Human.new(:name => 'Frank')
+      catalog.medias << Robot.new(:serial_number => '1B')
+
+      catalog.medias.size.should == 2
+      catalog.medias[0].name.should == 'Frank'
+      catalog.medias[0].class.should == Human
+      catalog.medias[1].serial_number.should == '1B'
+      catalog.medias[1].class.should == Robot
+
+      catalog.save.should be_true
+      catalog.reload
+
+      catalog.medias.size.should == 2
+      catalog.medias[0].name.should == 'Frank'
+      catalog.medias[0].class.should == Human
+      catalog.medias[1].serial_number.should == '1B'
+      catalog.medias[1].class.should == Robot
+    end
+  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)
+    ]
+    catalog.save.should be_true
+
+    catalog = catalog.reload
+    catalog.medias.size.should == 3
+    catalog.medias[0].file.should == 'video.mpg'
+    catalog.medias[0].length.should == 3600
+    catalog.medias[1].file.should == 'music.mp3'
+    catalog.medias[1].bitrate.should == "128kbps"
+    catalog.medias[2].file.should == 'image.png'
+    catalog.medias[2].width.should == 800
+    catalog.medias[2].height.should == 600
+  end
+
+  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},
+        ]
+      }
+
+      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].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].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].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].icu.should be_true
+      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].year.should == 2008
+    end
+
+    should "default reader to empty array" do
+      fleet = TrModels::Fleet.new
+      fleet.transports.should == []
+    end
+
+    should "allow adding to association like it was an array" do
+      fleet = TrModels::Fleet.new
+      fleet.transports << TrModels::Car.new
+      fleet.transports.push TrModels::Bus.new
+      fleet.transports.size.should == 2
+    end
+
+    should "be able to replace the association" do
+      fleet = TrModels::Fleet.new
+      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'
+    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)
+      ]
+      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].year.should == 2003
+      fleet.transports[1].license_plate.should == 'XYZ9090'
+      fleet.transports[1].max_passengers.should == 51
+      fleet.transports[2].license_plate.should == 'HDD3030'
+      fleet.transports[2].icu.should == true
+    end
+  end
+
+  context "extending the association" do
+    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)
+      ]
+      catalog.save
+      catalog.medias.visible.should == [medias[0], medias[1]]
+    end
+
+    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)
+      ]
+      fleet.save
+      fleet.transports.to_be_replaced.should == [transports[1]]
+    end
+  end
+end

data/test/functional/associations/test_many_embedded_proxy.rb

@@ -0,0 +1,289 @@
+require 'test_helper'
+require 'models'
+
+class ManyEmbeddedProxyTest < Test::Unit::TestCase
+  def setup
+    @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
+    @post_class.new.comments.should == []
+  end
+
+  should "allow adding to association like it was an array" do
+    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
+    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
+
+    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
+    @klass = Doc()
+    @klass.key :person, @person_class
+
+    meg             = @person_class.new(:name => 'Meg')
+    meg.child       = @person_class.new(:name => 'Steve')
+    meg.child.child = @person_class.new(:name => 'Linda')
+
+    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
+    person_attributes = {
+      'name' => 'Mr. Pet Lover',
+      'pets' => [
+        {'name' => 'Jimmy', 'species' => 'Cocker Spainel'},
+        {'name' => 'Sasha', 'species' => 'Siberian Husky'},
+      ]
+    }
+
+    owner = @owner_class.new(person_attributes)
+    owner.name.should == 'Mr. Pet Lover'
+    owner.pets[0].id.class.should == BSON::ObjectId
+    owner.pets[0].name.should == 'Jimmy'
+    owner.pets[0].species.should == 'Cocker Spainel'
+    owner.pets[1].id.class.should == BSON::ObjectId
+    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].id.class.should == BSON::ObjectId
+    owner.pets[0].name.should == 'Jimmy'
+    owner.pets[0].species.should == 'Cocker Spainel'
+    owner.pets[1].id.class.should == BSON::ObjectId
+    owner.pets[1].name.should == 'Sasha'
+    owner.pets[1].species.should == 'Siberian Husky'
+  end
+
+  context "passing documents between versions of code" do
+    setup do
+      @old_klass = Doc do
+        set_collection_name 'generic_parents'
+        key :name, String
+      end
+
+      @updated_klass = Doc do
+        set_collection_name 'generic_parents'
+        key :name, String
+      end
+      @updated_klass.many :pets, :class => @pet_class
+    end
+
+    should "not break many embedded proxy" do
+      @old_klass.collection.drop
+      created_by_new_code = @updated_klass.create!
+      created_by_new_code.pets.should == []
+
+      @old_klass.first # ensure_key_exists calls @old_klass.key(:embedded_docs) (not @old_klass.many(:embedded_docs))
+      @old_klass.create!(:name => 'created in old code') # creates doc with {embedded_docs : null}
+
+      lambda {
+        loaded_in_new_code = @updated_klass.find_by_name('created in old code')
+        loaded_in_new_code.pets.should == []
+      }.should_not raise_error
+    end
+  end
+
+  context "embedding many embedded documents" do
+    setup do
+      @klass = Doc()
+      @klass.many :people, :class => @person_class
+    end
+
+    should "persist all embedded documents" do
+      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.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'
+    end
+
+    should "create a reference to the root 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._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_class.new(:name => 'Sparky', :species => 'Dog')
+      meg    = @person_class.new(:name => 'Meg', :pets => [sparky])
+      doc    = @klass.new
+      doc.people << meg
+      doc.save
+
+      doc.reload
+      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_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 = EDoc do
+        key :address, String
+        key :city, String
+        key :state, String
+        key :zip, Integer
+      end
+
+      @project_class = Doc do
+        key :name, String
+      end
+    end
+
+    should "work using a block passed to many" do
+      @project_class.many :addresses, :class => @address_class do
+        def find_all_by_state(state)
+          find_all { |a| a.state == state }
+        end
+      end
+
+      addr1 = @address_class.new(:address => "Gate-3 Lankershim Blvd.", :city => "Universal City", :state => "CA", :zip => "91608")
+      addr2 = @address_class.new(:address => "3000 W. Alameda Ave.", :city => "Burbank", :state => "CA", :zip => "91523")
+      addr3 = @address_class.new(:address => "111 Some Ln", :city => "Nashville", :state => "TN", :zip => "37211")
+      project = @project_class.create(:name => "Some Project", :addresses => [addr1, addr2, addr3])
+
+      project.addresses.find_all_by_state("CA").should == [addr1, addr2]
+    end
+
+    should "work using many's :extend option" do
+      module FindByCity
+        def find_by_city(city)
+          find_all { |a| a.city == city }
+        end
+      end
+      @project_class.many :addresses, :class => @address_class, :extend => FindByCity
+
+      addr1 = @address_class.new(:address => "Gate-3 Lankershim Blvd.", :city => "Universal City", :state => "CA", :zip => "91608")
+      addr2 = @address_class.new(:address => "3000 W. Alameda Ave.", :city => "Burbank", :state => "CA", :zip => "91523")
+      addr3 = @address_class.new(:address => "111 Some Ln", :city => "Nashville", :state => "TN", :zip => "37211")
+      project = @project_class.create(:name => "Some Project", :addresses => [addr1, addr2, addr3])
+
+      project.addresses.find_by_city('Burbank').should == [addr2]
+    end
+  end
+end