From c3e162a8f5ddc4e56ceca83364b86218cafd4f10 Mon Sep 17 00:00:00 2001 From: kaitoozawa Date: Thu, 18 Dec 2025 09:39:14 +1000 Subject: [PATCH] address review comments --- spec/beef/core/main/handlers/browser_details_handler_spec.rb | 2 +- spec/beef/extensions/websocket_hooked_browser_spec.rb | 2 +- spec/spec_helper.rb | 2 ++ 3 files changed, 4 insertions(+), 2 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 9d7ff1080..bf6f761ee 100644 --- a/spec/beef/core/main/handlers/browser_details_handler_spec.rb +++ b/spec/beef/core/main/handlers/browser_details_handler_spec.rb @@ -106,7 +106,7 @@ RSpec.describe 'Browser Details Handler', run_on_browserstack: true do after(:all) do server_teardown(@driver, @pid, @pids) disconnect_all_active_record! - @__ar_config_snapshot = SpecActiveRecordConnection.snapshot + SpecActiveRecordConnection.restore!(@__ar_config_snapshot) end it 'can successfully hook a browser' do diff --git a/spec/beef/extensions/websocket_hooked_browser_spec.rb b/spec/beef/extensions/websocket_hooked_browser_spec.rb index 4058429b4..5d4978e05 100644 --- a/spec/beef/extensions/websocket_hooked_browser_spec.rb +++ b/spec/beef/extensions/websocket_hooked_browser_spec.rb @@ -105,7 +105,7 @@ RSpec.describe 'Browser hooking with Websockets', run_on_browserstack: true do after(:all) do server_teardown(@driver, @pid, @pids) disconnect_all_active_record! - @__ar_config_snapshot = SpecActiveRecordConnection.snapshot + SpecActiveRecordConnection.restore!(@__ar_config_snapshot) end it 'confirms a websocket server has been started' do diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb index 73aa41101..b9a260367 100644 --- a/spec/spec_helper.rb +++ b/spec/spec_helper.rb @@ -271,6 +271,8 @@ require 'socket' # Generate a token for the server to respond with BeEF::Core::Crypto::api_token + disconnect_all_active_record! + # Initiate server start-up BeEF::API::Registrar.instance.fire(BeEF::API::Server, 'pre_http_start', http_hook_server) pid = fork do