jnunemaker-mongomapper 0.2.0 → 0.3.0

data/test/functional/test_pagination.rb

@@ -0,0 +1,82 @@
+require 'test_helper'
+
+class PaginationTest < Test::Unit::TestCase
+  context "Paginating" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        collection 'users'
+
+        key :first_name, String
+        key :last_name, String
+        key :age, Integer
+      end
+
+      @document.collection.clear
+      
+      @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 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)
+      result.size.should == 2
+      result.subject.should == [@doc1, @doc2]
+      result.should == [@doc1, @doc2]
+    end
+
+    should "accept conditions" do
+      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', 
+        :conditions => {: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',
+        :conditions => {: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',
+        :conditions => {: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,31 @@
+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.clear
+    end
+  
+    should "have to_param that returns id" do
+      id = MongoID.new
+      instance = Order.create('_id' => id.to_s)
+      instance.to_param.should == id.to_s
+    end
+
+    should "alias new to new_record?" do
+      instance = Order.new
+      instance.new_record?.should == instance.new?
+    end
+  end
+end

data/test/functional/test_validations.rb

@@ -0,0 +1,172 @@
+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
+        key :name, String, :required => true
+      end
+      @document.collection.clear
+    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 "Saving a document that is invalid (destructive)" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        key :name, String, :required => true
+      end
+      @document.collection.clear
+    end
+    
+    should "raise error" do
+      doc = @document.new
+      lambda { doc.save! }.should raise_error(MongoMapper::DocumentNotValid)
+    end
+  end
+  
+  context "Saving an existing document that is invalid" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+        key :name, String, :required => true
+      end
+      @document.collection.clear
+      
+      @doc = @document.create(:name => 'John Nunemaker')
+    end
+    
+    should "not update document" do
+      @doc.name = nil
+      @doc.save
+      @document.find(@doc.id).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
+        key :action, String
+        def action_present
+          errors.add(:action, 'is invalid') if action.blank?
+        end
+      end
+      @document.collection.clear
+    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
+        key :name, String
+        validates_uniqueness_of :name
+      end
+      @document.collection.clear
+    end
+
+    should "not fail if object is new" do
+      doc = @document.new
+      doc.should_not have_error_on(:name)
+    end
+
+    should "allow to update an object" do
+      doc = @document.new("name" => "joe")
+      doc.save.should be_true
+
+      @document \
+        .stubs(:find) \
+        .with(:first, :conditions => {:name => 'joe'}, :limit => 1) \
+        .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(:find) \
+        .with(:first, :conditions => {:name => 'joe'}, :limit => 1) \
+        .returns(doc)
+
+      doc2 = @document.new("name" => "joe")
+      doc2.should have_error_on(:name)
+    end
+  end
+  
+  context "validates uniqueness of with :unique shortcut" do
+    should "work" do
+      @document = Class.new do
+        include MongoMapper::Document
+        key :name, String, :unique => true
+      end
+      @document.collection.clear
+      
+      doc = @document.create(:name => 'John')
+      doc.should_not have_error_on(:name)
+      
+      @document \
+        .stubs(:find) \
+        .with(:first, :conditions => {:name => 'John'}, :limit => 1) \
+        .returns(doc)
+      
+      second_john = @document.create(:name => 'John')
+      second_john.should have_error_on(:name, 'has already been taken')
+    end
+  end
+end

data/test/models.rb

@@ -0,0 +1,92 @@
+class Address
+  include MongoMapper::EmbeddedDocument
+  key :address, String
+  key :city,    String
+  key :state,   String
+  key :zip,     Integer
+end
+
+class Project
+  include MongoMapper::Document
+  key :name, String
+  many :statuses
+  many :addresses
+end
+
+class Status
+  include MongoMapper::Document
+  belongs_to :project
+  belongs_to :target, :polymorphic => true
+  key :name, String
+end
+
+class RealPerson
+  include MongoMapper::Document
+  many :pets  
+  key :name, String
+end
+
+class Person
+  include MongoMapper::EmbeddedDocument
+  key :name, String
+  key :child, Person
+  many :pets
+end
+
+class Pet
+  include MongoMapper::EmbeddedDocument
+  key :name, String
+  key :species, String
+end
+
+class Media
+  include MongoMapper::EmbeddedDocument
+  key :file, String
+end
+
+class Video < Media
+  key :length, Integer
+end
+
+class Image < Media
+  key :width, Integer
+  key :height, Integer
+end
+
+class Music < Media
+  key :bitrate, String
+end
+
+class Catalog
+  include MongoMapper::Document
+  many :medias, :polymorphic => true
+end
+
+module TrModels
+  class Transport
+    include MongoMapper::EmbeddedDocument
+    key :license_plate, String
+  end
+  
+  class Car < TrModels::Transport
+    include MongoMapper::EmbeddedDocument
+    key :model, String
+    key :year, Integer
+  end
+  
+  class Bus < TrModels::Transport
+    include MongoMapper::EmbeddedDocument
+    key :max_passengers, Integer
+  end
+  
+  class Ambulance < TrModels::Transport
+    include MongoMapper::EmbeddedDocument
+    key :icu, Boolean
+  end
+  
+  class Fleet
+    include MongoMapper::Document
+    many :transports, :polymorphic => true, :class_name => "TrModels::Transport"
+    key :name, String    
+  end
+end

data/test/test_helper.rb

@@ -1,4 +1,5 @@
 require 'pathname'
+require 'pp'
 require 'rubygems'
 require 'test/unit'
 require 'shoulda'
@@ -14,6 +15,10 @@ dir = (Pathname(__FILE__).dirname +  '..' + 'lib').expand_path
 require dir + 'mongomapper'
 
 class Test::Unit::TestCase
+  def clear_all_collections
+    MongoMapper::Document.descendants.map(&:delete_all)
+  end
+  
   custom_matcher :be_nil do |receiver, matcher, args|
     matcher.positive_failure_message = "Expected #{receiver} to be nil but it wasn't"
     matcher.negative_failure_message = "Expected #{receiver} not to be nil but it was"

data/test/unit/test_association_base.rb

@@ -0,0 +1,131 @@
+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 "be class_name constantized" do
+      Base.new(:belongs_to, :foo_monster).klass.should == FooMonster
+    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 "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
+  
+  should "have belongs_to_key_name" do
+    Base.new(:belongs_to, :foo).belongs_to_key_name.should == 'foo_id'
+  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 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