From f3834f730bfec68adcdcee1c740c04c924b7acb1 Mon Sep 17 00:00:00 2001 From: zinduolis Date: Sat, 6 Sep 2025 10:52:04 +1000 Subject: [PATCH] Remove unnecessary server preparation step and ensure ActiveRecord disconnection in teardown --- spec/beef/extensions/websocket_hooked_browser_spec.rb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spec/beef/extensions/websocket_hooked_browser_spec.rb b/spec/beef/extensions/websocket_hooked_browser_spec.rb index 3c8ea5e7b..d25ff37d9 100644 --- a/spec/beef/extensions/websocket_hooked_browser_spec.rb +++ b/spec/beef/extensions/websocket_hooked_browser_spec.rb @@ -56,7 +56,6 @@ RSpec.describe 'Browser hooking with Websockets', run_on_browserstack: true do # Spawn HTTP Server print_info 'Starting HTTP Hook Server' http_hook_server = BeEF::Core::Server.instance - http_hook_server.prepare # Generate a token for the server to respond with @token = BeEF::Core::Crypto.api_token @@ -90,12 +89,13 @@ RSpec.describe 'Browser hooking with Websockets', run_on_browserstack: true do sleep 1 until wait.until { @driver.execute_script('return window.beef.session.get_hook_session_id().length') > 0 } - @session = @driver.execute_script('return window.beef.session.get_hook_session_id().length') + @session = @driver.execute_script('return window.beef.session.get_hook_session_id()') end end after(:all) do server_teardown(@driver, @pid, @pids) + disconnect_all_active_record! end it 'confirms a websocket server has been started' do