Extract Msf::Module::UI::*::Verbose and shared examples

MSP-11126

lib/msf/core/module.rb

@@ -176,31 +176,6 @@ class Module
     obj
   end
 
-  # Verbose version of #print_status
-  def vprint_status(msg)
-    print_status(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
-  end
-  # Verbose version of #print_error
-  def vprint_error(msg)
-    print_error(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
-  end
-  # Verbose version of #print_good
-  def vprint_good(msg)
-    print_good(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
-  end
-  # Verbose version of #print_line
-  def vprint_line(msg)
-    print_line(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
-  end
-  # Verbose version of #print_debug
-  def vprint_debug(msg)
-    print_debug(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
-  end
-  # Verbose version of #print_warning
-  def vprint_warning(msg)
-    print_warning(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
-  end
-
   #
   # Returns the module's framework full reference name.  This is the
   # short name that end-users work with (refname) plus the type

lib/msf/core/module/ui/line.rb

@@ -1,4 +1,8 @@
 module Msf::Module::UI::Line
+  autoload :Verbose, 'msf/core/module/ui/line/verbose'
+
+  include Msf::Module::UI::Line::Verbose
+
   def print_line(msg='')
     super(print_line_prefix + msg)
   end

lib/msf/core/module/ui/line/verbose.rb

@@ -0,0 +1,6 @@
+module Msf::Module::UI::Line::Verbose
+  # Verbose version of #print_line
+  def vprint_line(msg)
+    print_line(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
+  end
+end

lib/msf/core/module/ui/message.rb

@@ -1,5 +1,9 @@
 # Methods for print messages with status indicators
 module Msf::Module::UI::Message
+  autoload :Verbose, 'msf/core/module/ui/message/verbose'
+
+  include Msf::Module::UI::Message::Verbose
+
   def print_error(msg='')
     super(print_prefix + msg)
   end

lib/msf/core/module/ui/message/verbose.rb

@@ -0,0 +1,26 @@
+module Msf::Module::UI::Message::Verbose
+  # Verbose version of #print_debug
+  def vprint_debug(msg)
+    print_debug(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
+  end
+
+  # Verbose version of #print_error
+  def vprint_error(msg)
+    print_error(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
+  end
+
+  # Verbose version of #print_good
+  def vprint_good(msg)
+    print_good(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
+  end
+
+  # Verbose version of #print_status
+  def vprint_status(msg)
+    print_status(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
+  end
+
+  # Verbose version of #print_warning
+  def vprint_warning(msg)
+    print_warning(msg) if datastore['VERBOSE'] || framework.datastore['VERBOSE']
+  end
+end

spec/lib/msf/core/module_spec.rb

@@ -96,12 +96,6 @@ describe Msf::Module do
   it { is_expected.to respond_to :type }
   it { is_expected.to respond_to :update_info }
   it { is_expected.to respond_to :validate }
-  it { is_expected.to respond_to :vprint_debug }
-  it { is_expected.to respond_to :vprint_error }
-  it { is_expected.to respond_to :vprint_good }
-  it { is_expected.to respond_to :vprint_line }
-  it { is_expected.to respond_to :vprint_status }
-  it { is_expected.to respond_to :vprint_warning }
   it { is_expected.to respond_to :workspace }
 
   it_should_behave_like 'Msf::Module::Compatibility'

spec/support/shared/examples/msf/module/ui/line.rb

@@ -1,4 +1,6 @@
 shared_examples_for 'Msf::Module::UI::Line' do
+  it_should_behave_like 'Msf::Module::UI::Line::Verbose'
+
   it { is_expected.to respond_to :print_line }
   it { is_expected.to respond_to :print_line_prefix }
 end

spec/support/shared/examples/msf/module/ui/line/verbose.rb

@@ -0,0 +1,3 @@
+shared_examples_for 'Msf::Module::UI::Line::Verbose' do
+  it { is_expected.to respond_to :vprint_line }
+end

spec/support/shared/examples/msf/module/ui/message.rb

@@ -1,4 +1,6 @@
 shared_examples_for 'Msf::Module::UI::Message' do
+  it_should_behave_like 'Msf::Module::UI::Message::Verbose'
+
   it { is_expected.to respond_to :print_error }
   it { is_expected.to respond_to :print_good }
   it { is_expected.to respond_to :print_prefix }

spec/support/shared/examples/msf/module/ui/message/verbose.rb

@@ -0,0 +1,7 @@
+shared_examples_for 'Msf::Module::UI::Message::Verbose' do
+  it { is_expected.to respond_to :vprint_debug }
+  it { is_expected.to respond_to :vprint_error }
+  it { is_expected.to respond_to :vprint_good }
+  it { is_expected.to respond_to :vprint_status }
+  it { is_expected.to respond_to :vprint_warning }
+end