1 files changed, 16 insertions, 0 deletions
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