metasploit-gs/spec/support/shared/contexts/msf/ui_driver.rb

RSpec.shared_context 'Msf::UIDriver' do
  let(:driver) do
    double(
      'Driver',
      :framework => framework
    ).tap { |driver|
      allow(driver).to receive(:on_command_proc=).with(kind_of(Proc))
      allow(driver).to receive(:print_line).with(kind_of(String)) do |string|
        @output ||= []
        @output.concat string.split("\n")
      end
      allow(driver).to receive(:print_status).with(kind_of(String)) do |string|
        @output ||= []
        @output.concat string.split("\n")
      end
      allow(driver).to receive(:print_error).with(kind_of(String)) do |string|
        @error ||= []
        @error.concat string.split("\n")
      end
      allow(driver).to receive(:print_bad).with(kind_of(String)) do |string|
        @error ||= []
        @error.concat string.split("\n")
      end
    }
  end
end