tpitale-mongo_mapper 0.6.9

data/specs.watchr

@@ -0,0 +1,32 @@
+def run(cmd)
+  puts(cmd)
+  system(cmd)
+end
+
+def run_test_file(file)
+  run %Q(ruby -I"lib:test" -rubygems #{file})
+end
+
+def run_all_tests
+  run "rake test"
+end
+
+def related_test_files(path)
+  Dir['test/**/*.rb'].select { |file| file =~ /test_#{File.basename(path)}/ }
+end
+
+watch('test/test_helper\.rb') { run_all_tests }
+watch('test/.*/test_.*\.rb')  { |m| run_test_file(m[0]) }
+watch('lib/.*') do |m|
+  related_test_files(m[0]).each { |file| run_test_file(file) }
+end
+
+# Ctrl-\
+Signal.trap('QUIT') do
+  puts " --- Running all tests ---\n\n"
+  run_all_tests
+end
+
+# Ctrl-C
+Signal.trap('INT') { abort("\n") }
+

data/test/NOTE_ON_TESTING

@@ -0,0 +1 @@
+I am doing my best to keep unit and functional tests separate. As I see them, functional tests hit the database and should never care about internals. Unit tests do not hit the database.

data/test/functional/associations/test_belongs_to_polymorphic_proxy.rb

@@ -0,0 +1,55 @@
+require 'test_helper'
+require 'models'
+
+class BelongsToPolymorphicProxyTest < Test::Unit::TestCase
+  def setup
+    Status.collection.remove
+    Project.collection.remove
+  end
+  
+  should "default to nil" do
+    status = Status.new
+    status.target.nil?.should be_true
+    status.target.inspect.should == "nil"
+  end
+
+  should "be able to replace the association" do
+    status = Status.new(:name => 'Foo!')
+    project = Project.new(:name => "mongomapper")
+    status.target = project
+    status.save.should be_true
+
+    status = status.reload
+    status.target.nil?.should be_false
+    status.target_id.should == project._id
+    status.target_type.should == "Project"
+    status.target.name.should == "mongomapper"
+  end
+
+  should "unset the association" do
+    status = Status.new(:name => 'Foo!')
+    project = Project.new(:name => "mongomapper")
+    status.target = project
+    status.save.should be_true
+
+    status = status.reload
+    status.target = nil
+    status.target_type.nil?.should be_true
+    status.target_id.nil?.should be_true
+    status.target.nil?.should be_true
+  end
+  
+  context "association id set but document not found" do
+    setup do
+      @status = Status.new(:name => 'Foo!')
+      project = Project.new(:name => "mongomapper")
+      @status.target = project
+      @status.save.should be_true
+      project.destroy
+    end
+
+    should "return nil instead of raising error" do
+      @status.target.nil?.should be_true
+    end
+  end
+end

data/test/functional/associations/test_belongs_to_proxy.rb

@@ -0,0 +1,48 @@
+require 'test_helper'
+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
+      key :post_id, String
+    end
+    @comment_class.belongs_to :post, :class => @post_class
+    
+    @post_class.collection.remove
+    @comment_class.collection.remove
+  end
+  
+  should "default to nil" do
+    @comment_class.new.post.nil?.should be_true
+  end
+  
+  should "be able to replace the association" do
+    post = @post_class.new(:name => 'mongomapper')
+    comment = @comment_class.new(:name => 'Foo!', :post => post)
+    comment.save.should be_true
+    
+    comment = comment.reload
+    comment.post.should == post
+    comment.post.nil?.should be_false
+  end
+  
+  should "unset the association" do
+    post = @post_class.new(:name => 'mongomapper')
+    comment = @comment_class.new(:name => 'Foo!', :post => post)
+    comment.save.should be_true
+    
+    comment = comment.reload
+    comment.post = nil
+    comment.post.nil?.should be_true
+  end
+  
+  should "return nil if id set but document not found" do
+    id = Mongo::ObjectID.new
+    @comment_class.new(:name => 'Foo', :post_id => id).post.nil?.should be_true
+  end
+end

data/test/functional/associations/test_many_documents_as_proxy.rb

@@ -0,0 +1,246 @@
+require 'test_helper'
+require 'models'
+
+class ManyDocumentsAsProxyTest < Test::Unit::TestCase
+  def setup
+    Post.collection.remove
+    PostComment.collection.remove
+  end
+
+  should "default reader to empty array" do
+    Post.new.comments.should == []
+  end
+
+  should "add type and id key to polymorphic class base" do
+    PostComment.keys.keys.should include('commentable_type')
+    PostComment.keys.keys.should include('commentable_id')
+  end
+
+  should "allow adding to association like it was an array" do
+    post = Post.new
+    post.comments << PostComment.new(:body => 'foo bar')
+    post.comments << PostComment.new(:body => 'baz')
+    post.comments.concat PostComment.new(:body => 'baz')
+
+    post.comments.size.should == 3
+  end
+
+  should "be able to replace the association" do
+    post = Post.new
+
+    lambda {
+      post.comments = [
+        PostComment.new(:body => 'foo'),
+        PostComment.new(:body => 'bar'),
+        PostComment.new(:body => 'baz')
+      ]
+    }.should change { PostComment.count }.by(3)
+    
+    post = post.reload
+    post.comments.size.should == 3
+    bodies = post.comments.collect(&:body)
+    bodies.should include('foo')
+    bodies.should include('bar')
+    bodies.should include('baz')
+  end
+
+  context "build" do
+    should "assign foreign key" do
+      post = Post.new
+      comment = post.comments.build
+      comment.commentable_id.should == post._id
+    end
+
+    should "assign _type" do
+      post = Post.new
+      comment = post.comments.build
+      comment.commentable_type.should == "Post"
+    end
+
+    should "allow assigning attributes" do
+      post = Post.new
+      comment = post.comments.build(:body => 'foo bar')
+      comment.body.should == 'foo bar'
+    end
+  end
+
+  context "create" do
+    should "assign foreign key" do
+      post = Post.new
+      comment = post.comments.create
+      comment.commentable_id.should == post._id
+    end
+
+    should "assign _type" do
+      post = Post.new
+      comment = post.comments.create
+      comment.commentable_type.should == "Post"
+    end
+
+    should "save record" do
+      post = Post.new
+      lambda {
+        post.comments.create(:body => 'baz')
+      }.should change { PostComment.count }
+    end
+
+    should "allow passing attributes" do
+      post = Post.create
+      comment = post.comments.create(:body => 'foo bar')
+      comment.body.should == 'foo bar'
+    end
+  end
+
+  context "count" do
+    should "work scoped to association" do
+      post = Post.create
+      3.times { post.comments.create(:body => 'foo bar') }
+
+      other_post = Post.create
+      2.times { other_post.comments.create(:body => 'baz') }
+
+      post.comments.count.should == 3
+      other_post.comments.count.should == 2
+    end
+
+    should "work with conditions" do
+      post = Post.create
+      post.comments.create(:body => 'foo bar')
+      post.comments.create(:body => 'baz')
+      post.comments.create(:body => 'foo bar')
+
+      post.comments.count(:body => 'foo bar').should == 2
+    end
+  end
+
+  context "Finding scoped to association" do
+    setup do
+      @post = Post.new
+
+      @comment1 = PostComment.create(:body => 'comment1', :name => 'John')
+      @comment2 = PostComment.create(:body => 'comment2', :name => 'Steve')
+      @comment3 = PostComment.create(:body => 'comment3', :name => 'John')
+      @post.comments = [@comment1, @comment2]
+      @post.save
+
+      @post2 = Post.create(:body => "post #2")
+      @comment4 = PostComment.create(:body => 'comment1', :name => 'Chas')
+      @comment5 = PostComment.create(:body => 'comment2', :name => 'Dan')
+      @comment6 = PostComment.create(:body => 'comment3', :name => 'Ed')
+      @post2.comments = [@comment4, @comment5, @comment6]
+      @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)
+        @post.comments.all.should include(@comment2)
+      end
+
+      should "work with conditions" do
+        comments = @post.comments.all(:body => 'comment1')
+        comments.should == [@comment1]
+      end
+
+      should "work with order" do
+        comments = @post.comments.all(:order => 'body desc')
+        comments.should == [@comment2, @comment1]
+      end
+    end
+
+    context "with one id" do
+      should "work for id in association" do
+        @post.comments.find(@comment2._id).should == @comment2
+      end
+
+      should "not work for id not in association" do
+        lambda {
+          @post.comments.find!(@comment5._id)
+        }.should raise_error(MongoMapper::DocumentNotFound)
+      end
+    end
+
+    context "with multiple ids" do
+      should "work for ids in association" do
+        posts = @post.comments.find!(@comment1._id, @comment2._id)
+        posts.should == [@comment1, @comment2]
+      end
+
+      should "not work for ids not in association" do
+        lambda {
+          @post.comments.find!(@comment1._id, @comment2._id, @comment4._id)
+        }.should raise_error(MongoMapper::DocumentNotFound)
+      end
+    end
+    
+    context "dynamic finders" do
+      should "work with single key" do
+        @post.comments.find_by_body('comment1').should == @comment1
+        @post2.comments.find_by_body('comment1').should == @comment4
+      end
+      
+      should "work with multiple keys" do
+        @post.comments.find_by_body_and_name('comment1', 'John').should == @comment1
+        @post.comments.find_by_body_and_name('comment1', 'Frank').should be_nil
+      end
+      
+      should "raise error when using !" do
+        lambda {
+          @post.comments.find_by_body!('asdf')
+        }.should raise_error(MongoMapper::DocumentNotFound)
+      end
+      
+      context "find_or_create_by" do
+        should "not create document if found" do
+          lambda {
+            comment = @post.comments.find_or_create_by_name('Steve')
+            comment.commentable.should == @post
+            comment.should == @comment2
+          }.should_not change { PostComment.count }
+        end
+
+        should "create document if not found" do
+          lambda {
+            @post.comments.find_or_create_by_name('Chas')
+          }.should change { PostComment.count }.by(1)
+        end
+      end
+    end
+
+    context "with #paginate" do
+      setup do
+        @comments = @post2.comments.paginate(:per_page => 2, :page => 1, :order => 'name')
+      end
+
+      should "return total pages" do
+        @comments.total_pages.should == 2
+      end
+
+      should "return total entries" do
+        @comments.total_entries.should == 3
+      end
+
+      should "return the subject" do
+        @comments.should include(@comment4)
+        @comments.should include(@comment5)
+      end
+    end
+  end
+end