def setUp(self):
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
AMQPServiceConfigInstance = AmqpConfig()
AMQPServiceConfigInstance.reconnectOnConnectionLoss = False
self.amqpBroker = AmqpFactory(AMQPServiceConfigInstance)
yield self.amqpBroker.connect()
yield self.amqpBroker.getChannelReadyDeferred()
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
DLRThrowerConfigInstance = DLRThrowerConfig()
# Lower the timeout config to pass the timeout tests quickly
DLRThrowerConfigInstance.timeout = 2
DLRThrowerConfigInstance.retry_delay = 1
DLRThrowerConfigInstance.max_retries = 2
# Launch the DLRThrower
self.DLRThrower = DLRThrower(DLRThrowerConfigInstance)
# Add the broker to the DLRThrower
yield self.DLRThrower.addAmqpBroker(self.amqpBroker)
class DLRThrowerTestCases(unittest.TestCase):
@defer.inlineCallbacks
def setUp(self):
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
AMQPServiceConfigInstance = AmqpConfig()
AMQPServiceConfigInstance.reconnectOnConnectionLoss = False
self.amqpBroker = AmqpFactory(AMQPServiceConfigInstance)
yield self.amqpBroker.connect()
yield self.amqpBroker.getChannelReadyDeferred()
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
DLRThrowerConfigInstance = DLRThrowerConfig()
# Lower the timeout config to pass the timeout tests quickly
DLRThrowerConfigInstance.timeout = 2
DLRThrowerConfigInstance.retry_delay = 1
DLRThrowerConfigInstance.max_retries = 2
# Launch the DLRThrower
self.DLRThrower = DLRThrower(DLRThrowerConfigInstance)
# Add the broker to the DLRThrower
yield self.DLRThrower.addAmqpBroker(self.amqpBroker)
@defer.inlineCallbacks
def tearDown(self):
yield self.amqpBroker.disconnect()
yield self.DLRThrower.stopService()
class DLRThrowerTestCases(TestCase):
@defer.inlineCallbacks
def setUp(self):
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
AMQPServiceConfigInstance = AmqpConfig()
AMQPServiceConfigInstance.reconnectOnConnectionLoss = False
self.amqpBroker = AmqpFactory(AMQPServiceConfigInstance)
yield self.amqpBroker.connect()
yield self.amqpBroker.getChannelReadyDeferred()
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
DLRThrowerConfigInstance = DLRThrowerConfig()
# Lower the timeout config to pass the timeout tests quickly
DLRThrowerConfigInstance.timeout = 2
DLRThrowerConfigInstance.retry_delay = 1
DLRThrowerConfigInstance.max_retries = 2
# Launch the DLRThrower
self.DLRThrower = DLRThrower(DLRThrowerConfigInstance)
# Add the broker to the DLRThrower
yield self.DLRThrower.addAmqpBroker(self.amqpBroker)
@defer.inlineCallbacks
def tearDown(self):
for q in self.amqpBroker.queues:
yield self.amqpBroker.chan.queue_delete(queue=q)
yield self.amqpBroker.disconnect()
yield self.DLRThrower.stopService()
class SMPPClientManagerPBTestCase(HttpServerTestCase):
@defer.inlineCallbacks
def setUp(self):
HttpServerTestCase.setUp(self)
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
AMQPServiceConfigInstance = AmqpConfig()
AMQPServiceConfigInstance.reconnectOnConnectionLoss = False
# Launch AMQP Broker
self.amqpBroker = AmqpFactory(AMQPServiceConfigInstance)
self.amqpBroker.preConnect()
self.amqpClient = reactor.connectTCP(AMQPServiceConfigInstance.host, AMQPServiceConfigInstance.port, self.amqpBroker)
# Wait for AMQP Broker connection to get ready
yield self.amqpBroker.getChannelReadyDeferred()
# Add the broker to the RouterPB
yield self.pbRoot_f.addAmqpBroker(self.amqpBroker)
# Setup smpp client manager pb
yield self.clientManager_f.addAmqpBroker(self.amqpBroker)
p = portal.Portal(JasminPBRealm(self.clientManager_f))
p.registerChecker(AllowAnonymousAccess())
jPBPortalRoot = JasminPBPortalRoot(p)
self.CManagerServer = reactor.listenTCP(0, pb.PBServerFactory(jPBPortalRoot))
self.CManagerPort = self.CManagerServer.getHost().port
# Start DLRThrower
DLRThrowerConfigInstance = DLRThrowerConfig()
self.DLRThrower = DLRThrower()
self.DLRThrower.setConfig(DLRThrowerConfigInstance)
yield self.DLRThrower.addAmqpBroker(self.amqpBroker)
# Connect to redis server
RedisForJasminConfigInstance = RedisForJasminConfig()
self.redisClient = yield ConnectionWithConfiguration(RedisForJasminConfigInstance)
# Authenticate and select db
if RedisForJasminConfigInstance.password is not None:
yield self.redisClient.auth(RedisForJasminConfigInstance.password)
yield self.redisClient.select(RedisForJasminConfigInstance.dbid)
# Connect CM with RC:
self.clientManager_f.addRedisClient(self.redisClient)
# Set a smpp client manager proxy instance
self.SMPPClientManagerPBProxy = SMPPClientManagerPBProxy()
@defer.inlineCallbacks
def tearDown(self):
yield HttpServerTestCase.tearDown(self)
if self.SMPPClientManagerPBProxy.isConnected:
yield self.SMPPClientManagerPBProxy.disconnect()
yield self.CManagerServer.stopListening()
yield self.amqpClient.disconnect()
yield self.redisClient.disconnect()
def setUp(self):
HttpServerTestCase.setUp(self)
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
AMQPServiceConfigInstance = AmqpConfig()
AMQPServiceConfigInstance.reconnectOnConnectionLoss = False
# Launch AMQP Broker
self.amqpBroker = AmqpFactory(AMQPServiceConfigInstance)
self.amqpBroker.preConnect()
self.amqpClient = reactor.connectTCP(AMQPServiceConfigInstance.host,
AMQPServiceConfigInstance.port,
self.amqpBroker)
# Wait for AMQP Broker connection to get ready
yield self.amqpBroker.getChannelReadyDeferred()
# Add the broker to the RouterPB
yield self.pbRoot_f.addAmqpBroker(self.amqpBroker)
# Setup smpp client manager pb
yield self.clientManager_f.addAmqpBroker(self.amqpBroker)
p = portal.Portal(JasminPBRealm(self.clientManager_f))
p.registerChecker(AllowAnonymousAccess())
jPBPortalRoot = JasminPBPortalRoot(p)
self.CManagerServer = reactor.listenTCP(
0, pb.PBServerFactory(jPBPortalRoot))
self.CManagerPort = self.CManagerServer.getHost().port
# Start DLRThrower
DLRThrowerConfigInstance = DLRThrowerConfig()
self.DLRThrower = DLRThrower()
self.DLRThrower.setConfig(DLRThrowerConfigInstance)
yield self.DLRThrower.addAmqpBroker(self.amqpBroker)
# Connect to redis server
RedisForJasminConfigInstance = RedisForJasminConfig()
self.redisClient = yield ConnectionWithConfiguration(
RedisForJasminConfigInstance)
# Authenticate and select db
if RedisForJasminConfigInstance.password is not None:
yield self.redisClient.auth(RedisForJasminConfigInstance.password)
yield self.redisClient.select(RedisForJasminConfigInstance.dbid)
# Connect CM with RC:
self.clientManager_f.addRedisClient(self.redisClient)
# Set a smpp client manager proxy instance
self.SMPPClientManagerPBProxy = SMPPClientManagerPBProxy()
def startDLRThrowerService(self):
"Start DLRThrower"
DLRThrowerConfigInstance = DLRThrowerConfig(self.options['config'])
self.components['dlr-thrower'] = DLRThrower()
self.components['dlr-thrower'].setConfig(DLRThrowerConfigInstance)
self.components['dlr-thrower'].addSmpps(self.components['smpp-server-factory'])
# AMQP Broker is used to listen to DLRThrower queue
return self.components['dlr-thrower'].addAmqpBroker(self.components['amqp-broker-factory'])
def setUp(self):
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
AMQPServiceConfigInstance = AmqpConfig()
AMQPServiceConfigInstance.reconnectOnConnectionLoss = False
self.amqpBroker = AmqpFactory(AMQPServiceConfigInstance)
yield self.amqpBroker.connect()
yield self.amqpBroker.getChannelReadyDeferred()
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
DLRThrowerConfigInstance = DLRThrowerConfig()
# Lower the timeout config to pass the timeout tests quickly
DLRThrowerConfigInstance.timeout = 2
DLRThrowerConfigInstance.retry_delay = 1
DLRThrowerConfigInstance.max_retries = 2
# Launch the deliverSmHttpThrower
self.DLRThrower = DLRThrower()
self.DLRThrower.setConfig(DLRThrowerConfigInstance)
# Add the broker to the deliverSmHttpThrower
yield self.DLRThrower.addAmqpBroker(self.amqpBroker)
# Start http servers
self.Error404ServerResource = Error404Server()
self.Error404Server = reactor.listenTCP(0, server.Site(self.Error404ServerResource))
self.AckServerResource = AckServer()
self.AckServer = reactor.listenTCP(0, server.Site(self.AckServerResource))
self.NoAckServerResource = NoAckServer()
self.NoAckServer = reactor.listenTCP(0, server.Site(self.NoAckServerResource))
self.TimeoutLeafServerResource = TimeoutLeafServer()
self.TimeoutLeafServerResource.hangTime = 3
self.TimeoutLeafServer = reactor.listenTCP(0, server.Site(self.TimeoutLeafServerResource))
class DLRThrowerTestCase(unittest.TestCase):
@defer.inlineCallbacks
def setUp(self):
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
AMQPServiceConfigInstance = AmqpConfig()
AMQPServiceConfigInstance.reconnectOnConnectionLoss = False
self.amqpBroker = AmqpFactory(AMQPServiceConfigInstance)
yield self.amqpBroker.connect()
yield self.amqpBroker.getChannelReadyDeferred()
# Initiating config objects without any filename
# will lead to setting defaults and that's what we
# need to run the tests
DLRThrowerConfigInstance = DLRThrowerConfig()
# Lower the timeout config to pass the timeout tests quickly
DLRThrowerConfigInstance.timeout = 2
DLRThrowerConfigInstance.retry_delay = 1
DLRThrowerConfigInstance.max_retries = 2
# Launch the deliverSmHttpThrower
self.DLRThrower = DLRThrower()
self.DLRThrower.setConfig(DLRThrowerConfigInstance)
# Add the broker to the deliverSmHttpThrower
yield self.DLRThrower.addAmqpBroker(self.amqpBroker)
# Start http servers
self.Error404ServerResource = Error404Server()
self.Error404Server = reactor.listenTCP(0, server.Site(self.Error404ServerResource))
self.AckServerResource = AckServer()
self.AckServer = reactor.listenTCP(0, server.Site(self.AckServerResource))
self.NoAckServerResource = NoAckServer()
self.NoAckServer = reactor.listenTCP(0, server.Site(self.NoAckServerResource))
self.TimeoutLeafServerResource = TimeoutLeafServer()
self.TimeoutLeafServerResource.hangTime = 3
self.TimeoutLeafServer = reactor.listenTCP(0, server.Site(self.TimeoutLeafServerResource))
@defer.inlineCallbacks
def publishDLRContent(
self,
message_status,
msgid,
dlr_url,
dlr_level,
id_smsc="",
sub="",
dlvrd="",
subdate="",
donedate="",
err="",
text="",
method="POST",
trycount=0,
):
content = DLRContent(
message_status,
msgid,
dlr_url,
dlr_level,
id_smsc,
sub,
dlvrd,
subdate,
donedate,
err,
text,
method,
trycount,
)
yield self.amqpBroker.publish(exchange="messaging", routing_key="dlr_thrower.http", content=content)
@defer.inlineCallbacks
def tearDown(self):
yield self.amqpBroker.disconnect()
yield self.DLRThrower.stopService()
yield self.Error404Server.stopListening()
yield self.AckServer.stopListening()
yield self.NoAckServer.stopListening()
yield self.TimeoutLeafServer.stopListening()
@defer.inlineCallbacks
def test_throwing_http_connector_with_ack(self):
self.AckServerResource.render_POST = mock.Mock(wraps=self.AckServerResource.render_POST)
dlr_url = "http://127.0.0.1:%s/dlr" % self.AckServer.getHost().port
dlr_level = 1
msgid = "anything"
message_status = "DELIVRD"
self.publishDLRContent(message_status, msgid, dlr_url, dlr_level)
# Wait 3 seconds
exitDeferred = defer.Deferred()
reactor.callLater(3, exitDeferred.callback, None)
yield exitDeferred
# No message retries must be made since ACK was received
self.assertEqual(self.AckServerResource.render_POST.call_count, 1)
@defer.inlineCallbacks
def test_throwing_http_connector_without_ack(self):
self.NoAckServerResource.render_POST = mock.Mock(wraps=self.NoAckServerResource.render_POST)
dlr_url = "http://127.0.0.1:%s/dlr" % self.NoAckServer.getHost().port
dlr_level = 1
msgid = "anything"
message_status = "DELIVRD"
self.publishDLRContent(message_status, msgid, dlr_url, dlr_level)
# Wait 3 seconds
exitDeferred = defer.Deferred()
reactor.callLater(3, exitDeferred.callback, None)
yield exitDeferred
# Retries must be made when ACK is not received
self.assertTrue(self.NoAckServerResource.render_POST.call_count > 1)
@defer.inlineCallbacks
def test_throwing_http_connector_timeout_retry(self):
self.TimeoutLeafServerResource.render_POST = mock.Mock(wraps=self.TimeoutLeafServerResource.render_POST)
dlr_url = "http://127.0.0.1:%s/dlr" % self.TimeoutLeafServer.getHost().port
dlr_level = 1
msgid = "anything"
message_status = "DELIVRD"
self.publishDLRContent(message_status, msgid, dlr_url, dlr_level)
# Wait 9 seconds (timeout is set to 2 seconds in deliverSmHttpThrowerTestCase.setUp(self)
exitDeferred = defer.Deferred()
reactor.callLater(12, exitDeferred.callback, None)
yield exitDeferred
self.assertEqual(self.TimeoutLeafServerResource.render_POST.call_count, 3)
@defer.inlineCallbacks
def test_throwing_http_connector_404_error_noretry(self):
"""When receiving a 404 error, no further retries shall be made
"""
self.Error404ServerResource.render_POST = mock.Mock(wraps=self.Error404ServerResource.render_POST)
dlr_url = "http://127.0.0.1:%s/dlr" % self.Error404Server.getHost().port
dlr_level = 1
msgid = "anything"
message_status = "DELIVRD"
self.publishDLRContent(message_status, msgid, dlr_url, dlr_level)
# Wait 3 seconds
exitDeferred = defer.Deferred()
reactor.callLater(3, exitDeferred.callback, None)
yield exitDeferred
self.assertEqual(self.Error404ServerResource.render_POST.call_count, 1)
@defer.inlineCallbacks
def test_throwing_http_connector_dlr_level1(self):
self.AckServerResource.render_GET = mock.Mock(wraps=self.AckServerResource.render_GET)
dlr_url = "http://127.0.0.1:%s/dlr" % self.AckServer.getHost().port
dlr_level = 1
msgid = "anything"
message_status = "DELIVRD"
self.publishDLRContent(message_status, msgid, dlr_url, dlr_level, method="GET")
# Wait 3 seconds
exitDeferred = defer.Deferred()
reactor.callLater(3, exitDeferred.callback, None)
yield exitDeferred
# No message retries must be made since ACK was received
self.assertEqual(self.AckServerResource.render_GET.call_count, 1)
callArgs = self.AckServerResource.render_GET.call_args_list[0][0][0].args
self.assertEqual(callArgs["message_status"][0], message_status)
self.assertEqual(callArgs["id"][0], msgid)
self.assertEqual(callArgs["level"][0], str(dlr_level))
@defer.inlineCallbacks
def test_throwing_http_connector_dlr_level2(self):
self.AckServerResource.render_GET = mock.Mock(wraps=self.AckServerResource.render_GET)
dlr_url = "http://127.0.0.1:%s/dlr" % self.AckServer.getHost().port
dlr_level = 2
msgid = "anything"
message_status = "DELIVRD"
self.publishDLRContent(
message_status,
msgid,
dlr_url,
dlr_level,
id_smsc="abc",
sub="3",
dlvrd="3",
subdate="anydate",
donedate="anydate",
err="",
text="Any text",
method="GET",
)
# Wait 3 seconds
exitDeferred = defer.Deferred()
reactor.callLater(3, exitDeferred.callback, None)
yield exitDeferred
# No message retries must be made since ACK was received
self.assertEqual(self.AckServerResource.render_GET.call_count, 1)
callArgs = self.AckServerResource.render_GET.call_args_list[0][0][0].args
self.assertEqual(callArgs["message_status"][0], message_status)
self.assertEqual(callArgs["id"][0], msgid)
self.assertEqual(callArgs["level"][0], str(dlr_level))
self.assertEqual(callArgs["id_smsc"][0], "abc")
self.assertEqual(callArgs["sub"][0], "3")
self.assertEqual(callArgs["dlvrd"][0], "3")
self.assertEqual(callArgs["subdate"][0], "anydate")
self.assertEqual(callArgs["donedate"][0], "anydate")
self.assertEqual(callArgs["err"][0], "")
self.assertEqual(callArgs["text"][0], "Any text")
