93 lines
2.1 KiB
Ruby
93 lines
2.1 KiB
Ruby
require File.expand_path("../../../base", __FILE__)
|
|
|
|
describe Vagrant::Action::Warden do
|
|
let(:data) { { :data => [] } }
|
|
let(:instance) { described_class.new }
|
|
|
|
# This returns a proc that can be used with the builder
|
|
# that simply appends data to an array in the env.
|
|
def appender_proc(data)
|
|
Proc.new { |env| env[:data] << data }
|
|
end
|
|
|
|
it "calls the actions like normal" do
|
|
instance = described_class.new([appender_proc(1), appender_proc(2)], data)
|
|
instance.call(data)
|
|
|
|
data[:data].should == [1, 2]
|
|
end
|
|
|
|
it "starts a recovery sequence when an exception is raised" do
|
|
class Action
|
|
def initialize(app, env)
|
|
@app = app
|
|
end
|
|
|
|
def call(env)
|
|
@app.call(env)
|
|
end
|
|
|
|
def recover(env)
|
|
env[:recover] << 1
|
|
end
|
|
end
|
|
|
|
class ActionTwo
|
|
def initialize(app, env)
|
|
@app = app
|
|
end
|
|
|
|
def call(env)
|
|
@app.call(env)
|
|
end
|
|
|
|
def recover(env)
|
|
env[:recover] << 2
|
|
end
|
|
end
|
|
|
|
error_proc = Proc.new { raise "ERROR!" }
|
|
|
|
data = { :recover => [] }
|
|
instance = described_class.new([Action, ActionTwo, error_proc], data)
|
|
|
|
# The error should be raised back up
|
|
expect { instance.call(data) }.
|
|
to raise_error(RuntimeError)
|
|
|
|
# Verify the recovery process goes in reverse order
|
|
data[:recover].should == [2, 1]
|
|
|
|
# Verify that the error is available in the data
|
|
data["vagrant.error"].should be_kind_of(RuntimeError)
|
|
end
|
|
|
|
it "does not do a recovery sequence if SystemExit is raised" do
|
|
class Action
|
|
def initialize(app, env)
|
|
@app = app
|
|
end
|
|
|
|
def call(env)
|
|
@app.call(env)
|
|
end
|
|
|
|
def recover(env)
|
|
env[:recover] = true
|
|
end
|
|
end
|
|
|
|
# Make a proc that just calls "abort" which raises a
|
|
# SystemExit exception.
|
|
error_proc = Proc.new { abort }
|
|
|
|
instance = described_class.new([Action, error_proc], data)
|
|
|
|
# The SystemExit should come through
|
|
expect { instance.call(data) }.to raise_error(SystemExit)
|
|
|
|
# The recover should not have been called
|
|
data.has_key?(:recover).should_not be
|
|
end
|
|
end
|