Updated docstrings.
This commit is contained in:
parent
477a6eb4a2
commit
314b92b26e
|
@ -603,6 +603,12 @@ class StreamHandlerTest(BaseTest):
|
||||||
# -- if it proves to be of wider utility than just test_logging
|
# -- if it proves to be of wider utility than just test_logging
|
||||||
|
|
||||||
class TestSMTPChannel(smtpd.SMTPChannel):
|
class TestSMTPChannel(smtpd.SMTPChannel):
|
||||||
|
"""
|
||||||
|
This derived class has had to be created because smtpd does not
|
||||||
|
support use of custom channel maps, although they are allowed by
|
||||||
|
asyncore's design. Issue #11959 has been raised to address this,
|
||||||
|
and if resolved satisfactorily, some of this code can be removed.
|
||||||
|
"""
|
||||||
def __init__(self, server, conn, addr, sockmap):
|
def __init__(self, server, conn, addr, sockmap):
|
||||||
asynchat.async_chat.__init__(self, conn, sockmap)
|
asynchat.async_chat.__init__(self, conn, sockmap)
|
||||||
self.smtp_server = server
|
self.smtp_server = server
|
||||||
|
@ -630,6 +636,25 @@ class TestSMTPChannel(smtpd.SMTPChannel):
|
||||||
|
|
||||||
|
|
||||||
class TestSMTPServer(smtpd.SMTPServer):
|
class TestSMTPServer(smtpd.SMTPServer):
|
||||||
|
"""
|
||||||
|
This class implements a test SMTP server.
|
||||||
|
|
||||||
|
:param addr: A (host, port) tuple which the server listens on.
|
||||||
|
You can specify a port value of zero: the server's
|
||||||
|
*port* attribute will hold the actual port number
|
||||||
|
used, which can be used in client connections.
|
||||||
|
:param handler: A callable which will be called to process
|
||||||
|
incoming messages. The handler will be passed
|
||||||
|
the client address tuple, who the message is from,
|
||||||
|
a list of recipients and the message data.
|
||||||
|
:param poll_interval: The interval, in seconds, used in the underlying
|
||||||
|
:func:`select` or :func:`poll` call by
|
||||||
|
:func:`asyncore.loop`.
|
||||||
|
:param sockmap: A dictionary which will be used to hold
|
||||||
|
:class:`asyncore.dispatcher` instances used by
|
||||||
|
:func:`asyncore.loop`. This avoids changing the
|
||||||
|
:mod:`asyncore` module's global state.
|
||||||
|
"""
|
||||||
channel_class = TestSMTPChannel
|
channel_class = TestSMTPChannel
|
||||||
|
|
||||||
def __init__(self, addr, handler, poll_interval, sockmap):
|
def __init__(self, addr, handler, poll_interval, sockmap):
|
||||||
|
@ -654,22 +679,51 @@ class TestSMTPServer(smtpd.SMTPServer):
|
||||||
self.poll_interval = poll_interval
|
self.poll_interval = poll_interval
|
||||||
|
|
||||||
def handle_accepted(self, conn, addr):
|
def handle_accepted(self, conn, addr):
|
||||||
print('Incoming connection from %s' % repr(addr), file=smtpd.DEBUGSTREAM)
|
"""
|
||||||
|
Redefined only because the base class does not pass in a
|
||||||
|
map, forcing use of a global in :mod:`asyncore`.
|
||||||
|
"""
|
||||||
channel = self.channel_class(self, conn, addr, self.sockmap)
|
channel = self.channel_class(self, conn, addr, self.sockmap)
|
||||||
|
|
||||||
def process_message(self, peer, mailfrom, rcpttos, data):
|
def process_message(self, peer, mailfrom, rcpttos, data):
|
||||||
|
"""
|
||||||
|
Delegates to the handler passed in to the server's constructor.
|
||||||
|
|
||||||
|
Typically, this will be a test case method.
|
||||||
|
:param peer: The client (host, port) tuple.
|
||||||
|
:param mailfrom: The address of the sender.
|
||||||
|
:param rcpttos: The addresses of the recipients.
|
||||||
|
:param data: The message.
|
||||||
|
"""
|
||||||
self._handler(peer, mailfrom, rcpttos, data)
|
self._handler(peer, mailfrom, rcpttos, data)
|
||||||
|
|
||||||
def start(self):
|
def start(self):
|
||||||
|
"""
|
||||||
|
Start the server running on a separate daemon thread.
|
||||||
|
"""
|
||||||
self._thread = t = threading.Thread(target=self.serve_forever,
|
self._thread = t = threading.Thread(target=self.serve_forever,
|
||||||
args=(self.poll_interval,))
|
args=(self.poll_interval,))
|
||||||
t.setDaemon(True)
|
t.setDaemon(True)
|
||||||
t.start()
|
t.start()
|
||||||
|
|
||||||
def serve_forever(self, poll_interval):
|
def serve_forever(self, poll_interval):
|
||||||
|
"""
|
||||||
|
Run the :mod:`asyncore` loop until normal termination
|
||||||
|
conditions arise.
|
||||||
|
:param poll_interval: The interval, in seconds, used in the underlying
|
||||||
|
:func:`select` or :func:`poll` call by
|
||||||
|
:func:`asyncore.loop`.
|
||||||
|
"""
|
||||||
asyncore.loop(poll_interval, map=self.sockmap)
|
asyncore.loop(poll_interval, map=self.sockmap)
|
||||||
|
|
||||||
def stop(self, timeout=None):
|
def stop(self, timeout=None):
|
||||||
|
"""
|
||||||
|
Stop the thread by closing the server instance.
|
||||||
|
Wait for the server thread to terminate.
|
||||||
|
|
||||||
|
:param timeout: How long to wait for the server thread
|
||||||
|
to terminate.
|
||||||
|
"""
|
||||||
self.close()
|
self.close()
|
||||||
self._thread.join(timeout)
|
self._thread.join(timeout)
|
||||||
self._thread = None
|
self._thread = None
|
||||||
|
@ -706,12 +760,19 @@ class ControlMixin(object):
|
||||||
t.start()
|
t.start()
|
||||||
|
|
||||||
def serve_forever(self, poll_interval):
|
def serve_forever(self, poll_interval):
|
||||||
|
"""
|
||||||
|
Run the server. Set the ready flag before entering the
|
||||||
|
service loop.
|
||||||
|
"""
|
||||||
self.ready.set()
|
self.ready.set()
|
||||||
super(ControlMixin, self).serve_forever(poll_interval)
|
super(ControlMixin, self).serve_forever(poll_interval)
|
||||||
|
|
||||||
def stop(self, timeout=None):
|
def stop(self, timeout=None):
|
||||||
"""
|
"""
|
||||||
Tell the server thread to stop, and wait for it to do so.
|
Tell the server thread to stop, and wait for it to do so.
|
||||||
|
|
||||||
|
:param timeout: How long to wait for the server thread
|
||||||
|
to terminate.
|
||||||
"""
|
"""
|
||||||
self.shutdown()
|
self.shutdown()
|
||||||
if self._thread is not None:
|
if self._thread is not None:
|
||||||
|
@ -729,6 +790,7 @@ class TestHTTPServer(ControlMixin, HTTPServer):
|
||||||
single parameter - the request - in order to
|
single parameter - the request - in order to
|
||||||
process the request.
|
process the request.
|
||||||
:param poll_interval: The polling interval in seconds.
|
:param poll_interval: The polling interval in seconds.
|
||||||
|
:param log: Pass ``True`` to enable log messages.
|
||||||
"""
|
"""
|
||||||
def __init__(self, addr, handler, poll_interval=0.5, log=False):
|
def __init__(self, addr, handler, poll_interval=0.5, log=False):
|
||||||
class DelegatingHTTPRequestHandler(BaseHTTPRequestHandler):
|
class DelegatingHTTPRequestHandler(BaseHTTPRequestHandler):
|
||||||
|
|
Loading…
Reference in New Issue