JezK
Edit File: test_sci.rb
require 'test_helper' class SciTest < Test::Unit::TestCase context "Single collection inheritance (document)" do setup do class ::DocParent include MongoMapper::Document key :name, String end DocParent.collection.remove class ::DocDaughter < ::DocParent; end class ::DocSon < ::DocParent; end class ::DocGrandSon < ::DocSon; end DocSon.many :children, :class_name => 'DocGrandSon' @parent = DocParent.new({:name => "Daddy Warbucks"}) @daughter = DocDaughter.new({:name => "Little Orphan Annie"}) end teardown do Object.send :remove_const, 'DocParent' if defined?(::DocParent) Object.send :remove_const, 'DocDaughter' if defined?(::DocDaughter) Object.send :remove_const, 'DocSon' if defined?(::DocSon) Object.send :remove_const, 'DocGrandSon' if defined?(::DocGrandSon) end should "automatically add _type key to store class" do DocParent.key?(:_type).should be_true end should "use the same collection in the subclass" do DocDaughter.collection.name.should == DocParent.collection.name end context ".single_collection_inherited?" do should "be false if has not inherited" do DocParent.should_not be_single_collection_inherited end should "be true if inherited" do DocDaughter.should be_single_collection_inherited DocSon.should be_single_collection_inherited DocGrandSon.should be_single_collection_inherited end end should "set _type on initialize" do DocDaughter.new._type.should == 'DocDaughter' DocSon.new._type.should == 'DocSon' DocGrandSon.new._type.should == 'DocGrandSon' end should "set _type based on class and ignore assigned values" do DocSon.new(:_type => 'DocDaughter')._type.should == 'DocSon' end context "loading" do should "be based on _type" do @parent.save @daughter.save collection = DocParent.all collection.size.should == 2 collection.first.should be_kind_of(DocParent) collection.first.name.should == "Daddy Warbucks" collection.last.should be_kind_of(DocDaughter) collection.last.name.should == "Little Orphan Annie" end should "gracefully handle when _type cannot be constantized" do doc = DocParent.new(:name => 'Nunes') doc._type = 'FoobarBaz' doc.save collection = DocParent.all collection.last.should == doc collection.last.should be_kind_of(DocParent) end end context "querying" do should "find scoped to class" do john = DocSon.create(:name => 'John') steve = DocSon.create(:name => 'Steve') steph = DocDaughter.create(:name => 'Steph') carrie = DocDaughter.create(:name => 'Carrie') DocGrandSon.all(:order => 'name').should == [] DocSon.all(:order => 'name').should == [john, steve] DocDaughter.all(:order => 'name').should == [carrie, steph] DocParent.all(:order => 'name').should == [carrie, john, steph, steve] end should "work with nested hash conditions" do john = DocSon.create(:name => 'John') steve = DocSon.create(:name => 'Steve') DocSon.all(:name => {'$ne' => 'Steve'}).should == [john] end should "raise error if not found scoped to class" do john = DocSon.create(:name => 'John') steph = DocDaughter.create(:name => 'Steph') lambda { DocSon.find!(steph._id) }.should raise_error(MongoMapper::DocumentNotFound) end should "not raise error for find with parent" do john = DocSon.create(:name => 'John') DocParent.find!(john._id).should == john end should "count scoped to class" do john = DocSon.create(:name => 'John') steve = DocSon.create(:name => 'Steve') steph = DocDaughter.create(:name => 'Steph') carrie = DocDaughter.create(:name => 'Carrie') DocGrandSon.count.should == 0 DocSon.count.should == 2 DocDaughter.count.should == 2 DocParent.count.should == 4 end should "not be able to destroy each other" do john = DocSon.create(:name => 'John') steph = DocDaughter.create(:name => 'Steph') lambda { DocSon.destroy(steph._id) }.should raise_error(MongoMapper::DocumentNotFound) end should "not be able to delete each other" do john = DocSon.create(:name => 'John') steph = DocDaughter.create(:name => 'Steph') lambda { DocSon.delete(steph._id) }.should_not change { DocParent.count } end should "be able to destroy using parent" do john = DocSon.create(:name => 'John') steph = DocDaughter.create(:name => 'Steph') lambda { DocParent.destroy_all }.should change { DocParent.count }.by(-2) end should "be able to delete using parent" do john = DocSon.create(:name => 'John') steph = DocDaughter.create(:name => 'Steph') lambda { DocParent.delete_all }.should change { DocParent.count }.by(-2) end end should "be able to reload single collection inherited parent class" do brian = DocParent.create(:name => 'Brian') brian.name = 'B-Dawg' brian.reload brian.name.should == 'Brian' end end context "Single collection inheritance (embedded document)" do setup do class ::Grandparent include MongoMapper::EmbeddedDocument key :grandparent, String end class ::Parent < ::Grandparent include MongoMapper::EmbeddedDocument key :parent, String end class ::Child < ::Parent include MongoMapper::EmbeddedDocument key :child, String end class ::OtherChild < ::Parent include MongoMapper::EmbeddedDocument key :other_child, String end end teardown do Object.send :remove_const, 'Grandparent' if defined?(::Grandparent) Object.send :remove_const, 'Parent' if defined?(::Parent) Object.send :remove_const, 'Child' if defined?(::Child) Object.send :remove_const, 'OtherChild' if defined?(::OtherChild) end should "automatically add _type key" do Grandparent.key?(:_type).should be_true end context ".single_collection_inherited?" do should "be false if has not inherited" do Grandparent.should_not be_single_collection_inherited end should "be true if inherited" do Parent.should be_single_collection_inherited Child.should be_single_collection_inherited OtherChild.should be_single_collection_inherited end end should "set _type on initialize" do Parent.new._type.should == 'Parent' Child.new._type.should == 'Child' OtherChild.new._type.should == 'OtherChild' end should "set _type based on class and ignore assigned values" do Child.new(:_type => 'OtherChild')._type.should == 'Child' end end end