fix a test

nats-io · Feb 26, 2019 · efc5716 · efc5716
1 parent 4df24db
commit efc5716
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/nats/io/client.py b/nats/io/client.py
@@ -323,6 +323,8 @@ def connect(self,
                 self._status = Client.CONNECTING
                 yield self._process_connect_init()
                 break
+            except ErrNoServers:
+                raise
             except Exception as e:
                 self._status = Client.DISCONNECTED
                 self._err = e

diff --git a/tests/client_test.py b/tests/client_test.py
@@ -1082,10 +1082,13 @@ def disconnected_cb(self):
             "read_chunk_size": 10,
             "error_cb": c.error_cb,
             "close_cb": c.close_cb,
-            "disconnected_cb": c.disconnected_cb
+            "disconnected_cb": c.disconnected_cb,
+            "max_reconnect_attempts": 1,
         }
-        with self.assertRaises(tornado.iostream.StreamBufferFullError):
+        with self.assertRaises(ErrNoServers):
             yield c.nc.connect(**options)
+        self.assertEqual(
+            tornado.iostream.StreamBufferFullError, c.nc.last_error().__class__)
         self.assertFalse(c.nc.is_connected)
         self.assertEqual(1024, c.nc._max_read_buffer_size)
         self.assertEqual(50, c.nc._max_write_buffer_size)