2008-06-13 19:38:33 -03:00
|
|
|
import asyncore
|
|
|
|
import unittest
|
|
|
|
import select
|
|
|
|
import os
|
|
|
|
import socket
|
|
|
|
import sys
|
|
|
|
import time
|
2010-05-06 14:57:06 -03:00
|
|
|
import warnings
|
2010-05-18 17:04:31 -03:00
|
|
|
import errno
|
2012-03-22 12:06:43 -03:00
|
|
|
import struct
|
2008-06-13 19:38:33 -03:00
|
|
|
|
|
|
|
from test import test_support
|
2013-08-21 19:39:46 -03:00
|
|
|
from test.test_support import TESTFN, run_unittest, unlink, HOST
|
2008-06-13 19:38:33 -03:00
|
|
|
from StringIO import StringIO
|
|
|
|
|
2010-04-27 20:55:59 -03:00
|
|
|
try:
|
|
|
|
import threading
|
|
|
|
except ImportError:
|
|
|
|
threading = None
|
|
|
|
|
2008-06-13 19:38:33 -03:00
|
|
|
|
|
|
|
class dummysocket:
|
|
|
|
def __init__(self):
|
|
|
|
self.closed = False
|
|
|
|
|
|
|
|
def close(self):
|
|
|
|
self.closed = True
|
|
|
|
|
|
|
|
def fileno(self):
|
|
|
|
return 42
|
|
|
|
|
|
|
|
class dummychannel:
|
|
|
|
def __init__(self):
|
|
|
|
self.socket = dummysocket()
|
|
|
|
|
|
|
|
def close(self):
|
|
|
|
self.socket.close()
|
|
|
|
|
|
|
|
class exitingdummy:
|
|
|
|
def __init__(self):
|
|
|
|
pass
|
|
|
|
|
|
|
|
def handle_read_event(self):
|
|
|
|
raise asyncore.ExitNow()
|
|
|
|
|
|
|
|
handle_write_event = handle_read_event
|
2008-07-11 20:26:59 -03:00
|
|
|
handle_close = handle_read_event
|
2008-06-13 19:38:33 -03:00
|
|
|
handle_expt_event = handle_read_event
|
|
|
|
|
|
|
|
class crashingdummy:
|
|
|
|
def __init__(self):
|
|
|
|
self.error_handled = False
|
|
|
|
|
|
|
|
def handle_read_event(self):
|
|
|
|
raise Exception()
|
|
|
|
|
|
|
|
handle_write_event = handle_read_event
|
2008-07-11 20:26:59 -03:00
|
|
|
handle_close = handle_read_event
|
2008-06-13 19:38:33 -03:00
|
|
|
handle_expt_event = handle_read_event
|
|
|
|
|
|
|
|
def handle_error(self):
|
|
|
|
self.error_handled = True
|
|
|
|
|
|
|
|
# used when testing senders; just collects what it gets until newline is sent
|
|
|
|
def capture_server(evt, buf, serv):
|
|
|
|
try:
|
|
|
|
serv.listen(5)
|
|
|
|
conn, addr = serv.accept()
|
|
|
|
except socket.timeout:
|
|
|
|
pass
|
|
|
|
else:
|
|
|
|
n = 200
|
|
|
|
while n > 0:
|
|
|
|
r, w, e = select.select([conn], [], [])
|
|
|
|
if r:
|
|
|
|
data = conn.recv(10)
|
|
|
|
# keep everything except for the newline terminator
|
|
|
|
buf.write(data.replace('\n', ''))
|
|
|
|
if '\n' in data:
|
|
|
|
break
|
|
|
|
n -= 1
|
|
|
|
time.sleep(0.01)
|
|
|
|
|
|
|
|
conn.close()
|
|
|
|
finally:
|
|
|
|
serv.close()
|
|
|
|
evt.set()
|
|
|
|
|
|
|
|
|
|
|
|
class HelperFunctionTests(unittest.TestCase):
|
|
|
|
def test_readwriteexc(self):
|
|
|
|
# Check exception handling behavior of read, write and _exception
|
|
|
|
|
|
|
|
# check that ExitNow exceptions in the object handler method
|
|
|
|
# bubbles all the way up through asyncore read/write/_exception calls
|
|
|
|
tr1 = exitingdummy()
|
|
|
|
self.assertRaises(asyncore.ExitNow, asyncore.read, tr1)
|
|
|
|
self.assertRaises(asyncore.ExitNow, asyncore.write, tr1)
|
|
|
|
self.assertRaises(asyncore.ExitNow, asyncore._exception, tr1)
|
|
|
|
|
|
|
|
# check that an exception other than ExitNow in the object handler
|
|
|
|
# method causes the handle_error method to get called
|
|
|
|
tr2 = crashingdummy()
|
|
|
|
asyncore.read(tr2)
|
|
|
|
self.assertEqual(tr2.error_handled, True)
|
|
|
|
|
|
|
|
tr2 = crashingdummy()
|
|
|
|
asyncore.write(tr2)
|
|
|
|
self.assertEqual(tr2.error_handled, True)
|
|
|
|
|
|
|
|
tr2 = crashingdummy()
|
|
|
|
asyncore._exception(tr2)
|
|
|
|
self.assertEqual(tr2.error_handled, True)
|
|
|
|
|
|
|
|
# asyncore.readwrite uses constants in the select module that
|
|
|
|
# are not present in Windows systems (see this thread:
|
|
|
|
# http://mail.python.org/pipermail/python-list/2001-October/109973.html)
|
|
|
|
# These constants should be present as long as poll is available
|
|
|
|
|
2010-07-27 20:45:05 -03:00
|
|
|
@unittest.skipUnless(hasattr(select, 'poll'), 'select.poll required')
|
|
|
|
def test_readwrite(self):
|
|
|
|
# Check that correct methods are called by readwrite()
|
|
|
|
|
|
|
|
attributes = ('read', 'expt', 'write', 'closed', 'error_handled')
|
|
|
|
|
|
|
|
expected = (
|
|
|
|
(select.POLLIN, 'read'),
|
|
|
|
(select.POLLPRI, 'expt'),
|
|
|
|
(select.POLLOUT, 'write'),
|
|
|
|
(select.POLLERR, 'closed'),
|
|
|
|
(select.POLLHUP, 'closed'),
|
|
|
|
(select.POLLNVAL, 'closed'),
|
|
|
|
)
|
|
|
|
|
|
|
|
class testobj:
|
|
|
|
def __init__(self):
|
|
|
|
self.read = False
|
|
|
|
self.write = False
|
|
|
|
self.closed = False
|
|
|
|
self.expt = False
|
|
|
|
self.error_handled = False
|
|
|
|
|
|
|
|
def handle_read_event(self):
|
|
|
|
self.read = True
|
|
|
|
|
|
|
|
def handle_write_event(self):
|
|
|
|
self.write = True
|
|
|
|
|
|
|
|
def handle_close(self):
|
|
|
|
self.closed = True
|
|
|
|
|
|
|
|
def handle_expt_event(self):
|
|
|
|
self.expt = True
|
|
|
|
|
|
|
|
def handle_error(self):
|
|
|
|
self.error_handled = True
|
|
|
|
|
|
|
|
for flag, expectedattr in expected:
|
|
|
|
tobj = testobj()
|
|
|
|
self.assertEqual(getattr(tobj, expectedattr), False)
|
|
|
|
asyncore.readwrite(tobj, flag)
|
|
|
|
|
|
|
|
# Only the attribute modified by the routine we expect to be
|
|
|
|
# called should be True.
|
|
|
|
for attr in attributes:
|
|
|
|
self.assertEqual(getattr(tobj, attr), attr==expectedattr)
|
|
|
|
|
|
|
|
# check that ExitNow exceptions in the object handler method
|
|
|
|
# bubbles all the way up through asyncore readwrite call
|
|
|
|
tr1 = exitingdummy()
|
|
|
|
self.assertRaises(asyncore.ExitNow, asyncore.readwrite, tr1, flag)
|
|
|
|
|
|
|
|
# check that an exception other than ExitNow in the object handler
|
|
|
|
# method causes the handle_error method to get called
|
|
|
|
tr2 = crashingdummy()
|
|
|
|
self.assertEqual(tr2.error_handled, False)
|
|
|
|
asyncore.readwrite(tr2, flag)
|
|
|
|
self.assertEqual(tr2.error_handled, True)
|
2008-06-13 19:38:33 -03:00
|
|
|
|
|
|
|
def test_closeall(self):
|
|
|
|
self.closeall_check(False)
|
|
|
|
|
|
|
|
def test_closeall_default(self):
|
|
|
|
self.closeall_check(True)
|
|
|
|
|
|
|
|
def closeall_check(self, usedefault):
|
|
|
|
# Check that close_all() closes everything in a given map
|
|
|
|
|
|
|
|
l = []
|
|
|
|
testmap = {}
|
|
|
|
for i in range(10):
|
|
|
|
c = dummychannel()
|
|
|
|
l.append(c)
|
|
|
|
self.assertEqual(c.socket.closed, False)
|
|
|
|
testmap[i] = c
|
|
|
|
|
|
|
|
if usedefault:
|
|
|
|
socketmap = asyncore.socket_map
|
|
|
|
try:
|
|
|
|
asyncore.socket_map = testmap
|
|
|
|
asyncore.close_all()
|
|
|
|
finally:
|
|
|
|
testmap, asyncore.socket_map = asyncore.socket_map, socketmap
|
|
|
|
else:
|
|
|
|
asyncore.close_all(testmap)
|
|
|
|
|
|
|
|
self.assertEqual(len(testmap), 0)
|
|
|
|
|
|
|
|
for c in l:
|
|
|
|
self.assertEqual(c.socket.closed, True)
|
|
|
|
|
|
|
|
def test_compact_traceback(self):
|
|
|
|
try:
|
|
|
|
raise Exception("I don't like spam!")
|
|
|
|
except:
|
|
|
|
real_t, real_v, real_tb = sys.exc_info()
|
|
|
|
r = asyncore.compact_traceback()
|
|
|
|
else:
|
|
|
|
self.fail("Expected exception")
|
|
|
|
|
|
|
|
(f, function, line), t, v, info = r
|
|
|
|
self.assertEqual(os.path.split(f)[-1], 'test_asyncore.py')
|
|
|
|
self.assertEqual(function, 'test_compact_traceback')
|
|
|
|
self.assertEqual(t, real_t)
|
|
|
|
self.assertEqual(v, real_v)
|
|
|
|
self.assertEqual(info, '[%s|%s|%s]' % (f, function, line))
|
|
|
|
|
|
|
|
|
|
|
|
class DispatcherTests(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
|
|
pass
|
|
|
|
|
|
|
|
def tearDown(self):
|
|
|
|
asyncore.close_all()
|
|
|
|
|
|
|
|
def test_basic(self):
|
|
|
|
d = asyncore.dispatcher()
|
|
|
|
self.assertEqual(d.readable(), True)
|
|
|
|
self.assertEqual(d.writable(), True)
|
|
|
|
|
|
|
|
def test_repr(self):
|
|
|
|
d = asyncore.dispatcher()
|
|
|
|
self.assertEqual(repr(d), '<asyncore.dispatcher at %#x>' % id(d))
|
|
|
|
|
|
|
|
def test_log(self):
|
|
|
|
d = asyncore.dispatcher()
|
|
|
|
|
|
|
|
# capture output of dispatcher.log() (to stderr)
|
|
|
|
fp = StringIO()
|
|
|
|
stderr = sys.stderr
|
|
|
|
l1 = "Lovely spam! Wonderful spam!"
|
|
|
|
l2 = "I don't like spam!"
|
|
|
|
try:
|
|
|
|
sys.stderr = fp
|
|
|
|
d.log(l1)
|
|
|
|
d.log(l2)
|
|
|
|
finally:
|
|
|
|
sys.stderr = stderr
|
|
|
|
|
|
|
|
lines = fp.getvalue().splitlines()
|
2010-07-27 20:45:05 -03:00
|
|
|
self.assertEqual(lines, ['log: %s' % l1, 'log: %s' % l2])
|
2008-06-13 19:38:33 -03:00
|
|
|
|
|
|
|
def test_log_info(self):
|
|
|
|
d = asyncore.dispatcher()
|
|
|
|
|
|
|
|
# capture output of dispatcher.log_info() (to stdout via print)
|
|
|
|
fp = StringIO()
|
|
|
|
stdout = sys.stdout
|
|
|
|
l1 = "Have you got anything without spam?"
|
|
|
|
l2 = "Why can't she have egg bacon spam and sausage?"
|
|
|
|
l3 = "THAT'S got spam in it!"
|
|
|
|
try:
|
|
|
|
sys.stdout = fp
|
|
|
|
d.log_info(l1, 'EGGS')
|
|
|
|
d.log_info(l2)
|
|
|
|
d.log_info(l3, 'SPAM')
|
|
|
|
finally:
|
|
|
|
sys.stdout = stdout
|
|
|
|
|
|
|
|
lines = fp.getvalue().splitlines()
|
2009-04-12 22:06:46 -03:00
|
|
|
expected = ['EGGS: %s' % l1, 'info: %s' % l2, 'SPAM: %s' % l3]
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2010-07-27 20:45:05 -03:00
|
|
|
self.assertEqual(lines, expected)
|
2008-06-13 19:38:33 -03:00
|
|
|
|
|
|
|
def test_unhandled(self):
|
|
|
|
d = asyncore.dispatcher()
|
2009-03-31 22:28:11 -03:00
|
|
|
d.ignore_log_types = ()
|
2008-06-13 19:38:33 -03:00
|
|
|
|
|
|
|
# capture output of dispatcher.log_info() (to stdout via print)
|
|
|
|
fp = StringIO()
|
|
|
|
stdout = sys.stdout
|
|
|
|
try:
|
|
|
|
sys.stdout = fp
|
|
|
|
d.handle_expt()
|
|
|
|
d.handle_read()
|
|
|
|
d.handle_write()
|
|
|
|
d.handle_connect()
|
2012-10-18 08:44:43 -03:00
|
|
|
d.handle_accept()
|
2008-06-13 19:38:33 -03:00
|
|
|
finally:
|
|
|
|
sys.stdout = stdout
|
|
|
|
|
|
|
|
lines = fp.getvalue().splitlines()
|
2009-03-31 22:28:11 -03:00
|
|
|
expected = ['warning: unhandled incoming priority event',
|
2008-06-13 19:38:33 -03:00
|
|
|
'warning: unhandled read event',
|
|
|
|
'warning: unhandled write event',
|
2012-10-18 08:44:43 -03:00
|
|
|
'warning: unhandled connect event',
|
|
|
|
'warning: unhandled accept event']
|
2010-07-27 20:45:05 -03:00
|
|
|
self.assertEqual(lines, expected)
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2010-05-06 14:57:06 -03:00
|
|
|
def test_issue_8594(self):
|
|
|
|
# XXX - this test is supposed to be removed in next major Python
|
|
|
|
# version
|
|
|
|
d = asyncore.dispatcher(socket.socket())
|
|
|
|
# make sure the error message no longer refers to the socket
|
|
|
|
# object but the dispatcher instance instead
|
|
|
|
self.assertRaisesRegexp(AttributeError, 'dispatcher instance',
|
|
|
|
getattr, d, 'foo')
|
|
|
|
# cheap inheritance with the underlying socket is supposed
|
|
|
|
# to still work but a DeprecationWarning is expected
|
|
|
|
with warnings.catch_warnings(record=True) as w:
|
|
|
|
warnings.simplefilter("always")
|
|
|
|
family = d.family
|
|
|
|
self.assertEqual(family, socket.AF_INET)
|
2010-07-27 20:45:05 -03:00
|
|
|
self.assertEqual(len(w), 1)
|
2010-05-06 14:57:06 -03:00
|
|
|
self.assertTrue(issubclass(w[0].category, DeprecationWarning))
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2010-05-18 17:04:31 -03:00
|
|
|
def test_strerror(self):
|
|
|
|
# refers to bug #8573
|
|
|
|
err = asyncore._strerror(errno.EPERM)
|
|
|
|
if hasattr(os, 'strerror'):
|
|
|
|
self.assertEqual(err, os.strerror(errno.EPERM))
|
|
|
|
err = asyncore._strerror(-1)
|
2011-02-25 16:07:51 -04:00
|
|
|
self.assertTrue(err != "")
|
2010-05-18 17:04:31 -03:00
|
|
|
|
2008-06-13 19:38:33 -03:00
|
|
|
|
|
|
|
class dispatcherwithsend_noread(asyncore.dispatcher_with_send):
|
|
|
|
def readable(self):
|
|
|
|
return False
|
|
|
|
|
|
|
|
def handle_connect(self):
|
|
|
|
pass
|
|
|
|
|
|
|
|
class DispatcherWithSendTests(unittest.TestCase):
|
|
|
|
usepoll = False
|
|
|
|
|
|
|
|
def setUp(self):
|
|
|
|
pass
|
|
|
|
|
|
|
|
def tearDown(self):
|
|
|
|
asyncore.close_all()
|
|
|
|
|
2010-04-27 20:55:59 -03:00
|
|
|
@unittest.skipUnless(threading, 'Threading required for this test.')
|
2009-10-27 15:36:47 -03:00
|
|
|
@test_support.reap_threads
|
2008-06-13 19:38:33 -03:00
|
|
|
def test_send(self):
|
2009-10-27 15:36:47 -03:00
|
|
|
evt = threading.Event()
|
|
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
sock.settimeout(3)
|
|
|
|
port = test_support.bind_port(sock)
|
2008-06-13 19:38:33 -03:00
|
|
|
|
|
|
|
cap = StringIO()
|
2009-10-27 15:36:47 -03:00
|
|
|
args = (evt, cap, sock)
|
|
|
|
t = threading.Thread(target=capture_server, args=args)
|
|
|
|
t.start()
|
|
|
|
try:
|
|
|
|
# wait a little longer for the server to initialize (it sometimes
|
|
|
|
# refuses connections on slow machines without this wait)
|
|
|
|
time.sleep(0.2)
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2009-10-27 15:36:47 -03:00
|
|
|
data = "Suppose there isn't a 16-ton weight?"
|
|
|
|
d = dispatcherwithsend_noread()
|
|
|
|
d.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
d.connect((HOST, port))
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2009-10-27 15:36:47 -03:00
|
|
|
# give time for socket to connect
|
|
|
|
time.sleep(0.1)
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2009-10-27 15:36:47 -03:00
|
|
|
d.send(data)
|
|
|
|
d.send(data)
|
|
|
|
d.send('\n')
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2009-10-27 15:36:47 -03:00
|
|
|
n = 1000
|
|
|
|
while d.out_buffer and n > 0:
|
|
|
|
asyncore.poll()
|
|
|
|
n -= 1
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2009-10-27 15:36:47 -03:00
|
|
|
evt.wait()
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2009-10-27 15:36:47 -03:00
|
|
|
self.assertEqual(cap.getvalue(), data*2)
|
|
|
|
finally:
|
|
|
|
t.join()
|
2008-06-13 19:38:33 -03:00
|
|
|
|
|
|
|
|
|
|
|
class DispatcherWithSendTests_UsePoll(DispatcherWithSendTests):
|
|
|
|
usepoll = True
|
|
|
|
|
2010-07-27 20:45:05 -03:00
|
|
|
@unittest.skipUnless(hasattr(asyncore, 'file_wrapper'),
|
|
|
|
'asyncore.file_wrapper required')
|
|
|
|
class FileWrapperTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
|
|
self.d = "It's not dead, it's sleeping!"
|
2011-01-16 09:03:37 -04:00
|
|
|
with file(TESTFN, 'w') as h:
|
|
|
|
h.write(self.d)
|
2010-07-27 20:45:05 -03:00
|
|
|
|
|
|
|
def tearDown(self):
|
|
|
|
unlink(TESTFN)
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2010-07-27 20:45:05 -03:00
|
|
|
def test_recv(self):
|
|
|
|
fd = os.open(TESTFN, os.O_RDONLY)
|
|
|
|
w = asyncore.file_wrapper(fd)
|
|
|
|
os.close(fd)
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2010-07-27 20:45:05 -03:00
|
|
|
self.assertNotEqual(w.fd, fd)
|
|
|
|
self.assertNotEqual(w.fileno(), fd)
|
|
|
|
self.assertEqual(w.recv(13), "It's not dead")
|
|
|
|
self.assertEqual(w.read(6), ", it's")
|
|
|
|
w.close()
|
|
|
|
self.assertRaises(OSError, w.read, 1)
|
2008-06-13 19:38:33 -03:00
|
|
|
|
|
|
|
|
2010-07-27 20:45:05 -03:00
|
|
|
def test_send(self):
|
|
|
|
d1 = "Come again?"
|
|
|
|
d2 = "I want to buy some cheese."
|
|
|
|
fd = os.open(TESTFN, os.O_WRONLY | os.O_APPEND)
|
|
|
|
w = asyncore.file_wrapper(fd)
|
|
|
|
os.close(fd)
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2010-07-27 20:45:05 -03:00
|
|
|
w.write(d1)
|
|
|
|
w.send(d2)
|
|
|
|
w.close()
|
|
|
|
self.assertEqual(file(TESTFN).read(), self.d + d1 + d2)
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2010-08-01 18:18:52 -03:00
|
|
|
@unittest.skipUnless(hasattr(asyncore, 'file_dispatcher'),
|
|
|
|
'asyncore.file_dispatcher required')
|
|
|
|
def test_dispatcher(self):
|
|
|
|
fd = os.open(TESTFN, os.O_RDONLY)
|
|
|
|
data = []
|
|
|
|
class FileDispatcher(asyncore.file_dispatcher):
|
|
|
|
def handle_read(self):
|
|
|
|
data.append(self.recv(29))
|
|
|
|
s = FileDispatcher(fd)
|
|
|
|
os.close(fd)
|
|
|
|
asyncore.loop(timeout=0.01, use_poll=True, count=2)
|
|
|
|
self.assertEqual(b"".join(data), self.d)
|
|
|
|
|
2017-07-26 20:24:52 -03:00
|
|
|
def test_close_twice(self):
|
|
|
|
fd = os.open(TESTFN, os.O_RDONLY)
|
|
|
|
f = asyncore.file_wrapper(fd)
|
|
|
|
os.close(fd)
|
|
|
|
|
|
|
|
os.close(f.fd) # file_wrapper dupped fd
|
|
|
|
with self.assertRaises(OSError):
|
|
|
|
f.close()
|
|
|
|
|
|
|
|
self.assertEqual(f.fd, -1)
|
|
|
|
# calling close twice should not fail
|
|
|
|
f.close()
|
|
|
|
|
2008-06-13 19:38:33 -03:00
|
|
|
|
2010-05-10 12:33:22 -03:00
|
|
|
class BaseTestHandler(asyncore.dispatcher):
|
|
|
|
|
|
|
|
def __init__(self, sock=None):
|
|
|
|
asyncore.dispatcher.__init__(self, sock)
|
|
|
|
self.flag = False
|
|
|
|
|
|
|
|
def handle_accept(self):
|
|
|
|
raise Exception("handle_accept not supposed to be called")
|
|
|
|
|
|
|
|
def handle_connect(self):
|
|
|
|
raise Exception("handle_connect not supposed to be called")
|
|
|
|
|
|
|
|
def handle_expt(self):
|
|
|
|
raise Exception("handle_expt not supposed to be called")
|
|
|
|
|
|
|
|
def handle_close(self):
|
|
|
|
raise Exception("handle_close not supposed to be called")
|
|
|
|
|
|
|
|
def handle_error(self):
|
|
|
|
raise
|
|
|
|
|
|
|
|
|
|
|
|
class TCPServer(asyncore.dispatcher):
|
|
|
|
"""A server which listens on an address and dispatches the
|
|
|
|
connection to a handler.
|
|
|
|
"""
|
|
|
|
|
|
|
|
def __init__(self, handler=BaseTestHandler, host=HOST, port=0):
|
|
|
|
asyncore.dispatcher.__init__(self)
|
|
|
|
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
self.set_reuse_addr()
|
|
|
|
self.bind((host, port))
|
|
|
|
self.listen(5)
|
|
|
|
self.handler = handler
|
|
|
|
|
|
|
|
@property
|
|
|
|
def address(self):
|
|
|
|
return self.socket.getsockname()[:2]
|
|
|
|
|
2012-10-18 08:44:43 -03:00
|
|
|
def handle_accept(self):
|
2012-10-26 23:21:46 -03:00
|
|
|
pair = self.accept()
|
|
|
|
if pair is not None:
|
|
|
|
self.handler(pair[0])
|
2010-05-10 12:33:22 -03:00
|
|
|
|
|
|
|
def handle_error(self):
|
|
|
|
raise
|
|
|
|
|
|
|
|
|
|
|
|
class BaseClient(BaseTestHandler):
|
|
|
|
|
|
|
|
def __init__(self, address):
|
|
|
|
BaseTestHandler.__init__(self)
|
|
|
|
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
self.connect(address)
|
|
|
|
|
|
|
|
def handle_connect(self):
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
class BaseTestAPI(unittest.TestCase):
|
|
|
|
|
|
|
|
def tearDown(self):
|
|
|
|
asyncore.close_all()
|
|
|
|
|
|
|
|
def loop_waiting_for_flag(self, instance, timeout=5):
|
|
|
|
timeout = float(timeout) / 100
|
|
|
|
count = 100
|
|
|
|
while asyncore.socket_map and count > 0:
|
|
|
|
asyncore.loop(timeout=0.01, count=1, use_poll=self.use_poll)
|
|
|
|
if instance.flag:
|
|
|
|
return
|
|
|
|
count -= 1
|
|
|
|
time.sleep(timeout)
|
|
|
|
self.fail("flag not set")
|
|
|
|
|
|
|
|
def test_handle_connect(self):
|
|
|
|
# make sure handle_connect is called on connect()
|
|
|
|
|
|
|
|
class TestClient(BaseClient):
|
|
|
|
def handle_connect(self):
|
|
|
|
self.flag = True
|
|
|
|
|
|
|
|
server = TCPServer()
|
|
|
|
client = TestClient(server.address)
|
|
|
|
self.loop_waiting_for_flag(client)
|
|
|
|
|
|
|
|
def test_handle_accept(self):
|
|
|
|
# make sure handle_accept() is called when a client connects
|
|
|
|
|
|
|
|
class TestListener(BaseTestHandler):
|
|
|
|
|
|
|
|
def __init__(self):
|
|
|
|
BaseTestHandler.__init__(self)
|
|
|
|
self.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
self.bind((HOST, 0))
|
|
|
|
self.listen(5)
|
|
|
|
self.address = self.socket.getsockname()[:2]
|
|
|
|
|
|
|
|
def handle_accept(self):
|
|
|
|
self.flag = True
|
|
|
|
|
|
|
|
server = TestListener()
|
|
|
|
client = BaseClient(server.address)
|
|
|
|
self.loop_waiting_for_flag(server)
|
|
|
|
|
|
|
|
def test_handle_read(self):
|
|
|
|
# make sure handle_read is called on data received
|
|
|
|
|
|
|
|
class TestClient(BaseClient):
|
|
|
|
def handle_read(self):
|
|
|
|
self.flag = True
|
|
|
|
|
|
|
|
class TestHandler(BaseTestHandler):
|
|
|
|
def __init__(self, conn):
|
|
|
|
BaseTestHandler.__init__(self, conn)
|
|
|
|
self.send('x' * 1024)
|
|
|
|
|
|
|
|
server = TCPServer(TestHandler)
|
|
|
|
client = TestClient(server.address)
|
|
|
|
self.loop_waiting_for_flag(client)
|
|
|
|
|
|
|
|
def test_handle_write(self):
|
|
|
|
# make sure handle_write is called
|
|
|
|
|
|
|
|
class TestClient(BaseClient):
|
|
|
|
def handle_write(self):
|
|
|
|
self.flag = True
|
|
|
|
|
|
|
|
server = TCPServer()
|
|
|
|
client = TestClient(server.address)
|
|
|
|
self.loop_waiting_for_flag(client)
|
|
|
|
|
|
|
|
def test_handle_close(self):
|
|
|
|
# make sure handle_close is called when the other end closes
|
|
|
|
# the connection
|
|
|
|
|
|
|
|
class TestClient(BaseClient):
|
|
|
|
|
|
|
|
def handle_read(self):
|
|
|
|
# in order to make handle_close be called we are supposed
|
|
|
|
# to make at least one recv() call
|
|
|
|
self.recv(1024)
|
|
|
|
|
|
|
|
def handle_close(self):
|
|
|
|
self.flag = True
|
|
|
|
self.close()
|
|
|
|
|
|
|
|
class TestHandler(BaseTestHandler):
|
|
|
|
def __init__(self, conn):
|
|
|
|
BaseTestHandler.__init__(self, conn)
|
|
|
|
self.close()
|
|
|
|
|
|
|
|
server = TCPServer(TestHandler)
|
|
|
|
client = TestClient(server.address)
|
|
|
|
self.loop_waiting_for_flag(client)
|
|
|
|
|
|
|
|
@unittest.skipIf(sys.platform.startswith("sunos"),
|
|
|
|
"OOB support is broken on Solaris")
|
|
|
|
def test_handle_expt(self):
|
|
|
|
# Make sure handle_expt is called on OOB data received.
|
|
|
|
# Note: this might fail on some platforms as OOB data is
|
|
|
|
# tenuously supported and rarely used.
|
|
|
|
|
2017-04-03 13:11:22 -03:00
|
|
|
if sys.platform == "darwin" and self.use_poll:
|
|
|
|
self.skipTest("poll may fail on macOS; see issue #28087")
|
|
|
|
|
2010-05-10 12:33:22 -03:00
|
|
|
class TestClient(BaseClient):
|
|
|
|
def handle_expt(self):
|
|
|
|
self.flag = True
|
|
|
|
|
|
|
|
class TestHandler(BaseTestHandler):
|
|
|
|
def __init__(self, conn):
|
|
|
|
BaseTestHandler.__init__(self, conn)
|
|
|
|
self.socket.send(chr(244), socket.MSG_OOB)
|
|
|
|
|
|
|
|
server = TCPServer(TestHandler)
|
|
|
|
client = TestClient(server.address)
|
|
|
|
self.loop_waiting_for_flag(client)
|
|
|
|
|
|
|
|
def test_handle_error(self):
|
|
|
|
|
|
|
|
class TestClient(BaseClient):
|
|
|
|
def handle_write(self):
|
|
|
|
1.0 / 0
|
|
|
|
def handle_error(self):
|
|
|
|
self.flag = True
|
|
|
|
try:
|
|
|
|
raise
|
|
|
|
except ZeroDivisionError:
|
|
|
|
pass
|
|
|
|
else:
|
|
|
|
raise Exception("exception not raised")
|
|
|
|
|
|
|
|
server = TCPServer()
|
|
|
|
client = TestClient(server.address)
|
|
|
|
self.loop_waiting_for_flag(client)
|
|
|
|
|
|
|
|
def test_connection_attributes(self):
|
|
|
|
server = TCPServer()
|
|
|
|
client = BaseClient(server.address)
|
|
|
|
|
|
|
|
# we start disconnected
|
|
|
|
self.assertFalse(server.connected)
|
|
|
|
self.assertTrue(server.accepting)
|
2010-05-11 21:29:27 -03:00
|
|
|
# this can't be taken for granted across all platforms
|
|
|
|
#self.assertFalse(client.connected)
|
2010-05-10 12:33:22 -03:00
|
|
|
self.assertFalse(client.accepting)
|
|
|
|
|
|
|
|
# execute some loops so that client connects to server
|
|
|
|
asyncore.loop(timeout=0.01, use_poll=self.use_poll, count=100)
|
|
|
|
self.assertFalse(server.connected)
|
|
|
|
self.assertTrue(server.accepting)
|
|
|
|
self.assertTrue(client.connected)
|
|
|
|
self.assertFalse(client.accepting)
|
|
|
|
|
|
|
|
# disconnect the client
|
|
|
|
client.close()
|
|
|
|
self.assertFalse(server.connected)
|
|
|
|
self.assertTrue(server.accepting)
|
|
|
|
self.assertFalse(client.connected)
|
|
|
|
self.assertFalse(client.accepting)
|
|
|
|
|
|
|
|
# stop serving
|
|
|
|
server.close()
|
|
|
|
self.assertFalse(server.connected)
|
|
|
|
self.assertFalse(server.accepting)
|
|
|
|
|
|
|
|
def test_create_socket(self):
|
|
|
|
s = asyncore.dispatcher()
|
|
|
|
s.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
self.assertEqual(s.socket.family, socket.AF_INET)
|
2012-10-18 08:44:43 -03:00
|
|
|
self.assertEqual(s.socket.type, socket.SOCK_STREAM)
|
2010-05-10 12:33:22 -03:00
|
|
|
|
|
|
|
def test_bind(self):
|
|
|
|
s1 = asyncore.dispatcher()
|
|
|
|
s1.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
s1.bind((HOST, 0))
|
|
|
|
s1.listen(5)
|
|
|
|
port = s1.socket.getsockname()[1]
|
|
|
|
|
|
|
|
s2 = asyncore.dispatcher()
|
|
|
|
s2.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
# EADDRINUSE indicates the socket was correctly bound
|
|
|
|
self.assertRaises(socket.error, s2.bind, (HOST, port))
|
|
|
|
|
|
|
|
def test_set_reuse_addr(self):
|
|
|
|
sock = socket.socket()
|
|
|
|
try:
|
|
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|
|
|
except socket.error:
|
|
|
|
unittest.skip("SO_REUSEADDR not supported on this platform")
|
|
|
|
else:
|
|
|
|
# if SO_REUSEADDR succeeded for sock we expect asyncore
|
|
|
|
# to do the same
|
|
|
|
s = asyncore.dispatcher(socket.socket())
|
|
|
|
self.assertFalse(s.socket.getsockopt(socket.SOL_SOCKET,
|
|
|
|
socket.SO_REUSEADDR))
|
|
|
|
s.create_socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
|
|
s.set_reuse_addr()
|
|
|
|
self.assertTrue(s.socket.getsockopt(socket.SOL_SOCKET,
|
|
|
|
socket.SO_REUSEADDR))
|
|
|
|
finally:
|
|
|
|
sock.close()
|
|
|
|
|
2012-03-22 12:06:43 -03:00
|
|
|
@unittest.skipUnless(threading, 'Threading required for this test.')
|
|
|
|
@test_support.reap_threads
|
|
|
|
def test_quick_connect(self):
|
|
|
|
# see: http://bugs.python.org/issue10340
|
|
|
|
server = TCPServer()
|
|
|
|
t = threading.Thread(target=lambda: asyncore.loop(timeout=0.1, count=500))
|
|
|
|
t.start()
|
2013-01-13 08:55:13 -04:00
|
|
|
self.addCleanup(t.join)
|
2012-03-22 12:06:43 -03:00
|
|
|
|
|
|
|
for x in xrange(20):
|
|
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
2012-03-23 11:04:27 -03:00
|
|
|
s.settimeout(.2)
|
2012-03-22 12:06:43 -03:00
|
|
|
s.setsockopt(socket.SOL_SOCKET, socket.SO_LINGER,
|
|
|
|
struct.pack('ii', 1, 0))
|
2012-03-23 11:04:27 -03:00
|
|
|
try:
|
|
|
|
s.connect(server.address)
|
|
|
|
except socket.error:
|
|
|
|
pass
|
|
|
|
finally:
|
|
|
|
s.close()
|
2012-03-22 12:06:43 -03:00
|
|
|
|
2010-05-10 12:33:22 -03:00
|
|
|
|
|
|
|
class TestAPI_UseSelect(BaseTestAPI):
|
|
|
|
use_poll = False
|
|
|
|
|
2010-07-27 20:45:05 -03:00
|
|
|
@unittest.skipUnless(hasattr(select, 'poll'), 'select.poll required')
|
2010-05-10 12:33:22 -03:00
|
|
|
class TestAPI_UsePoll(BaseTestAPI):
|
|
|
|
use_poll = True
|
|
|
|
|
|
|
|
|
2008-06-13 19:38:33 -03:00
|
|
|
def test_main():
|
|
|
|
tests = [HelperFunctionTests, DispatcherTests, DispatcherWithSendTests,
|
2010-07-27 20:45:05 -03:00
|
|
|
DispatcherWithSendTests_UsePoll, TestAPI_UseSelect,
|
|
|
|
TestAPI_UsePoll, FileWrapperTest]
|
2008-06-13 19:38:33 -03:00
|
|
|
run_unittest(*tests)
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
test_main()
|