94 lines
2.3 KiB
Python
94 lines
2.3 KiB
Python
|
# test asynchat -- requires threading
|
||
|
|
|
||
|
|
import thread # If this fails, we can't test this module
|
||
|
|
import asyncore, asynchat, socket, threading, time
|
||
|
|
import unittest
|
||
|
|
from test import test_support
|
||
|
|
|
||
|
|
HOST = "127.0.0.1"
|
||
|
|
PORT = 54322
|
||
|
|
|
||
|
|
class echo_server(threading.Thread):
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||
|
|
global PORT
|
||
|
|
PORT = test_support.bind_port(sock, HOST, PORT)
|
||
|
|
sock.listen(1)
|
||
|
|
conn, client = sock.accept()
|
||
|
|
buffer = ""
|
||
|
|
while "\n" not in buffer:
|
||
|
|
data = conn.recv(1)
|
||
|
|
if not data:
|
||
|
|
break
|
||
|
|
buffer = buffer + data
|
||
|
|
while buffer:
|
||
|
|
n = conn.send(buffer)
|
||
|
|
buffer = buffer[n:]
|
||
|
|
conn.close()
|
||
|
|
sock.close()
|
||
|
|
|
||
|
|
class echo_client(asynchat.async_chat):
|
||
|
|
|
||
|
|
def __init__(self, terminator):
|
||
|
|
asynchat.async_chat.__init__(self)
|
||
|
|
self.contents = None
|
||
|
|
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
|
|
self.connect((HOST, PORT))
|
||
|
|
self.set_terminator(terminator)
|
||
|
|
self.buffer = ""
|
||
|
|
|
||
|
|
def handle_connect(self):
|
||
|
|
pass
|
||
|
|
##print "Connected"
|
||
|
|
|
||
|
|
def collect_incoming_data(self, data):
|
||
|
|
self.buffer = self.buffer + data
|
||
|
|
|
||
|
|
def found_terminator(self):
|
||
|
|
#print "Received:", repr(self.buffer)
|
||
|
|
self.contents = self.buffer
|
||
|
|
self.buffer = ""
|
||
|
|
self.close()
|
||
|
|
|
||
|
|
|
||
|
|
class TestAsynchat(unittest.TestCase):
|
||
|
|
def setUp (self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def tearDown (self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def test_line_terminator(self):
|
||
|
|
s = echo_server()
|
||
|
|
s.start()
|
||
|
|
time.sleep(1) # Give server time to initialize
|
||
|
|
c = echo_client('\n')
|
||
|
|
c.push("hello ")
|
||
|
|
c.push("world\n")
|
||
|
|
asyncore.loop()
|
||
|
|
s.join()
|
||
|
|
|
||
|
|
self.assertEqual(c.contents, 'hello world')
|
||
|
|
|
||
|
|
def test_numeric_terminator(self):
|
||
|
|
# Try reading a fixed number of bytes
|
||
|
|
s = echo_server()
|
||
|
|
s.start()
|
||
|
|
time.sleep(1) # Give server time to initialize
|
||
|
|
c = echo_client(6L)
|
||
|
|
c.push("hello ")
|
||
|
|
c.push("world\n")
|
||
|
|
asyncore.loop()
|
||
|
|
s.join()
|
||
|
|
|
||
|
|
self.assertEqual(c.contents, 'hello ')
|
||
|
|
|
||
|
|
|
||
|
|
def test_main(verbose=None):
|
||
|
|
test_support.run_unittest(TestAsynchat)
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
test_main(verbose=True)
|