Behavior of AR::Base.new{} fixed (#34)

lib/protector/adapters/active_record/relation.rb

@@ -94,7 +94,13 @@ def new_with_protector(*args, &block)
             Protector::ActiveRecord::StrongParameters.sanitize! args, true, protector_meta
           end
 
-          new_without_protector(*args, &block).restrict!(protector_subject)
+          unless block_given?
+            new_without_protector(*args).restrict!(protector_subject)
+          else
+            new_without_protector(*args) do |instance|
+              block.call instance.restrict!(protector_subject)
+            end
+          end
         end
 
         def create_with_protector(*args, &block)

spec/lib/protector/adapters/active_record_spec.rb

@@ -57,6 +57,7 @@ module ProtectionCase
     describe Protector::Adapters::ActiveRecord::Base do
       let(:dummy) do
         Class.new(ActiveRecord::Base) do
+          def self.name; 'Dummy'; end
           def self.model_name; ActiveModel::Name.new(self, nil, "dummy"); end
           self.table_name = "dummies"
           scope :none, where('1 = 0') unless respond_to?(:none)
@@ -93,6 +94,18 @@ def self.model_name; ActiveModel::Name.new(self, nil, "dummy"); end
         expect { dummy.restrict!('!').create!(string: 'test').delete }.to raise_error
       end
 
+      it "validates on new{}" do
+        dummy.instance_eval do
+          protect do; end
+        end
+
+        result = dummy.restrict!('!').new do |instance|
+          instance.protector_subject.should == '!'
+        end
+
+        result.protector_subject.should == '!'
+      end
+
       it "finds with scope on id column" do
         dummy.instance_eval do
           protect do