23 lines
499 B
Ruby
23 lines
499 B
Ruby
class TestUI < Vagrant::UI::Interface
|
|
attr_reader :messages
|
|
|
|
METHODS = [:clear_line, :report_progress, :warn, :error, :info, :success]
|
|
|
|
def initialize
|
|
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
|