From 15fd00a69a80d8085dd515fa898734faef501ee7 Mon Sep 17 00:00:00 2001 From: zinduolis Date: Sat, 6 Sep 2025 09:58:00 +1000 Subject: [PATCH] Refactor server start-up process in AutoRunEngine and Websocket tests for improved initialization --- spec/beef/core/main/autorun_engine/autorun_engine_spec.rb | 5 ++--- spec/beef/extensions/websocket_hooked_browser_spec.rb | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) 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 6b42e8eb6..d94f14727 100644 --- a/spec/beef/core/main/autorun_engine/autorun_engine_spec.rb +++ b/spec/beef/core/main/autorun_engine/autorun_engine_spec.rb @@ -72,10 +72,9 @@ RSpec.describe 'AutoRunEngine Test', run_on_browserstack: true do disconnect_all_active_record! # Initiate server start-up - @pids = fork do - BeEF::API::Registrar.instance.fire(BeEF::API::Server, 'pre_http_start', http_hook_server) - end @pid = fork do + http_hook_server.prepare + BeEF::API::Registrar.instance.fire(BeEF::API::Server, 'pre_http_start', http_hook_server) http_hook_server.start end diff --git a/spec/beef/extensions/websocket_hooked_browser_spec.rb b/spec/beef/extensions/websocket_hooked_browser_spec.rb index 45b02cc89..3c8ea5e7b 100644 --- a/spec/beef/extensions/websocket_hooked_browser_spec.rb +++ b/spec/beef/extensions/websocket_hooked_browser_spec.rb @@ -65,10 +65,9 @@ RSpec.describe 'Browser hooking with Websockets', run_on_browserstack: true do disconnect_all_active_record! # Initiate server start-up - @pids = fork do - BeEF::API::Registrar.instance.fire(BeEF::API::Server, 'pre_http_start', http_hook_server) - end @pid = fork do + http_hook_server.prepare + BeEF::API::Registrar.instance.fire(BeEF::API::Server, 'pre_http_start', http_hook_server) http_hook_server.start end