From 6c65b0176223c90e08903ebc1a9a1075659f2c43 Mon Sep 17 00:00:00 2001 From: Jesse Storimer Date: Thu, 20 Jan 2011 00:37:05 -0500 Subject: Allow database specific locking clauses to be used --- test/visitors/test_mysql.rb | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) (limited to 'test/visitors/test_mysql.rb') diff --git a/test/visitors/test_mysql.rb b/test/visitors/test_mysql.rb index c3b79ca667..c22cbaff19 100644 --- a/test/visitors/test_mysql.rb +++ b/test/visitors/test_mysql.rb @@ -29,11 +29,16 @@ module Arel sql.must_be_like "SELECT FROM DUAL" end - it 'uses FOR UPDATE when locking' do - stmt = Nodes::SelectStatement.new - stmt.lock = Nodes::Lock.new - sql = @visitor.accept(stmt) - sql.must_be_like "SELECT FROM DUAL FOR UPDATE" + describe 'locking' do + it 'defaults to FOR UPDATE when locking' do + node = Nodes::Lock.new + @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') + @visitor.accept(node).must_be_like "LOCK IN SHARE MODE" + end end end end -- cgit v1.2.3