test/test_cometio.rb in sinatra-cometio-0.1.5 vs test/test_cometio.rb in sinatra-cometio-0.1.6

@@ -24,10 +24,11 @@
 
     30.times do
       break if res != nil
       sleep 0.1
     end
+    client.close
     assert res != nil, 'server not respond'
     assert res["time"] == post_data[:time]
     assert res["msg"] == post_data[:msg]
   end
 
@@ -48,10 +49,12 @@
         post_data['to'] = session2
         client.push :message, post_data
       end
       client2.on :message do |data|
         res2 = data
+        client2.close
+        client.close
       end
     end
 
     30.times do
       break if res != nil
@@ -69,19 +72,21 @@
     client = CometIO::Client.new(App.cometio_url).connect
     post_data = {:time => Time.now.to_s, :msg => 'hello!!'}
     res = nil
     client.on :broadcast do |data|
       res = data
+      client.close
     end
 
     res2 = nil
     client.on :connect do |session|
       client2 = CometIO::Client.new(App.cometio_url).connect
       client2.on :connect do |session2|
         client.push :broadcast, post_data
       end
       client2.on :broadcast do |data|
         res2 = data
+        client2.close
       end
     end
 
     30.times do
       break if res != nil