lib/ws-io.rb in ws-io-1.0.0 vs lib/ws-io.rb in ws-io-1.0.1

@@ -34,10 +34,11 @@
 
       threads << Thread.start do
         begin
           yield
         ensure
+          unfake_io
           stop_server
         end
       end
 
       threads << Thread.start do
@@ -54,13 +55,17 @@
       end
 
       threads.each do |thread|
         thread.join
       end
-    rescue SignalException, StandardError
+    rescue SignalException, StandardError => e
+      g e
       unfake_io
-    rescue Exception
+      stop_server
+    rescue Exception => e
+      g e
       unfake_io
+      stop_server
       raise
     end
 
     def fake_io
       @in_read, @in_write = IO.pipe