def test_destinationValidMappings(self):
validMappings = []
validMappings.append({
"prefix": "",
"destination": "/queue/test",
"internaldestination": "/queue/test"
})
validMappings.append({
"prefix": "AABBXX",
"destination": "/queue/test",
"internaldestination": "/queue/AABBXXtest"
})
configDict = {
"Type": "Stomp",
"Username": "******",
"Password": "******",
"ConnectionString": "stomp+ssl://aa:1234",
"skipConnectionCheck": True
}
for x in validMappings:
configDict2 = copy.deepcopy(configDict)
configDict2["DestinationPrefix"] = x["prefix"]
mqClient = mq_client_abstraction.createMQClientInstance(
configDict=configDict2)
self.assertEqual(mqClient._mapToInternalDestination(
x["destination"]),
x["internaldestination"],
msg="Map to destination failed")
self.assertEqual(mqClient._mapFromInternalDestination(
x["internaldestination"]),
x["destination"],
msg="Map from destination failed")
def test_initInvalidDestinationPrefixFails(self):
configDict = {"Type": "Memory", "DestinationPrefix": "fr%$£"}
with self.assertRaises(Exception) as context:
mqClient = mq_client_abstraction.createMQClientInstance(
configDict=configDict)
self.checkGotRightExceptionType(
context, mq_client_abstraction.MqClientExceptionClass)
self.assertEqual(str(context.exception), "Invalid DestinationPrefix")
def getMqClient():
configDict = {
"Type": "Stomp",
"Username": getFromEnvironment("RJMACTIVEMQ_USER"),
"Password": getFromEnvironment("RJMACTIVEMQ_PASS"),
"ConnectionString": getFromEnvironment("RJMACTIVEMQ_CONNSTR")
}
return mq_client_abstraction.createMQClientInstance(configDict=configDict)
def createMqClientInstance(testContext, configDict):
_setupTestContext(testContext=testContext)
mq_client = None
with mock.patch('stomp.Connection',
return_value=testContext["mockConnectionObject"]
) as stompConnection_function:
mq_client = mq_client_abstraction.createMQClientInstance(
configDict=configDict)
return mq_client
def test_initWithMissingPassword(self):
configDict = {"Type": "Stomp", "Username": "******"}
with self.assertRaises(Exception) as context:
mqClient = mq_client_abstraction.createMQClientInstance(
configDict=configDict)
self.checkGotRightExceptionType(
context, mq_client_abstraction.MqClientExceptionClass)
self.assertEqual(str(context.exception),
"Invalid mq client config - Missing Password")
def test_initNoParamFails(self):
configDict = {}
with self.assertRaises(Exception) as context:
mqClient = mq_client_abstraction.createMQClientInstance(
configDict=configDict)
self.checkGotRightExceptionType(
context, mq_client_abstraction.MqClientExceptionClass)
self.assertEqual(str(context.exception),
"Invalid mq client config - Type Missing")
def test_initWithAmazonListConnectionString(self):
configDict = {
"Type": "Stomp",
"Username": "******",
"Password": "******",
"ConnectionString":
"stomp+ssl://b-xx-xx-xx-xx-xx-1.mq.eu-west-2.amazonaws.com:61614",
"skipConnectionCheck": True
}
mqClient = mq_client_abstraction.createMQClientInstance(
configDict=configDict)
def test_initWithMissingPort(self):
configDict = {
"Type": "Stomp",
"Username": "******",
"Password": "******",
"ConnectionString": "stomp+ssl://aaa.mq.xxx.amazonaws.com"
}
with self.assertRaises(Exception) as context:
mqClient = mq_client_abstraction.createMQClientInstance(
configDict=configDict)
self.checkGotRightExceptionType(
context, mq_client_abstraction.MqClientExceptionClass)
self.assertEqual(
str(context.exception),
"Invalid mq client config - Invalid ConnectionString (Missing Port)"
)
def createMqClientInstance(testContext, configDict):
return mq_client_abstraction.createMQClientInstance(configDict=configDict)
def test_initDefaultGivesMemoryStore(self):
configDict = None
mqClient = mq_client_abstraction.createMQClientInstance(
configDict=configDict)
self.assertEqual(mqClient.getType(), "Memory")
