From c7eef391dbcf3364f3792c0e0bbffd4bc3e139d0 Mon Sep 17 00:00:00 2001
From: Aaron Patterson <aaron.patterson@gmail.com>
Date: Mon, 21 Feb 2011 15:40:38 -0800
Subject: refactoring custom lock logic to use sql literals

---
 test/visitors/test_mysql.rb    | 4 ++--
 test/visitors/test_postgres.rb | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

(limited to 'test/visitors')

diff --git a/test/visitors/test_mysql.rb b/test/visitors/test_mysql.rb
index ee70d4c174..3c15c218b2 100644
--- a/test/visitors/test_mysql.rb
+++ b/test/visitors/test_mysql.rb
@@ -31,12 +31,12 @@ module Arel
 
       describe 'locking' do
         it 'defaults to FOR UPDATE when locking' do
-          node = Nodes::Lock.new true
+          node = Nodes::Lock.new(Arel.sql('FOR UPDATE'))
           @visitor.accept(node).must_be_like "FOR UPDATE"
         end
 
         it 'allows a custom string to be used as a lock' do
-          node = Nodes::Lock.new('LOCK IN SHARE MODE')
+          node = Nodes::Lock.new(Arel.sql('LOCK IN SHARE MODE'))
           @visitor.accept(node).must_be_like "LOCK IN SHARE MODE"
         end
       end
diff --git a/test/visitors/test_postgres.rb b/test/visitors/test_postgres.rb
index 169510185e..74446c23ba 100644
--- a/test/visitors/test_postgres.rb
+++ b/test/visitors/test_postgres.rb
@@ -9,13 +9,13 @@ module Arel
 
       describe 'locking' do
         it 'defaults to FOR UPDATE' do
-          @visitor.accept(Nodes::Lock.new(true)).must_be_like %{
+          @visitor.accept(Nodes::Lock.new(Arel.sql('FOR UPDATE'))).must_be_like %{
             FOR UPDATE
           }
         end
 
         it 'allows a custom string to be used as a lock' do
-          node = Nodes::Lock.new('FOR SHARE')
+          node = Nodes::Lock.new(Arel.sql('FOR SHARE'))
           @visitor.accept(node).must_be_like %{
             FOR SHARE
           }
-- 
cgit v1.2.3