spec/support/mock_server.rb in ruby_fs-1.0.4 vs spec/support/mock_server.rb in ruby_fs-1.0.5

@@ -1,25 +1,27 @@
 MockServer = Class.new
 
 class ServerMock
   include Celluloid::IO
 
+  finalizer :finalize
+
   def initialize(host, port, mock_target = MockServer.new)
     @server = TCPServer.new host, port
     @mock_target = mock_target
     @clients = []
-    run!
+    async.run
   end
 
   def finalize
     Logger.debug "ServerMock finalizing"
     @server.close if @server
     @clients.each(&:close)
   end
 
   def run
     after(1) { terminate }
-    loop { handle_connection! @server.accept }
+    loop { async.handle_connection @server.accept }
   end
 
   def handle_connection(socket)
     @clients << socket
     _, port, host = socket.peeraddr