cartodb-4.42/spec/lib/carto/bolt_spec.rb

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