diff --git a/spec/beef/core/main/handlers/browser_details_handler_spec.rb b/spec/beef/core/main/handlers/browser_details_handler_spec.rb index a3c4abb98..f48f70837 100644 --- a/spec/beef/core/main/handlers/browser_details_handler_spec.rb +++ b/spec/beef/core/main/handlers/browser_details_handler_spec.rb @@ -99,11 +99,8 @@ RSpec.describe 'Browser Details Handler', :run_on_browserstack => true do @hook_request = RestClient.get "#{RESTAPI_HOOKS}?token=#{@token}" @hooks = JSON.parse(@hook_request) - if @hooks['hooked-browsers']['online'].empty? - @session = @driver.execute_script("return window.beef.session.get_hook_session_id()") - else - @session = @hooks['hooked-browsers']['online'][0]['session'] - end + @session = @hooks['hooked-browsers']['online'][0]['session'] + rescue => exception print_info "Exception: #{exception}" print_info "Exception Class: #{exception.class}" diff --git a/spec/beef/modules/debug/test_beef_debugs_spec.rb b/spec/beef/modules/debug/test_beef_debugs_spec.rb index 786d401b8..569aac970 100644 --- a/spec/beef/modules/debug/test_beef_debugs_spec.rb +++ b/spec/beef/modules/debug/test_beef_debugs_spec.rb @@ -99,11 +99,7 @@ RSpec.describe 'BeEF Debug Command Modules:', :run_on_browserstack => true do @hook_request = RestClient.get "#{RESTAPI_HOOKS}?token=#{@token}" @hooks = JSON.parse(@hook_request) - if @hooks['hooked-browsers']['online'].empty? - @session = @driver.execute_script("return window.beef.session.get_hook_session_id()") - else - @session = @hooks['hooked-browsers']['online'][0]['session'] - end + @session = @driver.execute_script("return window.beef.session.get_hook_session_id()") # Grab Command Module IDs as they can differ from machine to machine @debug_mod_ids = JSON.parse(RestClient.get "#{RESTAPI_MODULES}?token=#{@token}")