+ReconnectTestCase

evilkost · Apr 14, 2011 · bbed682 · bbed682
1 parent 57f8103
commit bbed682
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/server_commands.py b/tests/server_commands.py
@@ -78,6 +78,9 @@ def callback(result):
                         msg=source_line+'  Got:'+repr(result))
         return callback
 
+    def delayed(self, timeout, cb):
+        self.loop.add_timeout(time.time()+timeout, cb)
+
     def finish(self, *args):
         self.loop.stop()
 
@@ -713,5 +716,16 @@ def simulate(client, callbacks):
         self.loop.add_callback(lambda: simulate(self.client, self.finish))
         self.start()
 
+class ReconnectTestCase(TornadoTestCase):
+    def test_run_stop_redis(self):
+        self.client.set('foo', 'bar', self.expect(True))
+        self.delayed(10, lambda:
+            self.client.get('foo', [
+                self.expect('bar'),
+                self.finish
+            ])
+        )
+        self.start()
+
 if __name__ == '__main__':
     unittest.main()