From dc8e8677410573fc009b2d3f17259f10d49f3231 Mon Sep 17 00:00:00 2001
From: Sean Arietta <sean.arietta@gmail.com>
Date: Tue, 23 Dec 2014 21:39:48 -0800
Subject: [PATCH] Added tests that verify restart functionality is working as
 expected

---
 socketIO_client/tests.py | 68 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 65 insertions(+), 3 deletions(-)

diff --git a/socketIO_client/tests.py b/socketIO_client/tests.py
index ea20282..a0e6223 100644
--- a/socketIO_client/tests.py
+++ b/socketIO_client/tests.py
@@ -1,4 +1,5 @@
 import logging
+from subprocess import check_call
 import time
 from unittest import TestCase
 
@@ -15,6 +16,9 @@ class BaseMixin(object):
 
     def setUp(self):
         self.called_on_response = False
+        
+        # Start the server if it's not already started.
+        check_call("./start-test-server.sh");        
 
     def tearDown(self):
         del self.socketIO
@@ -27,6 +31,65 @@ class BaseMixin(object):
                 self.assertEqual(arg, DATA)
         self.called_on_response = True
 
+    def test_server_dies_during_define(self):
+        'Server dies in the middle of a send'
+        self.socketIO.wait(self.wait_time_in_seconds);
+        self.assertTrue(self.socketIO.connected);
+
+        check_call("./kill-test-server.sh");
+
+        news_namespace = self.socketIO.define(Namespace, '/news');
+        self.assertFalse(self.socketIO.connected);
+
+        check_call("./start-test-server.sh");
+
+        main_namespace = self.socketIO.define(Namespace);
+        chat_namespace = self.socketIO.define(Namespace, '/chat');
+        news_namespace = self.socketIO.define(Namespace, '/news');
+        news_namespace.emit('emit_with_payload', PAYLOAD);
+
+        self.socketIO.wait(self.wait_time_in_seconds);
+        self.assertEqual(main_namespace.args_by_event, {});
+        self.assertEqual(chat_namespace.args_by_event, {});
+        self.assertEqual(news_namespace.args_by_event, {
+            'emit_with_payload_response': (PAYLOAD,),
+        });
+
+    def test_server_dies_during_emit_and_re_emit(self):
+        'Server dies in the middle of a send'
+        namespace = self.socketIO.define(Namespace);
+
+        self.socketIO.wait(self.wait_time_in_seconds);
+        self.assertTrue(self.socketIO.connected);
+
+        check_call("./kill-test-server.sh");
+
+        self.socketIO.emit("emit");
+        self.assertFalse(self.socketIO.connected);
+        check_call("./start-test-server.sh");
+
+        #self.socketIO.emit("emit");
+        self.socketIO.wait(self.wait_time_in_seconds);
+        self.assertEqual(namespace.args_by_event, {
+            'emit_response': (),
+        });
+
+    def test_server_restart(self):
+        'Server restart'
+        self.assertTrue(self.socketIO.connected);
+
+        check_call("./kill-test-server.sh");
+        time.sleep(2);
+        self.socketIO.wait(self.wait_time_in_seconds)
+
+        self.assertFalse(self.socketIO.connected);
+
+        check_call("./start-test-server.sh");
+        time.sleep(2);
+        self.socketIO.wait(self.wait_time_in_seconds);
+
+        self.assertTrue(self.socketIO.connected);
+
     def test_disconnect(self):
         'Disconnect'
         self.assertTrue(self.socketIO.connected)
@@ -143,21 +206,20 @@ class Test_WebsocketTransport(BaseMixin, TestCase):
         self.socketIO = SocketIO(HOST, PORT)
         self.wait_time_in_seconds = 0.1
 
-
 class Test_XHR_PollingTransport(BaseMixin, TestCase):
 
     def setUp(self):
         super(Test_XHR_PollingTransport, self).setUp()
         self.socketIO = SocketIO(HOST, PORT)
         self.wait_time_in_seconds = 1
-
+"""
 class Test_JSONP_PollingTransport(TestCase, BaseMixin):
 
     def setUp(self):
         super(Test_JSONP_PollingTransport, self).setUp()
         self.socketIO = SocketIO(HOST, PORT, transports = ['jsonp-polling'])
         self.wait_time_in_seconds = 1;
-
+"""
 class Namespace(BaseNamespace):
 
     def initialize(self):