address review comments

This commit is contained in:
kaitoozawa
2025-12-18 09:39:14 +10:00
parent 7ec386881e
commit c3e162a8f5
3 changed files with 4 additions and 2 deletions

View File

@@ -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

View File

@@ -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

View File

@@ -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