danielharan-mongo_mapper 0.6.5

data/test/unit/test_association_base.rb

@@ -0,0 +1,166 @@
+require 'test_helper'
+require 'models'
+
+class FooMonster; end
+
+class AssociationBaseTest < Test::Unit::TestCase
+  include MongoMapper::Associations
+  
+  should "initialize with type and name" do
+    base = Base.new(:many, :foos)
+    base.type.should == :many
+    base.name.should == :foos
+  end
+  
+  should "also allow options when initializing" do
+    base = Base.new(:many, :foos, :polymorphic => true)
+    base.options[:polymorphic].should be_true
+  end
+  
+  context "class_name" do
+    should "work for belongs_to" do
+      Base.new(:belongs_to, :user).class_name.should == 'User'
+    end
+    
+    should "work for many" do
+      Base.new(:many, :smart_people).class_name.should == 'SmartPerson'
+    end
+    
+    should "be changeable using class_name option" do
+      base = Base.new(:many, :smart_people, :class_name => 'IntelligentPerson')
+      base.class_name.should == 'IntelligentPerson'
+    end
+  end
+  
+  context "klass" do
+    should "default to class_name constantized" do
+      Base.new(:belongs_to, :foo_monster).klass.should == FooMonster
+    end
+    
+    should "be the specified class" do
+      anonnymous_class = Class.new
+      Base.new(:belongs_to, :foo_monster, :class => anonnymous_class).klass.should == anonnymous_class
+    end
+  end
+  
+  context "many?" do
+    should "be true if many" do
+      Base.new(:many, :foos).many?.should be_true
+    end
+    
+    should "be false if not many" do
+      Base.new(:belongs_to, :foo).many?.should be_false
+    end
+  end
+  
+  context "finder_options" do
+    should "default to empty hash" do
+      base = Base.new(:many, :foos)
+      base.finder_options.should == {}
+    end
+    
+    should "work with order" do
+      base = Base.new(:many, :foos, :order => 'position')
+      base.finder_options.should == {:order => 'position'}
+    end
+    
+    should "correctly parse from options" do
+      base = Base.new(:many, :foos, :order => 'position', :somekey => 'somevalue')
+      base.finder_options.should == {:order => 'position', :somekey => 'somevalue'}
+    end
+  end
+  
+  context "belongs_to?" do
+    should "be true if belongs_to" do
+      Base.new(:belongs_to, :foo).belongs_to?.should be_true
+    end
+    
+    should "be false if not belongs_to" do
+      Base.new(:many, :foos).belongs_to?.should be_false
+    end
+  end
+  
+  context "polymorphic?" do
+    should "be true if polymorphic" do
+      Base.new(:many, :foos, :polymorphic => true).polymorphic?.should be_true
+    end
+    
+    should "be false if not polymorphic" do
+      Base.new(:many, :bars).polymorphic?.should be_false
+    end
+  end
+  
+  context "type_key_name" do
+    should "be _type for many" do
+      Base.new(:many, :foos).type_key_name.should == '_type'
+    end
+    
+    should "be association name _ type for belongs_to" do
+      Base.new(:belongs_to, :foo).type_key_name.should == 'foo_type'
+    end
+  end
+  
+  context "foreign_key" do
+    should "default to assocation_name_id" do
+      base = Base.new(:belongs_to, :foo)
+      base.foreign_key.should == 'foo_id'
+    end
+    
+    should "be overridable with :foreign_key option" do
+      base = Base.new(:belongs_to, :foo, :foreign_key => 'foobar_id')
+      base.foreign_key.should == 'foobar_id'
+    end
+  end
+  
+  should "have ivar that is association name" do
+    Base.new(:belongs_to, :foo).ivar.should == '@_foo'
+  end
+  
+  context "embeddable?" do
+    should "be true if class is embeddable" do
+      base = Base.new(:many, :medias)
+      base.embeddable?.should be_true
+    end
+    
+    should "be false if class is not embeddable" do
+      base = Base.new(:many, :statuses)
+      base.embeddable?.should be_false
+      
+      base = Base.new(:belongs_to, :project)
+      base.embeddable?.should be_false
+    end
+  end
+  
+  context "proxy_class" do
+    should "be ManyProxy for many" do      
+      base = Base.new(:many, :statuses)
+      base.proxy_class.should == ManyProxy
+    end
+    
+    should "be ManyPolymorphicProxy for polymorphic many" do
+      base = Base.new(:many, :messages, :polymorphic => true)
+      base.proxy_class.should == ManyPolymorphicProxy
+    end
+    
+    should "be ManyEmbeddedProxy for many embedded" do
+      base = Base.new(:many, :medias)
+      base.proxy_class.should == ManyEmbeddedProxy
+    end
+    
+    should "be ManyEmbeddedPolymorphicProxy for polymorphic many embedded" do
+      base = Base.new(:many, :medias, :polymorphic => true)
+      base.proxy_class.should == ManyEmbeddedPolymorphicProxy
+    end
+    
+    should "be BelongsToProxy for belongs_to" do
+      base = Base.new(:belongs_to, :project)
+      base.proxy_class.should == BelongsToProxy
+    end
+    
+    should "be BelongsToPolymorphicProxy for polymorphic belongs_to" do
+      base = Base.new(:belongs_to, :target, :polymorphic => true)
+      base.proxy_class.should == BelongsToPolymorphicProxy
+    end
+  end
+  
+end

