nil check and fix mock

Author: dwelch-r7

lib/msf/core/module/alert.rb

@@ -270,10 +270,13 @@ def alert_user
     # Temporarily set the prompt mode to false to ensure that there are not additional lines printed
     # A workaround for the prompting bug spotted in https://github.com/rapid7/metasploit-framework/pull/18761#issuecomment-1916645095
   def without_prompt(&block)
-    previous_prompting_value = user_output.prompting
-    user_output.prompting(false)
+    if user_output
+      previous_prompting_value = user_output.prompting?
+      user_output.prompting(false)
+    end
+
     yield
   ensure
-    user_output.prompting(previous_prompting_value)
+    user_output.prompting(previous_prompting_value) if user_output
   end
 end

spec/support/shared/contexts/msf/ui_driver.rb

@@ -13,7 +13,8 @@
   let(:driver_output) do
     instance = double(
       Rex::Ui::Text::Output,
-      prompting?: false
+      prompting?: false,
+      prompting: false
     )
 
     capture_logging(instance)