Always return self when restore

lib/paranoia.rb

@@ -34,11 +34,7 @@ module Paranoia
     alias :deleted :only_deleted
     alias :deleted :only_deleted
 
 
     def restore(id, opts = {})
     def restore(id, opts = {})
-      if id.is_a?(Array)
-        id.map { |one_id| restore(one_id, opts) }
-      else
-        only_deleted.find(id).restore!(opts)
-      end
+      Array(id).flatten.map { |one_id| only_deleted.find(one_id).restore!(opts) }
     end
     end
   end
   end
 
 
@@ -97,6 +93,8 @@ module Paranoia
         restore_associated_records if opts[:recursive]
         restore_associated_records if opts[:recursive]
       end
       end
     end
     end
+
+    self
   end
   end
   alias :restore :restore!
   alias :restore :restore!
 
 

test/paranoia_test.rb

@@ -290,6 +290,13 @@ class ParanoiaTest < test_framework
     assert_equal false, model.destroyed?
     assert_equal false, model.destroyed?
   end
   end
 
 
+  def test_restore_on_object_return_self
+    model = ParanoidModel.create
+    model.destroy
+
+    assert_equal model.class, model.restore.class
+  end
+
   # Regression test for #92
   # Regression test for #92
   def test_destroy_twice
   def test_destroy_twice
     model = ParanoidModel.new
     model = ParanoidModel.new