data/test/unit/test_document.rb

@@ -0,0 +1,204 @@
+require 'test_helper'
+require 'models'
+
+class DocumentTest < Test::Unit::TestCase
+  context "The Document Class" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+      end
+      @document.collection.remove
+    end
+    
+    should "have logger method" do
+      @document.logger.should == MongoMapper.logger
+      @document.logger.should be_instance_of(Logger)
+    end
+    
+    should "track its descendants" do
+      MongoMapper::Document.descendants.should include(@document)
+    end
+
+    should "use default database by default" do
+      @document.database.should == MongoMapper.database
+    end
+
+    should "have a connection" do
+      @document.connection.should be_instance_of(Mongo::Connection)
+    end
+
+    should "allow setting different connection without affecting the default" do
+      conn = Mongo::Connection.new
+      @document.connection conn
+      @document.connection.should == conn
+      @document.connection.should_not == MongoMapper.connection
+    end
+
+    should "allow setting a different database without affecting the default" do
+      @document.set_database_name 'test2'
+      @document.database_name.should == 'test2'
+      @document.database.name.should == 'test2'
+
+      another_document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+      end
+      another_document.database.should == MongoMapper.database
+    end
+    
+    should "default collection name to class name tableized" do
+      class ::Item
+        include MongoMapper::Document
+      end
+      
+      Item.collection.should be_instance_of(Mongo::Collection)
+      Item.collection.name.should == 'items'
+    end
+    
+    should "default collection name of namespaced class to tableized with dot separation" do
+      module ::BloggyPoo
+        class Post
+          include MongoMapper::Document
+        end
+      end
+
+      BloggyPoo::Post.collection.should be_instance_of(Mongo::Collection)
+      BloggyPoo::Post.collection.name.should == 'bloggy_poo.posts'
+    end
+
+    should "allow setting the collection name" do
+      @document.set_collection_name('foobar')
+      @document.collection.should be_instance_of(Mongo::Collection)
+      @document.collection.name.should == 'foobar'
+    end
+  end # Document class
+  
+  context "Documents that inherit from other documents" do
+    should "default collection name to inherited class" do
+      Message.collection_name.should == 'messages'
+      Enter.collection_name.should   == 'messages'
+      Exit.collection_name.should    == 'messages'
+      Chat.collection_name.should    == 'messages'
+    end
+    
+    should "default associations to inherited class" do
+     Message.associations.keys.should include("room")
+     Enter.associations.keys.should   include("room")
+     Exit.associations.keys.should    include("room")
+     Chat.associations.keys.should    include("room")
+   end
+    
+    should "track subclasses" do
+      Message.subclasses.should == [Enter, Exit, Chat]
+    end
+  end
+
+  context "An instance of a document" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+
+        key :name, String
+        key :age, Integer
+      end
+      @document.collection.remove
+    end
+    
+    should "have access to logger" do
+      doc = @document.new
+      doc.logger.should == @document.logger
+      doc.logger.should be_instance_of(Logger)
+    end
+
+    should "have access to the class's collection" do
+      doc = @document.new
+      doc.collection.name.should == @document.collection.name
+    end
+
+    should "use default values if defined for keys" do
+      @document.key :active, Boolean, :default => true
+
+      @document.new.active.should be_true
+      @document.new(:active => false).active.should be_false
+    end
+    
+    should "use default values if defined even when custom data type" do
+      @document.key :window, WindowSize, :default => WindowSize.new(600, 480)
+      
+      doc = @document.new
+      doc.window.should == WindowSize.new(600, 480)
+    end
+
+    context "root document" do
+      should "have a nil _root_document" do
+        @document.new._root_document.should be_nil
+      end
+
+      should "set self to the root document on embedded documents" do        
+        klass = Class.new do
+          include MongoMapper::Document
+        end
+        
+        pets = Class.new do
+          include MongoMapper::EmbeddedDocument
+        end
+        klass.many :pets, :class => pets
+        
+        doc = klass.new(:pets => [{}])
+        doc.pets.first._root_document.should == doc        
+      end
+    end
+
+    context "new?" do
+      should "be true if no id" do
+        @document.new.new?.should be_true
+      end
+      
+      should "be true if id but using custom id and not saved yet" do
+        @document.key :_id, String
+        doc = @document.new
+        doc.id = '1234'
+        doc.new?.should be_true
+      end
+    end
+
+    context "clone" do
+      should "not set the id" do
+        doc = @document.create(:name => "foo", :age => 27)
+        clone = doc.clone
+        clone.should be_new
+      end
+
+      should "copy the attributes" do
+        doc = @document.create(:name => "foo", :age => 27)
+        clone = doc.clone
+        clone.name.should == "foo"
+        clone.age.should == 27
+      end
+    end
+
+    context "equality" do
+      setup do
+        @oid = Mongo::ObjectID.new
+      end
+      should "be equal if id and class are the same" do
+        (@document.new('_id' => @oid) == @document.new('_id' => @oid)).should be(true)
+      end
+
+      should "not be equal if class same but id different" do
+        (@document.new('_id' => @oid) == @document.new('_id' => Mongo::ObjectID.new)).should be(false)
+      end
+
+      should "not be equal if id same but class different" do
+        @another_document = Class.new do
+          include MongoMapper::Document
+          set_collection_name 'test'
+        end
+
+        (@document.new('_id' => @oid) == @another_document.new('_id' => @oid)).should be(false)
+      end
+    end
+  end # instance of a document
+end # DocumentTest

