djsun-mongomapper 0.3.1

data/test/unit/serializers/test_json_serializer.rb

@@ -0,0 +1,103 @@
+require 'test_helper'
+
+class JsonSerializationTest < Test::Unit::TestCase
+  class Contact
+    include MongoMapper::Document
+    key :name, String
+    key :age, Integer
+    key :created_at, Time
+    key :awesome, Boolean
+    key :preferences, Hash
+  end
+  
+  def setup
+    Contact.include_root_in_json = false
+    @contact = Contact.new(
+      :name        => 'Konata Izumi',
+      :age         => 16,
+      :created_at  => Time.utc(2006, 8, 1),
+      :awesome     => true,
+      :preferences => { :shows => 'anime' }
+    )
+  end
+  
+  should "include demodulized root" do
+    Contact.include_root_in_json = true
+    assert_match %r{^\{"contact": \{}, @contact.to_json
+  end
+  
+  should "encode all encodable attributes" do
+    json = @contact.to_json
+
+    assert_match %r{"name":"Konata Izumi"}, json
+    assert_match %r{"age":16}, json
+    assert json.include?(%("created_at":#{ActiveSupport::JSON.encode(Time.utc(2006, 8, 1))}))
+    assert_match %r{"awesome":true}, json
+    assert_match %r{"preferences":\{"shows":"anime"\}}, json
+  end
+  
+  should "allow attribute filtering with only" do
+    json = @contact.to_json(:only => [:name, :age])
+
+    assert_match %r{"name":"Konata Izumi"}, json
+    assert_match %r{"age":16}, json
+    assert_no_match %r{"awesome"}, json
+    assert_no_match %r{"created_at"}, json
+    assert_no_match %r{"preferences"}, json
+  end
+  
+  should "allow attribute filtering with except" do
+    json = @contact.to_json(:except => [:name, :age])
+
+    assert_no_match %r{"name"}, json
+    assert_no_match %r{"age"}, json
+    assert_match %r{"awesome"}, json
+    assert_match %r{"created_at"}, json
+    assert_match %r{"preferences"}, json
+  end
+  
+  context "including methods" do
+    setup do
+      def @contact.label; "Has cheezburger"; end
+      def @contact.favorite_quote; "Constraints are liberating"; end
+    end
+    
+    should "include single method" do
+      assert_match %r{"label":"Has cheezburger"}, @contact.to_json(:only => :name, :methods => :label)
+    end
+    
+    should "include multiple methods" do
+      json = @contact.to_json(:only => :name, :methods => [:label, :favorite_quote])
+      assert_match %r{"label":"Has cheezburger"}, json
+      assert_match %r{"favorite_quote":"Constraints are liberating"}, json
+    end
+  end
+  
+  context "array of records" do
+    setup do
+      @contacts = [
+        Contact.new(:name => 'David', :age => 39),
+        Contact.new(:name => 'Mary', :age => 14)
+      ]
+    end
+
+    should "allow attribute filtering with only" do
+      assert_equal %([{"name":"David"},{"name":"Mary"}]), @contacts.to_json(:only => :name)
+    end
+    
+    should "allow attribute filtering with except" do
+      json = @contacts.to_json(:except => [:name, :preferences, :awesome, :created_at, :updated_at, :_id])
+      assert_equal %([{"id":"","age":39},{"id":"","age":14}]), json
+    end
+  end
+  
+  should "allow options for hash of records" do
+    contacts = {
+      1 => Contact.new(:name => 'David', :age => 39),
+      2 => Contact.new(:name => 'Mary', :age => 14)
+    }
+
+    assert_equal %({"1":{"name":"David"}}), contacts.to_json(:only => [1, :name])
+  end
+  
+end

data/test/unit/test_association_base.rb

@@ -0,0 +1,136 @@
+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 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,125 @@
+require 'test_helper'
+require 'models'
+
+class DocumentTest < Test::Unit::TestCase
+  context "The Document Class" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::Document
+      end
+    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(XGen::Mongo::Driver::Mongo)
+    end
+
+    should "allow setting different connection without affecting the default" do
+      conn = XGen::Mongo::Driver::Mongo.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.database AlternateDatabase
+      @document.database.name.should == AlternateDatabase
+
+      another_document = Class.new do
+        include MongoMapper::Document
+      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(XGen::Mongo::Driver::Collection)
+      Item.collection.name.should == 'items'
+    end
+
+    should "allow setting the collection name" do
+      @document.collection('foobar')
+      @document.collection.should be_instance_of(XGen::Mongo::Driver::Collection)
+      @document.collection.name.should == 'foobar'
+    end
+  end # Document class
+  
+  context "Documents that inherit from other documents" do
+    should "default collection 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 "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
+
+        key :name, String
+        key :age, Integer
+      end
+      @document.collection.clear
+    end
+
+    should "have access to the class's collection" do
+      doc = @document.new
+      doc.collection.should == @document.collection
+    end
+
+    should "automatically have a created_at key" do
+      @document.keys.keys.should include('created_at')
+    end
+
+    should "automatically have an updated_at key" do
+      @document.keys.keys.should include('updated_at')
+    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
+
+    context "new?" do
+      should "be true if no id" do
+        @document.new.new?.should be(true)
+      end
+    end
+    
+    context "equality" do
+      should "be equal if id and class are the same" do
+        (@document.new('_id' => 1) == @document.new('_id' => 1)).should be(true)
+      end
+
+      should "not be equal if class same but id different" do
+        (@document.new('_id' => 1) == @document.new('_id' => 2)).should be(false)
+      end
+
+      should "not be equal if id same but class different" do
+        @another_document = Class.new do
+          include MongoMapper::Document
+        end
+
+        (@document.new('_id' => 1) == @another_document.new('_id' => 1)).should be(false)
+      end
+    end
+  end # instance of a document
+end # DocumentTest

data/test/unit/test_embedded_document.rb

@@ -0,0 +1,370 @@
+require 'test_helper'
+
+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
+
+class EmbeddedDocumentTest < Test::Unit::TestCase
+  context "Including MongoMapper::EmbeddedDocument" do
+    setup do
+      @klass = Class.new do
+        include MongoMapper::EmbeddedDocument
+      end
+    end
+    
+    should "add _id key" do
+      @klass.keys['_id'].should_not be_nil
+    end
+  end
+  
+  context "parent_model" do
+    should "be nil if none of parents ancestors include EmbeddedDocument" do
+      parent = Class.new
+      document = Class.new(parent) do
+        include MongoMapper::EmbeddedDocument
+      end
+      document.parent_model.should be_nil
+    end
+    
+    should "find parent" do
+      Parent.parent_model.should == Grandparent
+      Child.parent_model.should == Parent
+    end
+  end
+  
+  context "defining a key" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::EmbeddedDocument
+      end
+    end
+    
+    should "work" do
+      key = @document.key(:name, String)
+      key.name.should == 'name'
+      key.type.should == String
+      key.should be_instance_of(MongoMapper::Key)
+    end
+    
+    should "work with options" do
+      key = @document.key(:name, String, :required => true)
+      key.options[:required].should be(true)
+    end
+    
+    should "be tracked per document" do
+      @document.key(:name, String)
+      @document.key(:age, Integer)
+      @document.keys['name'].name.should == 'name'
+      @document.keys['name'].type.should == String
+      @document.keys['age'].name.should == 'age'
+      @document.keys['age'].type.should == Integer
+    end
+    
+    should "be redefinable" do
+      @document.key(:foo, String)
+      @document.keys['foo'].type.should == String
+      @document.key(:foo, Integer)
+      @document.keys['foo'].type.should == Integer
+    end
+    
+    should "create reader method" do
+      @document.new.should_not respond_to(:foo)
+      @document.key(:foo, String)
+      @document.new.should respond_to(:foo)
+    end
+    
+    should "create reader before typecast method" do
+      @document.new.should_not respond_to(:foo_before_typecast)
+      @document.key(:foo, String)
+      @document.new.should respond_to(:foo_before_typecast)
+    end
+    
+    should "create writer method" do
+      @document.new.should_not respond_to(:foo=)
+      @document.key(:foo, String)
+      @document.new.should respond_to(:foo=)
+    end
+    
+    should "create boolean method" do
+      @document.new.should_not respond_to(:foo?)
+      @document.key(:foo, String)
+      @document.new.should respond_to(:foo?)
+    end
+  end
+  
+  context "keys" do
+    should "be inherited" do
+      Grandparent.keys.keys.sort.should == ['_id', 'grandparent']
+      Parent.keys.keys.sort.should == ['_id', 'grandparent', 'parent']
+      Child.keys.keys.sort.should  == ['_id', 'child', 'grandparent', 'parent']
+    end
+    
+    should "propogate to subclasses if key added after class definition" do
+      Grandparent.key :_type, String
+      
+      Grandparent.keys.keys.sort.should == ['_id', '_type', 'grandparent']
+      Parent.keys.keys.sort.should      == ['_id', '_type', 'grandparent', 'parent']
+      Child.keys.keys.sort.should       == ['_id', '_type', 'child', 'grandparent', 'parent']
+    end
+  end
+  
+  context "subclasses" do
+    should "default to nil" do
+      Child.subclasses.should be_nil
+    end
+    
+    should "be recorded" do
+      Grandparent.subclasses.should == [Parent]
+      Parent.subclasses.should      == [Child, OtherChild]
+    end
+  end
+
+  context "An instance of an embedded document" do
+    setup do
+      @document = Class.new do
+        include MongoMapper::EmbeddedDocument
+
+        key :name, String
+        key :age, Integer
+      end
+    end
+    
+    should "automatically have an _id key" do
+      @document.keys.keys.should include('_id')
+    end
+    
+    should "have id method that is string representation of _id" do
+      doc = @document.new
+      doc.id.should == doc._id.to_s
+    end
+    
+    should "be able to set _id using id=" do
+      id = MongoID.new
+      doc = @document.new(:id => id.to_s)
+      doc._id.should == id
+      doc.id.should == id.to_s
+    end
+
+    context "being initialized" do
+      should "accept a hash that sets keys and values" do
+        doc = @document.new(:name => 'John', :age => 23)
+        doc.attributes.keys.sort.should == ['_id', 'age', 'name']
+        doc.attributes['name'].should == 'John'
+        doc.attributes['age'].should == 23
+      end
+
+      should "not throw error if initialized with nil" do
+        lambda {
+          @document.new(nil)
+        }.should_not raise_error
+      end
+    end
+
+    context "mass assigning keys" do
+      should "update values for keys provided" do
+        doc = @document.new(:name => 'foobar', :age => 10)
+        doc.attributes = {:name => 'new value', :age => 5}
+        doc.attributes[:name].should == 'new value'
+        doc.attributes[:age].should == 5
+      end
+
+      should "not update values for keys that were not provided" do
+        doc = @document.new(:name => 'foobar', :age => 10)
+        doc.attributes = {:name => 'new value'}
+        doc.attributes[:name].should == 'new value'
+        doc.attributes[:age].should == 10
+      end
+
+      should "raise undefined method if no key exists" do
+        doc = @document.new(:name => 'foobar', :age => 10)
+        lambda {
+          doc.attributes = {:name => 'new value', :foobar => 'baz'}
+        }.should raise_error(NoMethodError)
+      end
+
+      should "not ignore keys that have methods defined" do
+        @document.class_eval do
+          attr_writer :password
+
+          def passwd
+            @password
+          end
+        end
+
+        doc = @document.new(:name => 'foobar', :password => 'secret')
+        doc.passwd.should == 'secret'
+      end
+
+      should "typecast key values" do
+        doc = @document.new(:name => 1234, :age => '21')
+        doc.name.should == '1234'
+        doc.age.should == 21
+      end
+    end
+
+    context "attributes" do
+      should "default to hash with _id" do
+        doc = @document.new
+        doc.attributes.keys.should == ['_id']
+      end
+
+      should "return all keys that aren't nil" do
+        doc = @document.new(:name => 'string', :age => nil)
+        doc.attributes.keys.sort.should == ['_id', 'name']
+        doc.attributes.values.should include('string')
+      end
+    end
+    
+    context "key shorcut access" do
+      should "be able to read key with []" do
+        doc = @document.new(:name => 'string')
+        doc[:name].should == 'string'
+      end
+      
+      should "be able to write key value with []=" do
+        doc = @document.new
+        doc[:name] = 'string'
+        doc[:name].should == 'string'
+      end
+    end
+    
+    context "indifferent access" do
+      should "be enabled for keys" do
+        doc = @document.new(:name => 'string')
+        doc.attributes[:name].should == 'string'
+        doc.attributes['name'].should == 'string'
+      end
+    end
+
+    context "reading an attribute" do
+      should "work for defined keys" do
+        doc = @document.new(:name => 'string')
+        doc.name.should == 'string'
+      end
+
+      should "raise no method error for undefined keys" do
+        doc = @document.new
+        lambda { doc.fart }.should raise_error(NoMethodError)
+      end
+
+      should "be accessible for use in the model" do
+        @document.class_eval do
+          def name_and_age
+            "#{read_attribute(:name)} (#{read_attribute(:age)})"
+          end
+        end
+
+        doc = @document.new(:name => 'John', :age => 27)
+        doc.name_and_age.should == 'John (27)'
+      end
+    end
+
+    context "reading an attribute before typcasting" do
+      should "work for defined keys" do
+        doc = @document.new(:name => 12)
+        doc.name_before_typecast.should == 12
+      end
+
+      should "raise no method error for undefined keys" do
+        doc = @document.new
+        lambda { doc.foo_before_typecast }.should raise_error(NoMethodError)
+      end
+
+      should "be accessible for use in a document" do
+        @document.class_eval do
+          def untypcasted_name
+            read_attribute_before_typecast(:name)
+          end
+        end
+
+        doc = @document.new(:name => 12)
+        doc.name.should == '12'
+        doc.untypcasted_name.should == 12
+      end
+    end
+
+    context "writing an attribute" do
+      should "work for defined keys" do
+        doc = @document.new
+        doc.name = 'John'
+        doc.name.should == 'John'
+      end
+
+      should "raise no method error for undefined keys" do
+        doc = @document.new
+        lambda { doc.fart = 'poof!' }.should raise_error(NoMethodError)
+      end
+
+      should "typecast value" do
+        doc = @document.new
+        doc.name = 1234
+        doc.name.should == '1234'
+        doc.age = '21'
+        doc.age.should == 21
+      end
+      
+      should "be accessible for use in the model" do
+        @document.class_eval do
+          def name_and_age=(new_value)
+            new_value.match(/([^\(\s]+) \((.*)\)/)
+            write_attribute :name, $1
+            write_attribute :age, $2
+          end
+        end
+
+        doc = @document.new
+        doc.name_and_age = 'Frank (62)'
+        doc.name.should == 'Frank'
+        doc.age.should == 62
+      end
+    end # writing an attribute
+    
+    context "checking if an attributes value is present" do
+      should "work for defined keys" do
+        doc = @document.new
+        doc.name?.should be_false
+        doc.name = 'John'
+        doc.name?.should be_true
+      end
+      
+      should "raise no method error for undefined keys" do
+        doc = @document.new
+        lambda { doc.fart? }.should raise_error(NoMethodError)
+      end
+    end
+    
+    context "equality" do
+      should "be equal if id and class are the same" do
+        (@document.new('_id' => 1) == @document.new('_id' => 1)).should be(true)
+      end
+
+      should "not be equal if class same but id different" do
+        (@document.new('_id' => 1) == @document.new('_id' => 2)).should be(false)
+      end
+
+      should "not be equal if id same but class different" do
+        @another_document = Class.new do
+          include MongoMapper::Document
+        end
+
+        (@document.new('_id' => 1) == @another_document.new('_id' => 1)).should be(false)
+      end
+    end
+  end # instance of a embedded document
+end