Allow for handler to be injected into ProbeServer

lib/good_job/http_server.rb

@@ -31,6 +31,10 @@ def running?
       @running.true?
     end
 
+    def build_future
+      Concurrent::Future.new { run }
+    end
+
     private
 
     def start_server

lib/good_job/probe_server.rb

@@ -8,14 +8,14 @@ def self.task_observer(time, output, thread_error) # rubocop:disable Lint/Unused
       GoodJob._on_thread_error(thread_error) if thread_error
     end
 
-    def initialize(port:, app: default_probe_server)
+    def initialize(port:, handler: nil, app: default_probe_server)
       @port = port
       @app = app
+      @handler = handler || default_handler
     end
 
     def start
-      @handler = HttpServer.new(@app, port: @port, logger: GoodJob.logger)
-      @future = Concurrent::Future.new { @handler.run }
+      @future = @handler.build_future
       @future.add_observer(self.class, :task_observer)
       @future.execute
     end
@@ -37,5 +37,9 @@ def default_probe_server
         run Middleware::Catchall.new
       end
     end
+
+    def default_handler
+      HttpServer.new(@app, port: @port, logger: GoodJob.logger)
+    end
   end
 end

spec/lib/good_job/cli_spec.rb

@@ -77,7 +77,6 @@
             options: {},
             daemonize?: false,
             shutdown_timeout: 100,
-            idle_timeout: 100
           )
           allow(GoodJob).to receive_messages(configuration: configuration_mock)
           cli = described_class.new([], [], {})