diff --git a/spec/beef/core/main/autorun_engine/autorun_engine_spec.rb b/spec/beef/core/main/autorun_engine/autorun_engine_spec.rb index 8e3495183..1103f220a 100644 --- a/spec/beef/core/main/autorun_engine/autorun_engine_spec.rb +++ b/spec/beef/core/main/autorun_engine/autorun_engine_spec.rb @@ -98,7 +98,7 @@ RSpec.describe 'AutoRunEngine Test', :run_on_browserstack => true do puts @driver.execute_script("return window.beef.session.get_hook_session_id()") @hooks = JSON.parse(RestClient.get "#{RESTAPI_HOOKS}?token=#{@token}") - @session = @hooks['hooked-browsers']['online']['0']['session'] + @session = @hooks['hooked-browsers']['online']['0']['session'] || @driver.execute_script("return window.beef.session.get_hook_session_id()") end after(:all) do 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 9b5b93044..33cbc4a7e 100644 --- a/spec/beef/core/main/handlers/browser_details_handler_spec.rb +++ b/spec/beef/core/main/handlers/browser_details_handler_spec.rb @@ -95,7 +95,7 @@ RSpec.describe 'Browser Details Handler', :run_on_browserstack => true do puts @driver.execute_script("return window.beef.session.get_hook_session_id()") @hooks = JSON.parse(RestClient.get "#{RESTAPI_HOOKS}?token=#{@token}") - @session = @hooks['hooked-browsers']['online']['0']['session'] + @session = @hooks['hooked-browsers']['online']['0']['session'] || @driver.execute_script("return window.beef.session.get_hook_session_id()") end after(:all) do diff --git a/spec/beef/modules/debug/test_beef_debugs_spec.rb b/spec/beef/modules/debug/test_beef_debugs_spec.rb index f1903cbb5..e841f7953 100644 --- a/spec/beef/modules/debug/test_beef_debugs_spec.rb +++ b/spec/beef/modules/debug/test_beef_debugs_spec.rb @@ -94,7 +94,7 @@ RSpec.describe 'BeEF Debug Command Modules:', :run_on_browserstack => true do puts @driver.execute_script("return window.beef.session.get_hook_session_id()") @hooks = JSON.parse(RestClient.get "#{RESTAPI_HOOKS}?token=#{@token}") - @session = @hooks['hooked-browsers']['online']['0']['session'] + @session = @hooks['hooked-browsers']['online']['0']['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}")