lib/search_cop/visitors/visitor.rb in search_cop-1.0.8 vs lib/search_cop/visitors/visitor.rb in search_cop-1.0.9

@@ -53,10 +53,14 @@
 
       def visit_SearchCopGrammar_Nodes_Not(node)
         "NOT (#{visit node.object})"
       end
 
+      def visit_SearchCopGrammar_Nodes_Generator(node)
+        instance_exec visit(node.left), node.right[:value], &node.right[:generator]
+      end
+
       def quote_table_name(name)
         connection.quote_table_name name
       end
 
       def quote_column_name(name)
@@ -88,9 +92,10 @@
       alias :visit_Time :quote
       alias :visit_Date :quote
       alias :visit_Float :quote
       alias :visit_Fixnum :quote
       alias :visit_Symbol :quote
+      alias :visit_Integer :quote
     end
   end
 end