From 7854d600a74179d5f92a2f382f01da10da99d4d5 Mon Sep 17 00:00:00 2001
From: Robey Pointer <robey@lag.net>
Date: Sat, 22 Mar 2008 19:07:09 -0700
Subject: [project @ robey@lag.net-20080323020709-ub8ya3xkh1ebywfa] add
 exit_status_ready for justin cook.

---
 tests/test_transport.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'tests')

diff --git a/tests/test_transport.py b/tests/test_transport.py
index 894c69d3..f46fda7f 100644
--- a/tests/test_transport.py
+++ b/tests/test_transport.py
@@ -283,6 +283,7 @@ class TransportTest (unittest.TestCase):
         schan = self.ts.accept(1.0)
         chan.exec_command('yes')
         schan.send('Hello there.\n')
+        self.assert_(not chan.exit_status_ready())
         # trigger an EOF
         schan.shutdown_read()
         schan.shutdown_write()
@@ -292,6 +293,12 @@ class TransportTest (unittest.TestCase):
         f = chan.makefile()
         self.assertEquals('Hello there.\n', f.readline())
         self.assertEquals('', f.readline())
+        count = 0
+        while not chan.exit_status_ready():
+            time.sleep(0.1)
+            count += 1
+            if count > 50:
+                raise Exception("timeout")
         self.assertEquals(23, chan.recv_exit_status())
         chan.close()
 
-- 
cgit v1.2.3