From ad16c18cc3a7db4b56611b9244140f0b495a1214 Mon Sep 17 00:00:00 2001
From: Aaron Patterson <aaron.patterson@gmail.com>
Date: Mon, 29 Nov 2010 14:30:06 -0800
Subject: insert statements supported

---
 lib/arel/nodes/insert_statement.rb |  2 +-
 lib/arel/nodes/select_statement.rb |  2 +-
 lib/arel/visitors/depth_first.rb   |  7 +++++++
 test/visitors/test_depth_first.rb  | 10 ++++++++++
 4 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/lib/arel/nodes/insert_statement.rb b/lib/arel/nodes/insert_statement.rb
index a5e1a0bb9b..37c12f011a 100644
--- a/lib/arel/nodes/insert_statement.rb
+++ b/lib/arel/nodes/insert_statement.rb
@@ -1,6 +1,6 @@
 module Arel
   module Nodes
-    class InsertStatement
+    class InsertStatement < Arel::Nodes::Node
       attr_accessor :relation, :columns, :values
 
       def initialize
diff --git a/lib/arel/nodes/select_statement.rb b/lib/arel/nodes/select_statement.rb
index 371fd7bf21..6881a66747 100644
--- a/lib/arel/nodes/select_statement.rb
+++ b/lib/arel/nodes/select_statement.rb
@@ -1,6 +1,6 @@
 module Arel
   module Nodes
-    class SelectStatement
+    class SelectStatement < Arel::Nodes::Node
       attr_reader :cores
       attr_accessor :limit, :orders, :lock, :offset
 
diff --git a/lib/arel/visitors/depth_first.rb b/lib/arel/visitors/depth_first.rb
index 534f9edaf5..d6d3984ce2 100644
--- a/lib/arel/visitors/depth_first.rb
+++ b/lib/arel/visitors/depth_first.rb
@@ -61,6 +61,13 @@ module Arel
       alias :visit_Time                  :terminal
       alias :visit_TrueClass             :terminal
 
+      def visit_Arel_Nodes_InsertStatement o
+        visit o.relation
+        visit o.columns
+        visit o.values
+        @block.call o
+      end
+
       def visit_Arel_Nodes_SelectCore o
         visit o.projections
         visit o.froms
diff --git a/test/visitors/test_depth_first.rb b/test/visitors/test_depth_first.rb
index 04c5096f5b..32778f7d4b 100644
--- a/test/visitors/test_depth_first.rb
+++ b/test/visitors/test_depth_first.rb
@@ -120,6 +120,16 @@ module Arel
           :e,
           ss], @collector.calls
       end
+
+      def test_insert_statement
+        stmt = Nodes::InsertStatement.new
+        stmt.relation = :a
+        stmt.columns << :b
+        stmt.values = :c
+
+        @visitor.accept stmt
+        assert_equal [:a, :b, stmt.columns, :c, stmt], @collector.calls
+      end
     end
   end
 end
-- 
cgit v1.2.3