Adding SelectManager#limit= and removing limit nodes when nil is assigned to limit

1 files changed, 9 insertions, 0 deletions

diff --git a/test/test_select_manager.rb b/test/test_select_manager.rb
index 4b4733db97..efea44d56f 100644
--- a/test/test_select_manager.rb
+++ b/test/test_select_manager.rb
@@ -743,6 +743,15 @@ module Arel
         manager = Arel::SelectManager.new Table.engine
         manager.take(1).must_equal manager
       end
+
+      it 'removes LIMIT when nil is passed' do
+        manager = Arel::SelectManager.new Table.engine
+        manager.limit = 10
+        assert_match('LIMIT', manager.to_sql)
+
+        manager.limit = nil
+        refute_match('LIMIT', manager.to_sql)
+      end
     end
 
     describe 'where' do