1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
# Copyright (C) 2009 by the Free Software Foundation, Inc.
#
# This file is part of GNU Mailman.
#
# GNU Mailman is free software: you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
#
# GNU Mailman is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# GNU Mailman. If not, see <http://www.gnu.org/licenses/>.
"""Fake MTA for testing purposes."""
from __future__ import absolute_import, unicode_literals
__metaclass__ = type
__all__ = [
'FakeMTA',
]
import logging
from Queue import Queue
from lazr.smtptest.controller import QueueController
from lazr.smtptest.server import Channel, QueueServer
from zope.interface import implements
from mailman.interfaces.mta import IMailTransportAgentAliases
log = logging.getLogger('lazr.smtptest')
class FakeMTA:
"""Fake MTA for testing purposes."""
implements(IMailTransportAgentAliases)
def create(self, mlist):
pass
def delete(self, mlist):
pass
def regenerate(self):
pass
class StatisticsChannel(Channel):
"""A channel that can answers to the fake STAT command."""
def smtp_STAT(self, arg):
"""Cause the server to send statistics to its controller."""
self._server.send_statistics()
self.push('250 Ok')
class ConnectionCountingServer(QueueServer):
"""Count the number of SMTP connections opened."""
def __init__(self, host, port, queue, oob_queue):
"""See `lazr.smtptest.server.QueueServer`."""
QueueServer.__init__(self, host, port, queue)
self._connection_count = 0
# The out-of-band queue is where the server sends statistics to the
# controller upon request.
self._oob_queue = oob_queue
def handle_accept(self):
"""See `lazr.smtp.server.Server`."""
connection, address = self.accept()
self._connection_count += 1
log.info('[ConnectionCountingServer] accepted: %s', address)
StatisticsChannel(self, connection, address)
def reset(self):
"""See `lazr.smtp.server.Server`."""
QueueServer.reset(self)
self._connection_count = 0
def send_statistics(self):
"""Send the current connection statistics to the controller."""
# Do not count the connection caused by the STAT connect.
self._connection_count -= 1
self._oob_queue.put(self._connection_count)
class ConnectionCountingController(QueueController):
"""Count the number of SMTP connections opened."""
def __init__(self, host, port):
"""See `lazr.smtptest.controller.QueueController`."""
self.oob_queue = Queue()
QueueController.__init__(self, host, port)
def _make_server(self, host, port):
"""See `lazr.smtptest.controller.QueueController`."""
self.server = ConnectionCountingServer(
host, port, self.queue, self.oob_queue)
def start(self):
"""See `lazr.smtptest.controller.QueueController`."""
QueueController.start(self)
# Reset the connection statistics, since the base class's start()
# method causes a connection to occur.
self.reset()
def get_connection_count(self):
"""Retrieve the number of connections.
:return: The number of connections to the server that have been made.
:rtype: integer
"""
smtpd = self._connect()
smtpd.docmd('STAT')
# An Empty exception will occur if the data isn't available in 10
# seconds. Let that propagate.
return self.oob_queue.get(block=True, timeout=10)
@property
def messages(self):
"""Return all the messages received by the SMTP server."""
for message in self:
yield message
def clear(self):
"""Clear all the messages from the queue."""
list(self)
|
