danielharan-mongo_mapper 0.6.5

data/test/functional/test_embedded_document.rb

@@ -0,0 +1,125 @@
+require 'test_helper'
+require 'models'
+
+class EmbeddedDocumentTest < Test::Unit::TestCase
+  def setup
+    @document = Class.new do
+      include MongoMapper::Document
+      set_collection_name 'users'
+
+      key :first_name, String
+      key :last_name, String
+    end
+    @document.collection.remove
+  end
+    
+  context "Saving a document with an embedded document" do
+    setup do
+      @document.class_eval do
+        key :foo, Address
+      end
+      
+      @address = Address.new(:city => 'South Bend', :state => 'IN')
+      @doc = @document.new(:foo => @address)
+    end
+    
+    should "embed embedded document" do
+      @doc.save
+      @doc.foo.city.should == 'South Bend'
+      @doc.foo.state.should == 'IN'
+
+      doc = @doc.reload
+      doc.foo.city.should == 'South Bend'
+      doc.foo.state.should == 'IN'
+    end
+  end
+  
+  context "Instantiating single collection inherited embedded documents" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        key :message, Message
+      end
+    end
+
+    should "work" do
+      doc1 = @document.create(:message => Enter.new)
+      doc2 = @document.create(:message => Exit.new)
+      doc3 = @document.create(:message => Chat.new)
+
+      doc1.reload.message.class.should be(Enter)
+      doc2.reload.message.class.should be(Exit)
+      doc3.reload.message.class.should be(Chat)
+    end
+  end
+  
+  context "new?" do
+    setup do
+      @document.class_eval do
+        key :foo, Address
+      end
+    end
+
+    should "be new until document is saved" do
+      address = Address.new(:city => 'South Bend', :state => 'IN')
+      doc = @document.new(:foo => address)
+      address.new?.should == true
+    end
+    
+    should "not be new after document is saved" do
+      address = Address.new(:city => 'South Bend', :state => 'IN')
+      doc = @document.new(:foo => address)
+      doc.save
+      doc.foo.new?.should == false
+    end
+    
+    should "not be new when document is read back" do
+      address = Address.new(:city => 'South Bend', :state => 'IN')
+      doc = @document.new(:foo => address)
+      doc.save
+      
+      doc = doc.reload
+      doc.foo.new?.should == false
+    end
+  end
+  
+  context "save" do
+    should "save the root document" do
+      person = RealPerson.create
+
+      pet = Pet.new :name => 'sparky'
+      person.pets << pet
+      pet.save
+
+      person = person.reload
+      person.pets.first.should == pet
+    end
+    
+    should "save new keys" do
+      person = RealPerson.new
+      person[:new_attribute] = 'foobar'
+      person.save
+      
+      person = person.reload
+      person.new_attribute.should == 'foobar'
+    end
+  end
+  
+  context "update_attributes" do
+    should "save the root document" do
+      person = RealPerson.create
+
+      pet = Pet.new(:name => 'sparky')
+      person.pets << pet
+      pet.save
+
+      person = person.reload
+      pet = person.pets.first
+      pet.update_attributes :name => 'koda'
+
+      person = person.reload
+      person.pets.first._id.should == pet._id
+      person.pets.first.name.should == 'koda'
+    end
+  end  
+end

data/test/functional/test_logger.rb

@@ -0,0 +1,20 @@
+require 'test_helper'
+
+class LoggerTest < Test::Unit::TestCase
+  context "with connection that has logger" do
+    setup do
+      @output = StringIO.new
+      @logger = Logger.new(@output)
+      MongoMapper.connection = Mongo::Connection.new('127.0.0.1', 27017, :logger => @logger)
+    end
+
+    should "be able to get access to that logger" do
+      MongoMapper.logger.should == @logger
+    end
+    
+    should "be able to log messages" do
+      MongoMapper.logger.debug 'testing'
+      @output.string.include?('testing').should be_true
+    end
+  end
+end

data/test/functional/test_pagination.rb

