Statement nodes deep-copy AST nodes

3 files changed, 60 insertions, 0 deletions

diff --git a/spec/arel/nodes/insert_statement_spec.rb b/spec/arel/nodes/insert_statement_spec.rb
new file mode 100644
index 0000000000..d8b02b6703
--- /dev/null
+++ b/spec/arel/nodes/insert_statement_spec.rb
@@ -0,0 +1,22 @@
+require 'spec_helper'
+
+describe Arel::Nodes::InsertStatement do
+  describe "clone" do
+    it "clones cores" do
+      statement = Arel::Nodes::InsertStatement.new
+      statement.columns = %w[a b c]
+      statement.values  = %w[x y z]
+
+      statement.columns.each_with_index do |o, j|
+        o.should_receive(:clone).and_return("#{o}#{j}")
+      end
+      statement.values.each_with_index do |o, j|
+        o.should_receive(:clone).and_return("#{o}#{j}")
+      end
+
+      dolly = statement.clone
+      dolly.columns.should == %w[a0 b1 c2]
+      dolly.values.should  == %w[x0 y1 z2]
+    end
+  end
+end
diff --git a/spec/arel/nodes/select_statement_spec.rb b/spec/arel/nodes/select_statement_spec.rb
new file mode 100644
index 0000000000..04b850823c
--- /dev/null
+++ b/spec/arel/nodes/select_statement_spec.rb
@@ -0,0 +1,16 @@
+require 'spec_helper'
+
+describe Arel::Nodes::SelectStatement do
+  describe "clone" do
+    it "clones cores" do
+      statement = Arel::Nodes::SelectStatement.new %w[a b c]
+
+      statement.cores.each_with_index do |o, j|
+        o.should_receive(:clone).and_return("#{o}#{j}")
+      end
+
+      dolly = statement.clone
+      dolly.cores.should == %w[a0 b1 c2]
+    end
+  end
+end
diff --git a/spec/arel/nodes/update_statement_spec.rb b/spec/arel/nodes/update_statement_spec.rb
new file mode 100644
index 0000000000..438eaf1dc2
--- /dev/null
+++ b/spec/arel/nodes/update_statement_spec.rb
@@ -0,0 +1,22 @@
+require 'spec_helper'
+
+describe Arel::Nodes::UpdateStatement do
+  describe "clone" do
+    it "clones cores" do
+      statement = Arel::Nodes::UpdateStatement.new
+      statement.wheres = %w[a b c]
+      statement.values = %w[x y z]
+
+      statement.wheres.each_with_index do |o, j|
+        o.should_receive(:clone).and_return("#{o}#{j}")
+      end
+      statement.values.each_with_index do |o, j|
+        o.should_receive(:clone).and_return("#{o}#{j}")
+      end
+
+      dolly = statement.clone
+      dolly.wheres.should == %w[a0 b1 c2]
+      dolly.values.should == %w[x0 y1 z2]
+    end
+  end
+end