Statement nodes deep-copy AST nodes

1 files changed, 22 insertions, 0 deletions

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