def test_add_client_custom(self):
def fake_class(addr, queue, **kwargs):
self.assertEqual(('testhost', 25), addr)
return 'success'
static = StaticSmtpRelay('testhost', client_class=fake_class)
self.assertEqual('success', static.add_client())
def test_add_remove_client(self):
static = StaticSmtpRelay(None, client_class=FakeClient)
static.queue.put(True)
static._add_client()
for client in static.pool:
client.join()
gevent.sleep(0)
self.assertFalse(static.pool)
def test_add_client(self):
static = StaticSmtpRelay('testhost')
ret = static.add_client()
self.assertIsInstance(ret, SmtpRelayClient)
def test_add_client(self):
static = StaticSmtpRelay('testhost')
ret = static.add_client()
self.assertIsInstance(ret, SmtpRelayClient)
def test_add_client_custom(self):
def fake_class(addr, queue, **kwargs):
self.assertEqual(('testhost', 25), addr)
return 'success'
static = StaticSmtpRelay('testhost', client_class=fake_class)
self.assertEqual('success', static.add_client())
def test_attempt(self):
env = Envelope()
static = StaticSmtpRelay(None, client_class=FakeClient)
ret = static.attempt(env, 0)
self.assertEqual('test', ret)
def _start_relay(self, name, options=None):
if name in self.relays:
return self.relays[name]
if not options:
options = getattr(self.cfg.relay, name)
new_relay = None
if options.type == 'mx':
from slimta.relay.smtp.mx import MxSmtpRelay
from .helpers import fill_hostname_template
kwargs = {}
kwargs['connect_timeout'] = options.get('connect_timeout', 30)
kwargs['command_timeout'] = options.get('command_timeout', 30)
kwargs['data_timeout'] = options.get('data_timeout', 60)
kwargs['idle_timeout'] = options.get('idle_timeout', 10)
kwargs['pool_size'] = options.get('concurrent_connections', 5)
kwargs['ehlo_as'] = fill_hostname_template(options.ehlo_as)
kwargs['context'] = self._get_client_ssl_context(options.tls)
if options.ipv4_only:
kwargs['socket_creator'] = build_ipv4_socket_creator([25])
new_relay = MxSmtpRelay(**kwargs)
if 'force_mx' in options:
for domain, dest in options.force_mx:
new_relay.force_mx(domain, dest)
elif options.type == 'static':
from slimta.relay.smtp.static import StaticSmtpRelay
from .helpers import fill_hostname_template, get_relay_credentials
kwargs = {}
kwargs['host'] = options.host
kwargs['port'] = options.get('port', 25)
kwargs['connect_timeout'] = options.get('connect_timeout', 30)
kwargs['command_timeout'] = options.get('command_timeout', 30)
kwargs['data_timeout'] = options.get('data_timeout', 60)
kwargs['idle_timeout'] = options.get('idle_timeout', 10)
kwargs['pool_size'] = options.get('concurrent_connections', 5)
kwargs['ehlo_as'] = fill_hostname_template(options.ehlo_as)
kwargs['context'] = self._get_client_ssl_context(options.tls)
if 'credentials' in options:
credentials = get_relay_credentials(options.credentials)
kwargs['credentials'] = credentials
if options.ipv4_only:
kwargs['socket_creator'] = \
build_ipv4_socket_creator([kwargs['port']])
new_relay = StaticSmtpRelay(**kwargs)
elif options.type == 'lmtp':
from slimta.relay.smtp.static import StaticLmtpRelay
from .helpers import fill_hostname_template, get_relay_credentials
kwargs = {}
kwargs['host'] = options.get('host', 'localhost')
kwargs['port'] = options.get('port', 24)
kwargs['connect_timeout'] = options.get('connect_timeout', 30)
kwargs['command_timeout'] = options.get('command_timeout', 30)
kwargs['data_timeout'] = options.get('data_timeout', 60)
kwargs['idle_timeout'] = options.get('idle_timeout', 10)
kwargs['pool_size'] = options.get('concurrent_connections', 5)
kwargs['ehlo_as'] = fill_hostname_template(options.ehlo_as)
kwargs['context'] = self._get_client_ssl_context(options.tls)
if 'credentials' in options:
credentials = get_relay_credentials(options.credentials)
kwargs['credentials'] = credentials
if options.ipv4_only:
kwargs['socket_creator'] = \
build_ipv4_socket_creator([kwargs['port']])
new_relay = StaticLmtpRelay(**kwargs)
elif options.type == 'http':
from slimta.relay.http import HttpRelay
from .helpers import fill_hostname_template
kwargs = {}
kwargs['ehlo_as'] = fill_hostname_template(options.ehlo_as)
kwargs['timeout'] = options.get('timeout', 60)
kwargs['idle_timeout'] = options.get('idle_timeout', 10)
kwargs['context'] = self._get_client_ssl_context(options.tls)
new_relay = HttpRelay(options.url, **kwargs)
elif options.type == 'blackhole':
from slimta.relay.blackhole import BlackholeRelay
new_relay = BlackholeRelay()
elif options.type == 'pipe':
from slimta.relay.pipe import PipeRelay
new_relay = PipeRelay(options.args)
elif options.type == 'maildrop':
from slimta.relay.pipe import MaildropRelay
path = options.path
new_relay = MaildropRelay(path)
elif options.type == 'dovecot':
from slimta.relay.pipe import DovecotLdaRelay
path = options.path
new_relay = DovecotLdaRelay(path)
elif options.type == 'custom':
new_relay = custom_factory(options)
else:
msg = 'relay type does not exist: '+options.type
raise ConfigValidationError(msg)
self.relays[name] = new_relay
return new_relay
