Contents

require 'proletariat'

# Public: Test fixture
class PingWorker < Proletariat::Worker
  listen_on :ping

  class << self
    attr_accessor :pinged
  end

  def work(message, routing_key, headers)
    self.class.pinged = true

    log 'PING'
    sleep 0.5
    publish 'pong'

    :ok
  end
end

# Public: Test fixture
class PongWorker < Proletariat::Worker
  listen_on :pong

  class << self
    attr_accessor :fail_mode
    attr_accessor :ponged
  end

  def work(message, routing_key, headers)
    if self.class.fail_mode == true
      fail 'Error' unless headers['x-death'] && headers['x-death'].length == 2
    end

    self.class.ponged = true

    log 'PONG'
    sleep 0.5
    publish 'ping'

    :ok
  end
end

describe Proletariat do
  before do
    Proletariat.configure do
      config.exchange_name  = 'proletariat-test-suite'
      config.logger         = Logger.new('/dev/null')
      config.test_mode!
      config.worker_classes = [PingWorker, PongWorker]
    end

    PongWorker.fail_mode = false

    Proletariat.run

    sleep 1
  end

  after do
    Proletariat.stop

    PingWorker.pinged = false
    PongWorker.ponged = false

    sleep 0.5
  end

  it 'should roughly work' do
    Proletariat.publish 'ping', ''
    sleep 1

    expect(PingWorker.pinged).to be_truthy
    expect(PongWorker.ponged).to be_truthy
  end

  it 'should purge between tests' do
    sleep 1

    expect(PingWorker.pinged).to be_falsey
    expect(PongWorker.ponged).to be_falsey
  end

  it 'should work in error conditions' do
    PongWorker.fail_mode = true
    Proletariat.publish 'ping', ''
    sleep 15

    expect(PingWorker.pinged).to be_truthy
    expect(PongWorker.ponged).to be_truthy
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
proletariat-0.1.2	spec/lib/proletariat_spec.rb
proletariat-0.1.1	spec/lib/proletariat_spec.rb
proletariat-0.1.0	spec/lib/proletariat_spec.rb