Sha256: da178df6a77a9fc7fd739f8b4b32adf9c957d0074b0abdb0dc52f632021665f2
Contents?: true
Size: 1.76 KB
Versions: 3
Compression:
Stored size: 1.76 KB
Contents
require 'evented-spec/ext/amqp'
module EventedSpec
module SpecHelper
# Represents spec running inside AMQP.start loop
# See {EventedExample} for details and method descriptions.
class AMQPExample < EMExample
# Run @block inside the AMQP.start loop.
# See {EventedExample#run}
def run
run_em_loop do
::AMQP.start_connection(@opts) do
run_hooks :amqp_before
@example_group_instance.instance_eval(&@block)
end
end
end
# Breaks the event loop and finishes the spec. It yields to any given block first,
# then stops AMQP, EM event loop and cleans up AMQP state.
#
# See {EventedExample#done}
def done(delay = nil)
delayed(delay) do
yield if block_given?
EM.next_tick do
run_hooks :amqp_after
if ::AMQP.connection && !::AMQP.closing?
::AMQP.stop_connection do
# Cannot call finish_em_loop before connection is marked as closed
# This callback is called before that happens.
EM.next_tick { finish_em_loop }
end
else
# Need this branch because if AMQP couldn't connect,
# the callback would never trigger
::AMQP.cleanup_state
EM.next_tick { finish_em_loop }
end
end
end
end
# Called from run_event_loop when event loop is finished, before any exceptions
# is raised or example returns. We ensure AMQP state cleanup here.
#
# See {EventedExample#run}
def finish_example
::AMQP.cleanup_state
super
end
end # class AMQPExample < EventedExample
end # module SpecHelper
end # module EventedExample
Version data entries
3 entries across 3 versions & 1 rubygems