update is working on the select manager

1 files changed, 48 insertions, 0 deletions

diff --git a/spec/arel/select_manager_spec.rb b/spec/arel/select_manager_spec.rb
index 928db49c52..e456cabc16 100644
--- a/spec/arel/select_manager_spec.rb
+++ b/spec/arel/select_manager_spec.rb
@@ -1,7 +1,55 @@
 require 'spec_helper'
 
 module Arel
+  class EngineProxy
+    attr_reader :executed
+
+    def initialize engine
+      @engine = engine
+      @executed = []
+    end
+
+    def connection
+      self
+    end
+
+    def quote_table_name thing; @engine.connection.quote_table_name thing end
+    def quote_column_name thing; @engine.connection.quote_column_name thing end
+    def quote thing, column; @engine.connection.quote thing, column end
+
+    def execute sql
+      @executed << sql
+    end
+  end
+
   describe 'select manager' do
+    describe 'update' do
+      it 'copies where clauses' do
+        engine  = EngineProxy.new Table.engine
+        table   = Table.new :users
+        manager = Arel::SelectManager.new engine
+        manager.where table[:id].eq 10
+        manager.from table
+        manager.update(table[:id] => 1)
+
+        engine.executed.last.should be_like %{
+          UPDATE "users" SET "id" = 1 WHERE "users"."id" = 10
+        }
+      end
+
+      it 'executes an update statement' do
+        engine  = EngineProxy.new Table.engine
+        table   = Table.new :users
+        manager = Arel::SelectManager.new engine
+        manager.from table
+        manager.update(table[:id] => 1)
+
+        engine.executed.last.should be_like %{
+          UPDATE "users" SET "id" = 1
+        }
+      end
+    end
+
     describe 'project' do
       it 'takes strings' do
         table   = Table.new :users