@@ -0,0 +1,95 @@
+require 'test_helper'
+
+class PaginationTest < Test::Unit::TestCase
+  context "Paginating" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'users'
+
+        key :first_name, String
+        key :last_name, String
+        key :age, Integer
+        
+        def self.per_page; 1 end
+      end
+      @document.collection.remove
+      
+      @doc1 = @document.create({:first_name => 'John', :last_name => 'Nunemaker', :age => '27'})
+      @doc2 = @document.create({:first_name => 'Steve', :last_name => 'Smith', :age => '28'})
+      @doc3 = @document.create({:first_name => 'Steph', :last_name => 'Nunemaker', :age => '26'})
+    end
+
+    should "return the total pages" do
+      result = @document.paginate(:per_page => 2, :page => 1)
+      result.total_pages.should == 2
+    end
+    
+    should "return the total pages when defaulting to the document class per_page" do
+      result = @document.paginate(:page => 1)
+      result.total_pages.should == 3
+    end
+    
+    should "return the total of records" do
+      result = @document.paginate(:per_page => 2, :page => 1)
+      result.total_entries.should == 3
+    end
+
+    should "return the items" do
+      result = @document.paginate(:per_page => 2, :page => 1, :order => 'first_name')
+      result.size.should == 2
+      result.should == [@doc1, @doc3]
+    end
+
+    should "accept conditions" do
+      result = @document.paginate({
+        :last_name => 'Nunemaker',
+        :order     => "age DESC",
+        :per_page  => 2, 
+        :page      => 1,
+      })
+      result.should == [@doc1, @doc3]
+      result.first.age.should == 27
+
+      result = @document.paginate({
+        :conditions => {:last_name => 'Nunemaker'},
+        :order      => "age DESC",
+        :per_page   => 2,
+        :page       => 1} )
+      result.should == [@doc1, @doc3]
+      result.first.age.should == 27
+    end
+
+    should "withstand rigor" do      
+      result = @document.paginate({
+        :per_page  => 1, 
+        :page      => 1,
+        :order     => 'age desc', 
+        :last_name => 'Nunemaker'
+      })
+      result.should == [@doc1]
+      result.total_entries.should == 2
+      result.total_pages.should == 2
+      
+      result = @document.paginate({
+        :per_page  => 1,
+        :page      => 2,
+        :order     => 'age desc',
+        :last_name => 'Nunemaker'
+      })
+      result.should == [@doc3]
+      result.total_entries.should == 2
+      result.total_pages.should == 2
+      
+      result = @document.paginate({
+        :per_page  => 2,
+        :page      => 1,
+        :order     => 'age desc',
+        :last_name => 'Nunemaker'
+      })
+      result.should == [@doc1, @doc3]
+      result.total_entries.should == 2
+      result.total_pages.should == 1
+    end
+  end
+end

data/test/functional/test_rails_compatibility.rb

@@ -0,0 +1,25 @@
+require 'test_helper'
+
+class TestRailsCompatibility < Test::Unit::TestCase
+  class Item
+    include MongoMapper::EmbeddedDocument
+    key :for_all, String
+  end
+  
+  class Order
+    include MongoMapper::Document
+    many :items, :class_name => 'TestRailsCompatibility::Item'
+    key :order_only, String
+  end
+  
+  context "Document" do
+    setup do
+      Order.collection.remove
+    end
+
+    should "alias new to new_record?" do
+      instance = Order.new
+      instance.new_record?.should == instance.new?
+    end
+  end
+end

data/test/functional/test_string_id_compatibility.rb

@@ -0,0 +1,72 @@
+require 'test_helper'
+
+class StringIdCompatibilityTest < Test::Unit::TestCase
+  def setup
+    @note_class = Class.new do
+      include MongoMapper::EmbeddedDocument
+      key :_id, String
+    end
+    
+    @task_class = Class.new do
+      include MongoMapper::Document
+      key :_id, String
+      key :project_id, String
+      belongs_to :project
+    end
+    
+    @project_class = Class.new do
+      include MongoMapper::Document
+      key :_id, String
+    end
+    
+    @task_class.belongs_to :project, :class => @project_class
+    @project_class.many :notes, :class => @note_class
+    @project_class.many :tasks, :class => @task_class, :foreign_key => 'project_id'
+    
+    @project_class.collection.remove
+    @task_class.collection.remove
+  end
+  
+  should "assign correct _id for documents" do
+    project = @project_class.create
+    project._id.should == project.id
+    project._id.should be_instance_of(String)
+    project.id.size.should == 24
+    lambda {
+      Mongo::ObjectID.from_string(project.id)
+    }.should_not raise_error
+  end
+  
+  should "assign correct _id for embedded documents" do
+    note = @note_class.new
+    note.id.should == note._id
+    note.id.size.should == 24
+  end
+  
+  should "find records" do
+    project = @project_class.create
+    @project_class.find(project.id).should == project
+  end
+    
+  should "save embedded docs" do
+    n1 = @note_class.new
+    n2 = @note_class.new
+    n3 = @note_class.new
+    project = @project_class.create(:notes => [n1, n2, n3])
+    
+    project = project.reload
+    project.notes.size.should == 3
+    project.notes.should == [n1, n2, n3]
+  end
+  
+  should "be able to associate records" do
+    t1 = @task_class.new(:body => 'First task')
+    t2 = @task_class.new(:body => 'Second task')
+    t3 = @task_class.new(:body => 'Third task')
+    project = @project_class.create(:name => 'MM', :tasks => [t1, t2, t3])
+    
+    project = project.reload
+    project.tasks.count.should == 3
+    project.tasks.should == [t1, t2, t3]
+  end
+end

