123 lines
3.3 KiB
Ruby
123 lines
3.3 KiB
Ruby
require_relative '../../spec_helper_min.rb'
|
|
require_relative '../../../lib/carto/bolt.rb'
|
|
|
|
module Carto
|
|
describe Bolt do
|
|
before(:each) do
|
|
@bolt = Carto::Bolt.new('manolo_bolt_locked')
|
|
end
|
|
|
|
after(:each) do
|
|
@bolt.send :retried
|
|
end
|
|
|
|
it 'should expect block' do
|
|
expect { @bolt.run_locked.should_raise }.to raise_error('no code block given')
|
|
end
|
|
|
|
it 'should allow access in block if locked and unlocked automatically' do
|
|
bolt = Carto::Bolt.new('manolo_bolt_locked', ttl_ms: 60000)
|
|
|
|
bolt.run_locked {}.should be_true
|
|
bolt.run_locked {}.should be_true
|
|
end
|
|
|
|
it 'should not allow access if locked' do
|
|
other_bolt = Carto::Bolt.new('manolo_bolt_locked')
|
|
|
|
@bolt.run_locked { other_bolt.run_locked {}.should be_false }.should be_true
|
|
end
|
|
|
|
it 'should wait for execution if we pass attempts parameters' do
|
|
main = Thread.new do
|
|
@bolt.run_locked {
|
|
sleep(2)
|
|
}.should be_true
|
|
end
|
|
sleep(0.5)
|
|
thr = Thread.new do
|
|
flag = false
|
|
Carto::Bolt.new('manolo_bolt_locked').run_locked(attempts: 5, timeout: 1000) { flag = true }
|
|
flag.should be_true
|
|
end
|
|
thr.join
|
|
main.join
|
|
end
|
|
|
|
it 'should wait for execution and exit without complete it if timeout and retries reach the limit' do
|
|
main = Thread.new do
|
|
@bolt.run_locked {
|
|
sleep(2)
|
|
}.should be_true
|
|
end
|
|
sleep(0.5)
|
|
thr = Thread.new do
|
|
flag = false
|
|
Carto::Bolt.new('manolo_bolt_locked').run_locked(attempts: 5, timeout: 1) { flag = true }
|
|
flag.should be_false
|
|
end
|
|
thr.join
|
|
main.join
|
|
end
|
|
|
|
it 'should retry an execution when other process tries to acquire bolt and has retriable flag set' do
|
|
flag = 0
|
|
rerun_func = lambda { flag += 1 }
|
|
main = Thread.new do
|
|
@bolt.run_locked(fail_function: rerun_func) {
|
|
flag += 1
|
|
sleep(2)
|
|
}.should be_true
|
|
end
|
|
sleep(0.5)
|
|
thr = Thread.new do
|
|
Carto::Bolt.new('manolo_bolt_locked').run_locked(fail_function: rerun_func) {}.should be_false
|
|
end
|
|
thr.join
|
|
main.join
|
|
flag.should eq(2)
|
|
end
|
|
|
|
it 'should execute once the fail_function part despite of the number of calls to acquire the lock' do
|
|
flag = 0
|
|
rerun_func = lambda do
|
|
flag += 1
|
|
end
|
|
main = Thread.new do
|
|
@bolt.run_locked(fail_function: rerun_func) {
|
|
flag += 1
|
|
sleep(2)
|
|
}.should be_true
|
|
end
|
|
sleep(0.5)
|
|
10.times do
|
|
t = Thread.new do
|
|
Carto::Bolt.new('manolo_bolt_locked').run_locked(fail_function: rerun_func) {}
|
|
sleep(0.1)
|
|
end
|
|
t.join
|
|
end
|
|
main.join
|
|
flag.should eq(2)
|
|
end
|
|
|
|
it 'should raise error if fail_function is not a lambda' do
|
|
expect {
|
|
@bolt.run_locked(fail_function: "lala") {}.should_raise
|
|
}.to raise_error('no proc/lambda passed as fail_function')
|
|
end
|
|
|
|
it 'should expire a lock after ttl_ms' do
|
|
ttl_ms = 200
|
|
|
|
bolt = Carto::Bolt.new('manolo_bolt_locked', ttl_ms: ttl_ms)
|
|
|
|
bolt.run_locked do
|
|
sleep((ttl_ms * 2 / 1000.0).second)
|
|
|
|
Carto::Bolt.new('manolo_bolt_locked').run_locked {}.should be_true
|
|
end
|
|
end
|
|
end
|
|
end
|