[4] Fix: Pass callbacks parameter to the _sudden_disconnect function.

brukva/client.py

@@ -174,7 +174,7 @@ def try_to_loop(self):
     def schedule(self, command, callbacks, *args, **kwargs):
         self.queue.append(Task(command, callbacks, args, kwargs))
 
-    def _sudden_disconnect(self):
+    def _sudden_disconnect(self, callbacks):
         self.connection.disconnect()
         self.call_callbacks(callbacks, (ConnectionError("Socket closed on remote end"), None))
 
@@ -199,14 +199,14 @@ def _process_response(self, callbacks=None):
 
     def _parse_value_response(self, callbacks, data):
         if not data:
-            self._sudden_disconnect()
+            self._sudden_disconnect(callbacks)
             return
         data = data[:-2] # strip \r\n
         self.call_callbacks(callbacks, (None, data))
 
     def _parse_command_response(self, callbacks, data):
         if not data:
-            self._sudden_disconnect()
+            self._sudden_disconnect(callbacks)
             return
         data = data[:-2] # strip \r\n
         if data == '$-1':
@@ -239,7 +239,7 @@ def execute_command(self, cmd, callbacks, *args, **kwargs):
         try:
             self.connection.write(self.format(cmd, *args, **kwargs))
         except IOError:
-            self._sudden_disconnect()
+            self._sudden_disconnect(callbacks)
             return
         self.schedule(cmd, callbacks, *args, **kwargs)
         self.try_to_loop()