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")