lib/dat-tcp/server_spy.rb in dat-tcp-0.6.0 vs lib/dat-tcp/server_spy.rb in dat-tcp-0.7.0

@@ -5,10 +5,12 @@
   class ServerSpy
 
     attr_reader :ip, :port, :file_descriptor
     attr_reader :client_file_descriptors
     attr_reader :logger
+    attr_reader :worker_start_procs, :worker_shutdown_procs
+    attr_reader :worker_sleep_procs, :worker_wakeup_procs
     attr_reader :waiting_for_pause, :waiting_for_stop, :waiting_for_halt
     attr_accessor :listen_called, :start_called
     attr_accessor :stop_listen_called, :pause_called
     attr_accessor :stop_called, :halt_called
 
@@ -19,13 +21,18 @@
       @port = nil
       @file_descriptor = nil
       @client_file_descriptors = []
       @logger = DatTCP::Logger::Null.new
 
+      @worker_start_procs    = []
+      @worker_shutdown_procs = []
+      @worker_sleep_procs    = []
+      @worker_wakeup_procs   = []
+
       @waiting_for_pause = nil
       @waiting_for_stop = nil
-      @waiting_for_hale = nil
+      @waiting_for_halt = nil
 
       @listen_called = false
       @stop_listen_called = false
       @start_called = false
       @pause_called = false
@@ -74,9 +81,14 @@
 
     def halt(wait = false)
       @waiting_for_halt = wait
       @halt_called = true
     end
+
+    def on_worker_start(&block);    @worker_start_procs << block;    end
+    def on_worker_shutdown(&block); @worker_shutdown_procs << block; end
+    def on_worker_sleep(&block);    @worker_sleep_procs << block;    end
+    def on_worker_wakeup(&block);   @worker_wakeup_procs << block;   end
 
   end
 
 end