diff options
-rw-r--r-- | lib/arel/update_manager.rb | 4 | ||||
-rw-r--r-- | test/test_update_manager.rb | 16 |
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 |