From dbb6cb13328231b4efbbecc66eb15d119febce25 Mon Sep 17 00:00:00 2001 From: Jack Walker Date: Wed, 27 May 2020 18:41:23 +1000 Subject: [PATCH] Missing key on hook variable assignment --- spec/beef/core/main/handlers/browser_details_handler_spec.rb | 2 +- spec/beef/modules/debug/test_beef_debugs_spec.rb | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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 bebb9b9e8..a3c4abb98 100644 --- a/spec/beef/core/main/handlers/browser_details_handler_spec.rb +++ b/spec/beef/core/main/handlers/browser_details_handler_spec.rb @@ -102,7 +102,7 @@ RSpec.describe 'Browser Details Handler', :run_on_browserstack => true do 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 = @hooks['hooked-browsers']['online'][0]['session'] end rescue => exception print_info "Exception: #{exception}" diff --git a/spec/beef/modules/debug/test_beef_debugs_spec.rb b/spec/beef/modules/debug/test_beef_debugs_spec.rb index b1208398e..786d401b8 100644 --- a/spec/beef/modules/debug/test_beef_debugs_spec.rb +++ b/spec/beef/modules/debug/test_beef_debugs_spec.rb @@ -98,11 +98,11 @@ 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 = @hooks['hooked-browsers']['online'][0]['session'] end # Grab Command Module IDs as they can differ from machine to machine