X-Git-Url: https://git.teslayout.com/public/public/public/?p=anna.git;a=blobdiff_plain;f=source%2Ftesting%2FTestManager.cpp;h=5f0f6f672916ccce3ad4e6e2dd40163ca1b40ecf;hp=6aeae9f9d16cc8b5210ff62cd862ac1f610d9371;hb=303d3e2584497a05fc715fbf32b57dcf81805f38;hpb=26ae04c7e1eba19d02823bcc87c0ee9f6ce76186 diff --git a/source/testing/TestManager.cpp b/source/testing/TestManager.cpp index 6aeae9f..5f0f6f6 100644 --- a/source/testing/TestManager.cpp +++ b/source/testing/TestManager.cpp @@ -104,33 +104,33 @@ TestManager::TestManager() : a_currentTestIt = a_testPool.end(); } -void TestManager::registerSessionId(const std::string &sessionId, const TestCase *testCase) throw(anna::RuntimeException) { +void TestManager::registerKey1(const std::string &key, const TestCase *testCase) throw(anna::RuntimeException) { - std::map::const_iterator it = a_sessionIdTestCaseMap.find(sessionId); - if (it != a_sessionIdTestCaseMap.end()) { // found + auto it = a_key1TestCaseMap.find(key); + if (it != a_key1TestCaseMap.end()) { // found unsigned int id = it->second->getId(); if (id != testCase->getId()) { - throw anna::RuntimeException(anna::functions::asString("There is another test case (id = %llu) which registered such sessionId: %s", id, sessionId.c_str()), ANNA_FILE_LOCATION); + throw anna::RuntimeException(anna::functions::asString("There is another test case (id = %llu) which registered such key1: %s", id, key.c_str()), ANNA_FILE_LOCATION); } } else { - a_sessionIdTestCaseMap[sessionId] = const_cast(testCase); - LOGDEBUG(anna::Logger::debug(anna::functions::asString("TestManager::registerSessionId for test case (id = %llu): %s)", testCase->getId(), sessionId.c_str()), ANNA_FILE_LOCATION)); + a_key1TestCaseMap[key] = const_cast(testCase); + LOGDEBUG(anna::Logger::debug(anna::functions::asString("TestManager::registerKey1 for test case (id = %llu): %s)", testCase->getId(), key.c_str()), ANNA_FILE_LOCATION)); } } -void TestManager::registerSubscriberId(const std::string &subscriberId, const TestCase *testCase) throw(anna::RuntimeException) { +void TestManager::registerKey2(const std::string &key, const TestCase *testCase) throw(anna::RuntimeException) { - std::map::const_iterator it = a_subscriberIdTestCaseMap.find(subscriberId); - if (it != a_subscriberIdTestCaseMap.end()) { // found + auto it = a_key2TestCaseMap.find(key); + if (it != a_key2TestCaseMap.end()) { // found unsigned int id = it->second->getId(); if (id != testCase->getId()) { - throw anna::RuntimeException(anna::functions::asString("There is another test case (id = %llu) which registered such subscriberId: %s", id, subscriberId.c_str()), ANNA_FILE_LOCATION); + throw anna::RuntimeException(anna::functions::asString("There is another test case (id = %llu) which registered such key2: %s", id, key.c_str()), ANNA_FILE_LOCATION); } } else { - a_subscriberIdTestCaseMap[subscriberId] = const_cast(testCase); - LOGDEBUG(anna::Logger::debug(anna::functions::asString("TestManager::registerSubscriberId for test case (id = %llu): %s)", testCase->getId(), subscriberId.c_str()), ANNA_FILE_LOCATION)); + a_key2TestCaseMap[key] = const_cast(testCase); + LOGDEBUG(anna::Logger::debug(anna::functions::asString("TestManager::registerKey2 for test case (id = %llu): %s)", testCase->getId(), key.c_str()), ANNA_FILE_LOCATION)); } } @@ -282,8 +282,8 @@ bool TestManager::clearPool() throw() { // TODO: stop the possible command threads or there will be a core dump a_testPool.clear(); - a_sessionIdTestCaseMap.clear(); - a_subscriberIdTestCaseMap.clear(); + a_key1TestCaseMap.clear(); + a_key2TestCaseMap.clear(); a_currentTestIt = a_testPool.end(); a_poolCycle = 1; configureTTPS(0); // stop @@ -299,8 +299,8 @@ bool TestManager::resetPool(bool hard) throw() { if (it->second->reset(hard)) result = true; } - //a_sessionIdTestCaseMap.clear(); - //a_subscriberIdTestCaseMap.clear(); + //a_key1TestCaseMap.clear(); + //a_key2TestCaseMap.clear(); return result; } @@ -375,7 +375,7 @@ bool TestManager::nextTestCase() throw() { } } -TestCase *TestManager::getTestCaseFromSessionId(const anna::DataBlock &message, std::string &sessionId) throw() { +TestCase *TestManager::getDiameterTestCaseFromSessionId(const anna::DataBlock &message, std::string &sessionId) throw() { try { sessionId = anna::diameter::helpers::base::functions::getSessionId(message); } @@ -384,15 +384,15 @@ TestCase *TestManager::getTestCaseFromSessionId(const anna::DataBlock &message, LOGDEBUG(anna::Logger::debug("Cannot get the Session-Id from received DataBlock in order to identify the Test Case", ANNA_FILE_LOCATION)); return NULL; } - std::map::const_iterator sessionIdIt = a_sessionIdTestCaseMap.find(sessionId); - if (sessionIdIt != a_sessionIdTestCaseMap.end()) + auto sessionIdIt = a_key1TestCaseMap.find(sessionId); + if (sessionIdIt != a_key1TestCaseMap.end()) return sessionIdIt->second; LOGDEBUG(anna::Logger::debug(anna::functions::asString("Cannot identify the Test Case for received Session-Id: %s", sessionId.c_str()), ANNA_FILE_LOCATION)); return NULL; } -TestCase *TestManager::getTestCaseFromSubscriberId(const anna::DataBlock &message, std::string &subscriberId) throw() { +TestCase *TestManager::getDiameterTestCaseFromSubscriberId(const anna::DataBlock &message, std::string &subscriberId) throw() { try { subscriberId = anna::diameter::helpers::dcca::functions::getSubscriptionIdData(message, anna::diameter::helpers::dcca::AVPVALUES__Subscription_Id_Type::END_USER_E164); if (subscriberId == "") // try with IMSI @@ -403,15 +403,15 @@ TestCase *TestManager::getTestCaseFromSubscriberId(const anna::DataBlock &messag LOGDEBUG(anna::Logger::debug("Cannot get the Subscriber-Id from received DataBlock in order to identify the Test Case", ANNA_FILE_LOCATION)); return NULL; } - std::map::const_iterator subscriberIdIt = a_subscriberIdTestCaseMap.find(subscriberId); - if (subscriberIdIt != a_subscriberIdTestCaseMap.end()) + auto subscriberIdIt = a_key2TestCaseMap.find(subscriberId); + if (subscriberIdIt != a_key2TestCaseMap.end()) return subscriberIdIt->second; LOGDEBUG(anna::Logger::debug(anna::functions::asString("Cannot identify the Test Case for received Subscriber-Id: %s", subscriberId.c_str()), ANNA_FILE_LOCATION)); return NULL; } -void TestManager::receiveMessage(const anna::DataBlock &message, const anna::diameter::comm::ClientSession *clientSession) throw(anna::RuntimeException) { +void TestManager::receiveDiameterMessage(const anna::DataBlock &message, const anna::diameter::comm::ClientSession *clientSession) throw(anna::RuntimeException) { // Testing disabled: if (!tests()) return; @@ -419,9 +419,9 @@ void TestManager::receiveMessage(const anna::DataBlock &message, const anna::dia // Identify the test case: std::string sessionId, subscriberId; TestCase *tc; - tc = getTestCaseFromSessionId(message, sessionId); + tc = getDiameterTestCaseFromSessionId(message, sessionId); if (!tc) - tc = getTestCaseFromSubscriberId(message, subscriberId); + tc = getDiameterTestCaseFromSubscriberId(message, subscriberId); if (!tc) { LOGWARNING(anna::Logger::warning(anna::comm::functions::asText("Cannot identify the Test Case for the message received from server: ", message), ANNA_FILE_LOCATION)); // this should not appear return; @@ -452,7 +452,7 @@ void TestManager::receiveMessage(const anna::DataBlock &message, const anna::dia } } -void TestManager::receiveMessage(const anna::DataBlock &message, const anna::diameter::comm::ServerSession *serverSession) throw(anna::RuntimeException) { +void TestManager::receiveDiameterMessage(const anna::DataBlock &message, const anna::diameter::comm::ServerSession *serverSession) throw(anna::RuntimeException) { // Testing disabled: if (!tests()) return; @@ -460,9 +460,9 @@ void TestManager::receiveMessage(const anna::DataBlock &message, const anna::dia // Identify the test case: std::string sessionId, subscriberId; TestCase *tc; - tc = getTestCaseFromSessionId(message, sessionId); + tc = getDiameterTestCaseFromSessionId(message, sessionId); if (!tc) - tc = getTestCaseFromSubscriberId(message, subscriberId); + tc = getDiameterTestCaseFromSubscriberId(message, subscriberId); if (!tc) { LOGWARNING(anna::Logger::warning(anna::comm::functions::asText("Cannot identify the Test Case for the message received from client: ", message), ANNA_FILE_LOCATION)); // this should not appear return;