data/test/functional/test_validations.rb

@@ -0,0 +1,369 @@
+require 'test_helper'
+
+class ValidationsTest < Test::Unit::TestCase  
+  context "Saving a new document that is invalid" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+        key :name, String, :required => true
+      end
+      @document.collection.remove
+    end
+    
+    should "not insert document" do
+      doc = @document.new
+      doc.save
+      @document.count.should == 0
+    end
+    
+    should "populate document's errors" do
+      doc = @document.new
+      doc.errors.size.should == 0
+      doc.save
+      doc.errors.full_messages.should == ["Name can't be empty"]
+    end
+  end
+
+  context "Skipping validations when saving" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+        key :name, String, :required => true
+      end
+      @document.collection.remove
+    end
+
+    should "insert document" do
+      doc = @document.new
+      doc.save(false)
+      @document.count.should == 1
+    end
+  end
+
+  context "Saving a document that is invalid (destructive)" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+        key :name, String, :required => true
+      end
+      @document.collection.remove
+    end
+    
+    should "raise error" do
+      doc = @document.new
+      lambda { doc.save! }.should raise_error(MongoMapper::DocumentNotValid)
+    end
+  end
+
+  context "Creating a document that is invalid (destructive)" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+        key :name, String, :required => true
+      end
+      @document.collection.remove
+    end
+    
+    should "raise error" do
+      lambda { @document.create! }.should raise_error(MongoMapper::DocumentNotValid)
+    end
+
+    should "create a new document" do
+      instance = @document.create!(:name => "James")
+      instance.new_record?.should be_false
+    end
+  end
+  
+  context "Saving an existing document that is invalid" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+        key :name, String, :required => true
+      end
+      @document.collection.remove
+      
+      @doc = @document.create(:name => 'John Nunemaker')
+    end
+    
+    should "not update document" do
+      @doc.name = nil
+      @doc.save
+      @doc.reload.name.should == 'John Nunemaker'
+    end
+    
+    should "populate document's errors" do
+      @doc.name = nil
+      @doc.save
+      @doc.errors.full_messages.should == ["Name can't be empty"]
+    end
+  end
+  
+  context "Adding validation errors" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+        
+        key :action, String
+        def action_present
+          errors.add(:action, 'is invalid') if action.blank?
+        end
+      end
+      @document.collection.remove
+    end
+    
+    should "work with validate_on_create callback" do
+      @document.validate_on_create :action_present
+      
+      doc = @document.new
+      doc.action = nil
+      doc.should have_error_on(:action)
+      
+      doc.action = 'kick'
+      doc.should_not have_error_on(:action)
+      doc.save
+      
+      doc.action = nil
+      doc.should_not have_error_on(:action)
+    end
+    
+    should "work with validate_on_update callback" do
+      @document.validate_on_update :action_present
+      
+      doc = @document.new
+      doc.action = nil
+      doc.should_not have_error_on(:action)
+      doc.save
+      
+      doc.action = nil
+      doc.should have_error_on(:action)
+      
+      doc.action = 'kick'
+      doc.should_not have_error_on(:action)
+    end
+  end
+  
+  context "validating uniqueness of" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+        
+        key :name, String
+        validates_uniqueness_of :name
+      end
+      @document.collection.remove
+    end
+
+    should "not fail if object is new" do
+      doc = @document.new
+      doc.should_not have_error_on(:name)
+    end
+
+    should "not fail when new object is out of scope" do
+      document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+        
+        key :name
+        key :adult
+        validates_uniqueness_of :name, :scope => :adult
+      end
+      doc = document.new("name" => "joe", :adult => true)
+      doc.save.should be_true
+
+      doc2 = document.new("name" => "joe", :adult => false)
+      doc2.should be_valid
+
+    end
+
+    should "allow to update an object" do
+      doc = @document.new("name" => "joe")
+      doc.save.should be_true
+
+      @document \
+        .stubs(:first) \
+        .with(:name => 'joe') \
+        .returns(doc)
+
+      doc.name = "joe"
+      doc.valid?.should be_true
+      doc.should_not have_error_on(:name)
+    end
+
+    should "fail if object name is not unique" do
+      doc = @document.new("name" => "joe")
+      doc.save.should be_true
+
+      @document \
+        .stubs(:first) \
+        .with(:name => 'joe') \
+        .returns(doc)
+
+      doc2 = @document.new("name" => "joe")
+      doc2.should have_error_on(:name)
+    end
+    
+    should "allow multiple blank entries if :allow_blank => true" do
+      document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+        
+        key :name
+        validates_uniqueness_of :name, :allow_blank => :true
+      end
+      
+      doc = document.new("name" => "")
+      doc.save.should be_true
+
+      document \
+        .stubs(:first) \
+        .with(:name => '') \
+        .returns(doc)
+
+      doc2 = document.new("name" => "")
+      doc2.should_not have_error_on(:name)
+    end
+
+    should "allow entries that differ only in case by default" do
+      document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+
+        key :name
+        validates_uniqueness_of :name
+      end
+
+      doc = document.new("name" => "BLAMMO")
+      doc.save.should be_true
+
+      doc2 = document.new("name" => "blammo")
+      doc2.should_not have_error_on(:name)
+    end
+
+    should "fail on entries that differ only in case if :case_sensitive => false" do
+      document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+
+        key :name
+        validates_uniqueness_of :name, :case_sensitive => false
+      end
+
+      doc = document.new("name" => "BLAMMO")
+      doc.save.should be_true
+
+      doc2 = document.new("name" => "blammo")
+      doc2.should have_error_on(:name)
+    end
+
+    context "scoped by a single attribute" do
+      setup do
+        @document = Class.new do
+          include MongoMapper::Document
+          set_collection_name 'test'
+          
+          key :name, String
+          key :scope, String
+          validates_uniqueness_of :name, :scope => :scope
+        end
+        @document.collection.remove
+      end
+
+      should "fail if the same name exists in the scope" do
+        doc = @document.new("name" => "joe", "scope" => "one")
+        doc.save.should be_true
+        
+        @document \
+          .stubs(:first) \
+          .with(:name => 'joe', :scope => "one") \
+          .returns(doc)
+
+        doc2 = @document.new("name" => "joe", "scope" => "one")
+        doc2.should have_error_on(:name)
+      end
+
+      should "pass if the same name exists in a different scope" do
+        doc = @document.new("name" => "joe", "scope" => "one")
+        doc.save.should be_true
+
+        @document \
+          .stubs(:first) \
+          .with(:name => 'joe', :scope => 'two') \
+          .returns(nil)
+
+        doc2 = @document.new("name" => "joe", "scope" => "two")
+        doc2.should_not have_error_on(:name)
+      end
+    end
+
+    context "scoped by a multiple attributes" do
+      setup do
+        @document = Class.new do
+          include MongoMapper::Document
+          set_collection_name 'test'
+          
+          key :name, String
+          key :first_scope, String
+          key :second_scope, String
+          validates_uniqueness_of :name, :scope => [:first_scope, :second_scope]
+        end
+        @document.collection.remove
+      end
+      
+      should "fail if the same name exists in the scope" do
+        doc = @document.new("name" => "joe", "first_scope" => "one", "second_scope" => "two")
+        doc.save.should be_true
+        
+        @document \
+          .stubs(:first) \
+          .with(:name => 'joe', :first_scope => 'one', :second_scope => 'two') \
+          .returns(doc)
+
+        doc2 = @document.new("name" => "joe", "first_scope" => "one", "second_scope" => "two")
+        doc2.should have_error_on(:name)
+      end
+      
+      should "pass if the same name exists in a different scope" do
+        doc = @document.new("name" => "joe", "first_scope" => "one", "second_scope" => "two")
+        doc.save.should be_true
+        
+        @document \
+          .stubs(:first) \
+          .with(:name => 'joe', :first_scope => 'one', :second_scope => 'one') \
+          .returns(nil)
+
+        doc2 = @document.new("name" => "joe", "first_scope" => "one", "second_scope" => "one")
+        doc2.should_not have_error_on(:name)
+      end
+    end
+  end
+  
+  context "validates uniqueness of with :unique shortcut" do
+    should "work" do
+      @document = Class.new do
+        include MongoMapper::Document
+        set_collection_name 'test'
+        
+        key :name, String, :unique => true
+      end
+      @document.collection.remove
+      
+      doc = @document.create(:name => 'John')
+      doc.should_not have_error_on(:name)
+      
+      @document \
+        .stubs(:first) \
+        .with(:name => 'John') \
+        .returns(doc)
+      
+      second_john = @document.create(:name => 'John')
+      second_john.should have_error_on(:name, 'has already been taken')
+    end
+  end
+end