1 files changed, 24 insertions, 0 deletions
diff --git a/activerecord/test/cases/arel/update_manager_test.rb b/activerecord/test/cases/arel/update_manager_test.rb
index cc1b9ac5b3..e13cb6aa52 100644
--- a/activerecord/test/cases/arel/update_manager_test.rb
+++ b/activerecord/test/cases/arel/update_manager_test.rb
@@ -122,5 +122,29 @@ module Arel
         @um.key.must_equal @table[:foo]
       end
     end
+
+    describe "comment" do
+      it "chains" do
+        manager = Arel::UpdateManager.new
+        manager.comment("updating").must_equal manager
+      end
+
+      it "appends a comment to the generated query" do
+        table   = Table.new :users
+
+        manager = Arel::UpdateManager.new
+        manager.table table
+
+        manager.comment("updating")
+        manager.to_sql.must_be_like %{
+          UPDATE "users" /* updating */
+        }
+
+        manager.comment("updating", "with", "comment")
+        manager.to_sql.must_be_like %{
+          UPDATE "users" /* updating */ /* with */ /* comment */
+        }
+      end
+    end
   end
 end