1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
# frozen_string_literal: true
require 'test_helper'
class SeparateThreadTest < GemTestCase
setup do
@lock_name = 'testing 1,2,3' # OMG COMMAS
@mutex = Mutex.new
@t1_acquired_lock = false
@t1_return_value = nil
@t1 = Thread.new do
Label.connection_pool.with_connection do
@t1_return_value = Label.with_advisory_lock(@lock_name) do
@mutex.synchronize { @t1_acquired_lock = true }
sleep
't1 finished'
end
end
end
# Wait for the thread to acquire the lock:
sleep(0.1) until @mutex.synchronize { @t1_acquired_lock }
Label.connection.reconnect!
end
teardown do
@t1.wakeup if @t1.status == 'sleep'
@t1.join
end
test '#with_advisory_lock with a 0 timeout returns false immediately' do
response = Label.with_advisory_lock(@lock_name, 0) do
raise 'should not be yielded to'
end
assert_not(response)
end
test '#with_advisory_lock yields to the provided block' do
assert(@t1_acquired_lock)
end
test '#advisory_lock_exists? returns true when another thread has the lock' do
assert(Tag.advisory_lock_exists?(@lock_name))
end
test 'can re-establish the lock after the other thread releases it' do
@t1.wakeup
@t1.join
assert_equal('t1 finished', @t1_return_value)
# We should now be able to acquire the lock immediately:
reacquired = false
lock_result = Label.with_advisory_lock(@lock_name, 0) do
reacquired = true
end
assert(lock_result)
assert(reacquired)
end
end
|
