2004-12-31 14:38:04 -05:00
|
|
|
require 'abstract_unit'
|
|
|
|
require 'fixtures/person'
|
2006-01-14 04:36:52 -05:00
|
|
|
require 'fixtures/legacy_thing'
|
2004-12-31 14:38:04 -05:00
|
|
|
|
2006-09-03 20:02:38 -04:00
|
|
|
class LockWithoutDefault < ActiveRecord::Base; end
|
|
|
|
|
|
|
|
class LockWithCustomColumnWithoutDefault < ActiveRecord::Base
|
2006-11-01 15:28:48 -05:00
|
|
|
set_table_name :lock_without_defaults_cust
|
2006-09-03 20:02:38 -04:00
|
|
|
set_locking_column :custom_lock_version
|
|
|
|
end
|
|
|
|
|
2006-06-19 18:48:51 -04:00
|
|
|
class OptimisticLockingTest < Test::Unit::TestCase
|
2006-01-14 04:36:52 -05:00
|
|
|
fixtures :people, :legacy_things
|
2004-12-31 14:38:04 -05:00
|
|
|
|
|
|
|
def test_lock_existing
|
|
|
|
p1 = Person.find(1)
|
|
|
|
p2 = Person.find(1)
|
2006-06-20 15:54:35 -04:00
|
|
|
assert_equal 0, p1.lock_version
|
|
|
|
assert_equal 0, p2.lock_version
|
2006-06-19 18:48:51 -04:00
|
|
|
|
2006-06-20 15:54:35 -04:00
|
|
|
p1.save!
|
|
|
|
assert_equal 1, p1.lock_version
|
|
|
|
assert_equal 0, p2.lock_version
|
2006-06-19 18:48:51 -04:00
|
|
|
|
2006-06-20 15:54:35 -04:00
|
|
|
assert_raises(ActiveRecord::StaleObjectError) { p2.save! }
|
2004-12-31 14:38:04 -05:00
|
|
|
end
|
|
|
|
|
|
|
|
def test_lock_new
|
2006-06-20 15:54:35 -04:00
|
|
|
p1 = Person.new(:first_name => 'anika')
|
|
|
|
assert_equal 0, p1.lock_version
|
|
|
|
|
|
|
|
p1.save!
|
2004-12-31 14:38:04 -05:00
|
|
|
p2 = Person.find(p1.id)
|
2006-06-20 15:54:35 -04:00
|
|
|
assert_equal 0, p1.lock_version
|
|
|
|
assert_equal 0, p2.lock_version
|
|
|
|
|
|
|
|
p1.save!
|
|
|
|
assert_equal 1, p1.lock_version
|
|
|
|
assert_equal 0, p2.lock_version
|
|
|
|
|
|
|
|
assert_raises(ActiveRecord::StaleObjectError) { p2.save! }
|
2006-01-14 04:36:52 -05:00
|
|
|
end
|
2006-06-19 18:48:51 -04:00
|
|
|
|
2006-01-14 04:36:52 -05:00
|
|
|
def test_lock_column_name_existing
|
|
|
|
t1 = LegacyThing.find(1)
|
|
|
|
t2 = LegacyThing.find(1)
|
2006-06-20 15:54:35 -04:00
|
|
|
assert_equal 0, t1.version
|
|
|
|
assert_equal 0, t2.version
|
|
|
|
|
|
|
|
t1.save!
|
|
|
|
assert_equal 1, t1.version
|
|
|
|
assert_equal 0, t2.version
|
|
|
|
|
|
|
|
assert_raises(ActiveRecord::StaleObjectError) { t2.save! }
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_lock_column_is_mass_assignable
|
|
|
|
p1 = Person.create(:first_name => 'bianca')
|
|
|
|
assert_equal 0, p1.lock_version
|
|
|
|
assert_equal p1.lock_version, Person.new(p1.attributes).lock_version
|
2006-01-14 04:36:52 -05:00
|
|
|
|
2006-06-20 15:54:35 -04:00
|
|
|
p1.save!
|
|
|
|
assert_equal 1, p1.lock_version
|
|
|
|
assert_equal p1.lock_version, Person.new(p1.attributes).lock_version
|
2006-06-19 18:48:51 -04:00
|
|
|
end
|
2006-12-05 17:07:55 -05:00
|
|
|
|
2006-09-03 20:02:38 -04:00
|
|
|
def test_lock_without_default_sets_version_to_zero
|
|
|
|
t1 = LockWithoutDefault.new
|
|
|
|
assert_equal 0, t1.lock_version
|
|
|
|
end
|
2006-12-05 17:07:55 -05:00
|
|
|
|
2006-09-03 20:02:38 -04:00
|
|
|
def test_lock_with_custom_column_without_default_sets_version_to_zero
|
|
|
|
t1 = LockWithCustomColumnWithoutDefault.new
|
|
|
|
assert_equal 0, t1.custom_lock_version
|
|
|
|
end
|
2006-06-19 18:48:51 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
# TODO: test against the generated SQL since testing locking behavior itself
|
|
|
|
# is so cumbersome. Will deadlock Ruby threads if the underlying db.execute
|
|
|
|
# blocks, so separate script called by Kernel#system is needed.
|
|
|
|
# (See exec vs. async_exec in the PostgreSQL adapter.)
|
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
# TODO: The SQL Server adapter currently has no support for pessimistic locking
|
2006-01-14 04:36:52 -05:00
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
unless current_adapter?(:SQLServerAdapter)
|
|
|
|
class PessimisticLockingTest < Test::Unit::TestCase
|
|
|
|
self.use_transactional_fixtures = false
|
2006-12-06 15:28:26 -05:00
|
|
|
fixtures :people, :readers
|
2006-12-05 17:07:55 -05:00
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
def setup
|
2006-12-06 15:28:26 -05:00
|
|
|
# Avoid introspection queries during tests.
|
|
|
|
Person.columns; Reader.columns
|
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
@allow_concurrency = ActiveRecord::Base.allow_concurrency
|
|
|
|
ActiveRecord::Base.allow_concurrency = true
|
2006-06-19 18:48:51 -04:00
|
|
|
end
|
2006-12-05 17:07:55 -05:00
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
def teardown
|
|
|
|
ActiveRecord::Base.allow_concurrency = @allow_concurrency
|
|
|
|
end
|
2006-12-05 17:07:55 -05:00
|
|
|
|
|
|
|
# Test typical find.
|
2006-07-10 14:24:35 -04:00
|
|
|
def test_sane_find_with_lock
|
|
|
|
assert_nothing_raised do
|
|
|
|
Person.transaction do
|
|
|
|
Person.find 1, :lock => true
|
2006-06-19 18:48:51 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
2006-12-05 17:07:55 -05:00
|
|
|
|
|
|
|
# Test scoped lock.
|
|
|
|
def test_sane_find_with_scoped_lock
|
2006-07-10 14:24:35 -04:00
|
|
|
assert_nothing_raised do
|
|
|
|
Person.transaction do
|
|
|
|
Person.with_scope(:find => { :lock => true }) do
|
|
|
|
Person.find 1
|
|
|
|
end
|
|
|
|
end
|
2006-06-19 21:58:36 -04:00
|
|
|
end
|
|
|
|
end
|
2006-12-05 17:07:55 -05:00
|
|
|
|
|
|
|
# PostgreSQL protests SELECT ... FOR UPDATE on an outer join.
|
|
|
|
unless current_adapter?(:PostgreSQLAdapter)
|
|
|
|
# Test locked eager find.
|
|
|
|
def test_eager_find_with_lock
|
|
|
|
assert_nothing_raised do
|
|
|
|
Person.transaction do
|
2006-12-06 15:28:26 -05:00
|
|
|
Person.find 1, :include => :readers, :lock => true
|
2006-12-05 17:07:55 -05:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
# Locking a record reloads it.
|
|
|
|
def test_sane_lock_method
|
|
|
|
assert_nothing_raised do
|
|
|
|
Person.transaction do
|
|
|
|
person = Person.find 1
|
|
|
|
old, person.first_name = person.first_name, 'fooman'
|
|
|
|
person.lock!
|
|
|
|
assert_equal old, person.first_name
|
|
|
|
end
|
|
|
|
end
|
2006-06-19 18:48:51 -04:00
|
|
|
end
|
2006-12-05 17:07:55 -05:00
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
if current_adapter?(:PostgreSQLAdapter, :OracleAdapter)
|
|
|
|
def test_no_locks_no_wait
|
|
|
|
first, second = duel { Person.find 1 }
|
|
|
|
assert first.end > second.end
|
|
|
|
end
|
2006-12-05 17:07:55 -05:00
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
def test_second_lock_waits
|
|
|
|
assert [0.2, 1, 5].any? { |zzz|
|
|
|
|
first, second = duel(zzz) { Person.find 1, :lock => true }
|
|
|
|
second.end > first.end
|
|
|
|
}
|
|
|
|
end
|
2006-12-05 17:07:55 -05:00
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
protected
|
|
|
|
def duel(zzz = 5)
|
|
|
|
t0, t1, t2, t3 = nil, nil, nil, nil
|
2006-12-05 17:07:55 -05:00
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
a = Thread.new do
|
|
|
|
t0 = Time.now
|
|
|
|
Person.transaction do
|
|
|
|
yield
|
|
|
|
sleep zzz # block thread 2 for zzz seconds
|
|
|
|
end
|
|
|
|
t1 = Time.now
|
2006-06-19 18:48:51 -04:00
|
|
|
end
|
2006-12-05 17:07:55 -05:00
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
b = Thread.new do
|
|
|
|
sleep zzz / 2.0 # ensure thread 1 tx starts first
|
|
|
|
t2 = Time.now
|
|
|
|
Person.transaction { yield }
|
|
|
|
t3 = Time.now
|
|
|
|
end
|
2006-12-05 17:07:55 -05:00
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
a.join
|
|
|
|
b.join
|
2006-12-05 17:07:55 -05:00
|
|
|
|
2006-07-10 14:24:35 -04:00
|
|
|
assert t1 > t0 + zzz
|
|
|
|
assert t2 > t0
|
|
|
|
assert t3 > t2
|
|
|
|
[t0.to_f..t1.to_f, t2.to_f..t3.to_f]
|
2006-06-19 18:48:51 -04:00
|
|
|
end
|
2006-07-10 14:24:35 -04:00
|
|
|
end
|
2006-06-19 18:48:51 -04:00
|
|
|
end
|
2005-06-10 10:58:02 -04:00
|
|
|
end
|