data/test/unit/test_dynamic_finder.rb

@@ -0,0 +1,125 @@
+require 'test_helper'
+
+class DynamicFinderTest < Test::Unit::TestCase
+  include MongoMapper
+  
+  should "initialize with method" do
+    finder = DynamicFinder.new(:foobar)
+    finder.method.should == :foobar
+  end
+  
+  context "found?" do
+    should "be true for find_by" do
+      DynamicFinder.new(:find_by_foo).found?.should be_true
+    end
+    
+    should "be true for find_by with !" do
+      DynamicFinder.new(:find_by_foo!).found?.should be_true
+    end
+    
+    should "be true for find_all_by" do
+      DynamicFinder.new(:find_all_by_foo).found?.should be_true
+    end
+    
+    should "be true for find_or_initialize_by" do
+      DynamicFinder.new(:find_or_initialize_by_foo).found?.should be_true
+    end
+    
+    should "be true for find_or_create_by" do
+      DynamicFinder.new(:find_or_create_by_foo).found?.should be_true
+    end
+    
+    should "be false for anything else" do
+      [:foobar, :bazwick].each do |method|
+        DynamicFinder.new(method).found?.should be_false
+      end
+    end
+  end
+  
+  context "find_all_by" do
+    should "parse one attribute" do
+      DynamicFinder.new(:find_all_by_foo).attributes.should == %w(foo)
+    end
+    
+    should "parse multiple attributes" do
+      DynamicFinder.new(:find_all_by_foo_and_bar).attributes.should == %w(foo bar)
+      DynamicFinder.new(:find_all_by_foo_and_bar_and_baz).attributes.should == %w(foo bar baz)
+    end
+    
+    should "set finder to :all" do
+      DynamicFinder.new(:find_all_by_foo_and_bar).finder.should == :all
+    end
+  end
+  
+  context "find_by" do
+    should "parse one attribute" do
+      DynamicFinder.new(:find_by_foo).attributes.should == %w(foo)
+    end
+    
+    should "parse multiple attributes" do
+      DynamicFinder.new(:find_by_foo_and_bar).attributes.should == %w(foo bar)
+    end
+    
+    should "set finder to :first" do
+      DynamicFinder.new(:find_by_foo).finder.should == :first
+    end
+    
+    should "set bang to false" do
+      DynamicFinder.new(:find_by_foo).bang.should be_false
+    end
+  end
+  
+  context "find_by with !" do
+    should "parse one attribute" do
+      DynamicFinder.new(:find_by_foo!).attributes.should == %w(foo)
+    end
+    
+    should "parse multiple attributes" do
+      DynamicFinder.new(:find_by_foo_and_bar!).attributes.should == %w(foo bar)
+    end
+    
+    should "set finder to :first" do
+      DynamicFinder.new(:find_by_foo!).finder.should == :first
+    end
+    
+    should "set bang to true" do
+      DynamicFinder.new(:find_by_foo!).bang.should be_true
+    end
+  end
+  
+  context "find_or_initialize_by" do
+    should "parse one attribute" do
+      DynamicFinder.new(:find_or_initialize_by_foo).attributes.should == %w(foo)
+    end
+    
+    should "parse multiple attributes" do
+      DynamicFinder.new(:find_or_initialize_by_foo_and_bar).attributes.should == %w(foo bar)
+    end
+    
+    should "set finder to :first" do
+      DynamicFinder.new(:find_or_initialize_by_foo).finder.should == :first
+    end
+    
+    should "set instantiator to new" do
+      DynamicFinder.new(:find_or_initialize_by_foo).instantiator.should == :new
+    end
+  end
+
+  context "find_or_create_by" do
+    should "parse one attribute" do
+      DynamicFinder.new(:find_or_create_by_foo).attributes.should == %w(foo)
+    end
+    
+    should "parse multiple attributes" do
+      DynamicFinder.new(:find_or_create_by_foo_and_bar).attributes.should == %w(foo bar)
+    end
+    
+    should "set finder to :first" do
+      DynamicFinder.new(:find_or_create_by_foo).finder.should == :first
+    end
+    
+    should "set instantiator to new" do
+      DynamicFinder.new(:find_or_create_by_foo).instantiator.should == :create
+    end
+  end
+end