add UpdateManager#key method to access the key

2 files changed, 20 insertions, 0 deletions

diff --git a/lib/arel/update_manager.rb b/lib/arel/update_manager.rb
index f13aeb0a8c..56e219040c 100644
--- a/lib/arel/update_manager.rb
+++ b/lib/arel/update_manager.rb
@@ -15,6 +15,10 @@ module Arel
       @ast.key = key
     end
 
+    def key
+      @ast.key
+    end
+
     def order *expr
       @ast.orders = expr
       self
diff --git a/test/test_update_manager.rb b/test/test_update_manager.rb
index 3099fb367e..f9704af425 100644
--- a/test/test_update_manager.rb
+++ b/test/test_update_manager.rb
@@ -95,5 +95,21 @@ module Arel
         um.where(table[:id].eq(1)).must_equal um
       end
     end
+
+    describe 'key' do
+      before do
+        @table = Table.new :users
+        @um = Arel::UpdateManager.new Table.engine
+        @um.key = @table[:foo]
+      end
+
+      it 'can be set' do
+        @um.ast.key.must_equal @table[:foo]
+      end
+
+      it 'can be accessed' do
+        @um.key.must_equal @table[:foo]
+      end
+    end
   end
 end