update test__backdoor.py

greentest/test__backdoor.py

@@ -17,14 +17,27 @@ def read_until(conn, postfix):
 class Test(greentest.TestCase):
 
     def test(self):
-        server = backdoor.BackdoorServer.spawn(('127.0.0.1', 7891))
-        gevent.sleep(0)
-        conn = socket.create_connection(('127.0.0.1', 7891))
+        server = backdoor.BackdoorServer(('127.0.0.1', 0))
+        server.start()
+        def connect():
+            conn = socket.create_connection(('127.0.0.1', server.server_port))
+            read_until(conn, '>>> ')
+            conn.sendall('2+2\r\n')
+            line = conn.makefile().readline()
+            assert line.strip() == '4', repr(line)
+        jobs = [gevent.spawn(connect) for _ in xrange(10)]
+        gevent.joinall(jobs)
+        server.kill()
+        #self.assertEqual(conn.recv(1), '')
+
+    def test_quit(self):
+        server = backdoor.BackdoorServer(('127.0.0.1', 0))
+        server.start()
+        conn = socket.create_connection(('127.0.0.1', server.server_port))
         read_until(conn, '>>> ')
-        conn.sendall('2+2\r\n')
-        line = conn.makefile().readline()
-        assert line.strip() == '4', repr(line)
-        server.kill(block=True)
+        conn.sendall('quit()\r\n')
+        line = conn.makefile().read()
+        self.assertEqual(line, '')
 
 
 if __name__ == '__main__':