Merge pull request #271 from jordansexton/patch-2

Added update and delete convenience methods

2 files changed, 24 insertions, 0 deletions

diff --git a/lib/arel/table.rb b/lib/arel/table.rb
index 3866637bd2..545e73e3ae 100644
--- a/lib/arel/table.rb
+++ b/lib/arel/table.rb
@@ -108,6 +108,14 @@ primary_key (#{caller.first}) is deprecated and will be removed in Arel 4.0.0
       InsertManager.new(@engine)
     end
 
+    def update_manager
+      UpdateManager.new(@engine)
+    end
+
+    def delete_manager
+      DeleteManager.new(@engine)
+    end
+
     def hash
       # Perf note: aliases, table alias and engine is excluded from the hash
       #  aliases can have a loop back to this table breaking hashes in parent
diff --git a/test/test_table.rb b/test/test_table.rb
index 286ad10167..14256475ec 100644
--- a/test/test_table.rb
+++ b/test/test_table.rb
@@ -67,6 +67,22 @@ module Arel
       end
     end
 
+    describe 'update_manager' do
+      it 'should return an update manager' do
+        um = @relation.update_manager
+        assert_kind_of Arel::UpdateManager, um
+        assert_equal um.engine, @relation.engine
+      end
+    end
+
+    describe 'delete_manager' do
+      it 'should return a delete manager' do
+        dm = @relation.delete_manager
+        assert_kind_of Arel::DeleteManager, dm
+        assert_equal dm.engine, @relation.engine
+      end
+    end
+
     describe 'having' do
       it 'adds a having clause' do
         mgr = @relation.having @relation[:id].eq(10)