mongo_mapper-unstable 2010.1.27 → 2010.1.28

data/test/functional/test_protected.rb

@@ -0,0 +1,139 @@
+require 'test_helper'
+
+class ProtectedTest < Test::Unit::TestCase
+  context 'A document with protected attributes' do
+    setup do
+      @doc_class = Doc do
+        key :name, String
+        key :admin, Boolean, :default => false
+
+        attr_protected :admin
+      end
+
+      @doc = @doc_class.create(:name => 'Steve Sloan')
+    end
+
+    should 'have protected attributes class method' do
+      @doc_class.protected_attributes.should == [:admin].to_set
+    end
+    
+    should "default protected attributes to nil" do
+      Doc().protected_attributes.should be_nil
+    end
+
+    should "have protected attributes instance method" do
+      @doc.protected_attributes.should equal(@doc_class.protected_attributes)
+    end
+
+    should "work with :protected shortcut when defining key" do
+      Doc() do
+        key :user_id, ObjectId, :protected => true
+      end.protected_attributes.should == [:user_id].to_set
+    end
+
+    should 'assign protected attribute through accessor' do
+      @doc.admin = true
+      @doc.admin.should be_true
+    end
+
+    should 'ignore protected attribute on #update_attributes' do
+      @doc.update_attributes(:name => 'Ren Hoek', :admin => true)
+      @doc.name.should == 'Ren Hoek'
+      @doc.admin.should be_false
+    end
+
+    should 'ignore protected attribute on #update_attributes!' do
+      @doc.update_attributes!(:name => 'Stimpson J. Cat', :admin => true)
+      @doc.name.should == 'Stimpson J. Cat'
+      @doc.admin.should be_false
+    end
+  end
+  
+  context "Single collection inherited protected attributes" do
+    setup do
+      class ::GrandParent
+        include MongoMapper::Document
+
+        key :_type, String
+        key :site_id, ObjectId
+
+        attr_protected :site_id
+      end
+      GrandParent.collection.remove
+
+      class ::Child < ::GrandParent
+        key :position, Integer
+
+        attr_protected :position
+      end
+
+      class ::GrandChild < ::Child; end
+
+      class ::OtherChild < ::GrandParent
+        key :blog_id, ObjectId
+
+        attr_protected :blog_id
+      end
+    end
+
+    teardown do
+      Object.send :remove_const, 'GrandParent' if defined?(::GrandParent)
+      Object.send :remove_const, 'Child'       if defined?(::Child)
+      Object.send :remove_const, 'GrandChild'  if defined?(::GrandChild)
+      Object.send :remove_const, 'OtherChild'  if defined?(::OtherChild)
+    end
+
+    should "share keys down the inheritance trail" do
+      GrandParent.protected_attributes.should == [:site_id].to_set
+      Child.protected_attributes.should == [:site_id, :position].to_set
+      GrandChild.protected_attributes.should == [:site_id, :position].to_set
+      OtherChild.protected_attributes.should == [:site_id, :blog_id].to_set
+    end
+  end
+
+  context 'An embedded document with protected attributes' do
+    setup do
+      @doc_class = Doc('Project')
+      @edoc_class = EDoc('Person') do
+        key :name, String
+        key :admin, Boolean, :default => false
+
+        attr_protected :admin
+      end
+      @doc_class.many :people, :class => @edoc_class
+
+      @doc = @doc_class.create(:title => 'MongoMapper')
+      @edoc = @edoc_class.new(:name => 'Steve Sloan')
+      @doc.people << @edoc
+    end
+
+    should 'have protected attributes class method' do
+      @edoc_class.protected_attributes.should == [:admin].to_set
+    end
+    
+    should "default protected attributes to nil" do
+      EDoc().protected_attributes.should be_nil
+    end
+
+    should "have protected attributes instance method" do
+      @edoc.protected_attributes.should equal(@edoc_class.protected_attributes)
+    end
+
+    should 'assign protected attribute through accessor' do
+      @edoc.admin = true
+      @edoc.admin.should be_true
+    end
+
+    should 'ignore protected attribute on #update_attributes' do
+      @edoc.update_attributes(:name => 'Ren Hoek', :admin => true)
+      @edoc.name.should == 'Ren Hoek'
+      @edoc.admin.should be_false
+    end
+
+    should 'ignore protected attribute on #update_attributes!' do
+      @edoc.update_attributes!(:name => 'Stimpson J. Cat', :admin => true)
+      @edoc.name.should == 'Stimpson J. Cat'
+      @edoc.admin.should be_false
+    end
+  end
+end

data/test/test_helper.rb

@@ -16,46 +16,40 @@ require 'support/timing'
 
 class Test::Unit::TestCase
   include CustomMatchers
-  
-  cattr_accessor :mm_document_count
-  self.mm_document_count = 0
-  
+
   def Doc(name=nil, &block)
-    Test::Unit::TestCase.mm_document_count += 1
-    
     klass = Class.new do
       include MongoMapper::Document
       set_collection_name "test#{rand(20)}"
-      
+
       if name
         class_eval "def self.name; '#{name}' end"
         class_eval "def self.to_s; '#{name}' end"
       end
-      
-      class_eval(&block) if block_given?
     end
+
+    klass.class_eval(&block) if block_given?
     klass.collection.remove
     klass
   end
-  
+
   def EDoc(name=nil, &block)
-    Class.new do
+    klass = Class.new do
       include MongoMapper::EmbeddedDocument
-      
+
       if name
         class_eval "def self.name; '#{name}' end"
         class_eval "def self.to_s; '#{name}' end"
       end
-      
-      class_eval(&block) if block_given?
     end
+
+    klass.class_eval(&block) if block_given?
+    klass
   end
 end
 
 test_dir = File.expand_path(File.dirname(__FILE__) + '/../tmp')
 FileUtils.mkdir_p(test_dir) unless File.exist?(test_dir)
 
-MongoMapper.connection = Mongo::Connection.new('127.0.0.1', 27017, {
-  :logger => Logger.new(test_dir + '/test.log')
-})
+MongoMapper.connection = Mongo::Connection.new('127.0.0.1', 27017, {:logger => Logger.new(test_dir + '/test.log')})
 MongoMapper.database = 'test'