23 lines
515 B
Ruby
23 lines
515 B
Ruby
|
class TestUI < Vagrant::UI::Interface
|
||
|
|
attr_reader :messages
|
||
|
|
|
||
|
|
METHODS = [:clear_line, :report_progress, :warn, :error, :info, :success]
|
||
|
|
|
||
|
|
def initialize(resource = nil)
|
||
|
|
super
|
||
|
|
@messages = METHODS.each_with_object({}) { |m, h| h[m] = [] }
|
||
|
|
end
|
||
|
|
|
||
|
|
def ask(*args)
|
||
|
|
super
|
||
|
|
# Automated tests should not depend on user input, obviously.
|
||
|
|
raise Errors::UIExpectsTTY
|
||
|
|
end
|
||
|
|
|
||
|
|
METHODS.each do |method|
|
||
|
|
define_method(method) do |message, *opts|
|
||
|
|
@messages[method].push message
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|