Dynamic lib selection and deployment
[anna.git] / example / diameter / launcher / testing / TestManager.cpp
index 6dfec1e..cf768ab 100644 (file)
 #include <anna/app/functions.hpp>
 #include <anna/timex/Engine.hpp>
 #include <anna/diameter/helpers/base/functions.hpp>
+#include <anna/diameter/helpers/dcca/functions.hpp>
 #include <anna/diameter.comm/ClientSession.hpp>
 #include <anna/diameter.comm/ServerSession.hpp>
 
 // Process
 #include <TestManager.hpp>
-#include <TestCase.hpp>
 #include <TestClock.hpp>
 #include <Launcher.hpp>
-#include <RealmNode.hpp>
+#include <OriginHost.hpp>
 
 
 class TestTimer;
 
 
+
+
+///////////////////////////////////////////////////////////////////////////////////////////////////
+void TestManager::StatSummary::newTCState(const TestCase::State::_v beginState, const TestCase::State::_v endState) throw() {
+
+  if ((beginState == TestCase::State::Initialized)&&(endState == TestCase::State::Initialized)) { // special case (new test case provisioning)
+    a_initializedTcs++;
+    return;
+  }
+
+  switch (beginState) {
+  case TestCase::State::Initialized: a_initializedTcs--; break;
+  case TestCase::State::InProgress: a_inprogressTcs--; break;
+  case TestCase::State::Failed: a_failedTcs--; break;
+  case TestCase::State::Success: a_sucessTcs--; break;
+  default: break;
+  }
+  switch (endState) {
+  case TestCase::State::Initialized: a_initializedTcs++; break;
+  case TestCase::State::InProgress: a_inprogressTcs++; break;
+  case TestCase::State::Failed: a_failedTcs++; break;
+  case TestCase::State::Success: a_sucessTcs++; break;
+  default: break;
+  }
+}
+
+void TestManager::StatSummary::clear() throw() {
+  a_initializedTcs = 0;
+  a_inprogressTcs = 0;
+  a_failedTcs = 0;
+  a_sucessTcs = 0;
+}
+
+anna::xml::Node *TestManager::StatSummary::asXML(anna::xml::Node* parent) const throw() {
+  anna::xml::Node* result = parent->createChild("StatSummary");
+
+  anna::xml::Node* tcs = result->createChild("TestCasesCounts");
+  tcs->createAttribute("Total", a_initializedTcs + a_inprogressTcs + a_failedTcs + a_sucessTcs);
+  tcs->createAttribute("Initialized", a_initializedTcs);
+  tcs->createAttribute("InProgress", a_inprogressTcs);
+  tcs->createAttribute("Failed", a_failedTcs);
+  tcs->createAttribute("Success", a_sucessTcs);
+
+  return result;
+}
+///////////////////////////////////////////////////////////////////////////////////////////////////
+
+
+
 TestManager::TestManager() :
-  anna::timex::TimeEventObserver("TestManager") {
+      anna::timex::TimeEventObserver("TestManager") {
   a_timeController = NULL;
   a_reportsDirectory = "./";
-  a_dumpReports = false;
+
+  a_dumpInProgressReports = false;
+  a_dumpInitializedReports = false;
+  a_dumpFailedReports = false;
+  a_dumpSuccessReports = false;
+
   a_dumpHexMessages = false;
   a_synchronousAmount = 1;
   a_poolRepeats = 0; // repeat disabled by default
   a_poolCycle = 1;
-  a_inProgressCount = 0;
   a_inProgressLimit = UINT_MAX; // no limit
   a_clock = NULL;
   //a_testPool.clear();
+  //a_statSummary.clear();
   a_currentTestIt = a_testPool.end();
 }
 
@@ -57,6 +111,22 @@ void TestManager::registerSessionId(const std::string &sessionId, const TestCase
   }
   else {
     a_sessionIdTestCaseMap[sessionId] = const_cast<TestCase*>(testCase);
+    LOGDEBUG(anna::Logger::debug(anna::functions::asString("TestManager::registerSessionId for test case (id = %llu): %s)", testCase->getId(), sessionId.c_str()), ANNA_FILE_LOCATION));
+  }
+}
+
+void TestManager::registerSubscriberId(const std::string &subscriberId, const TestCase *testCase)  throw(anna::RuntimeException) {
+
+  std::map<std::string /* subscriber id's */, TestCase*>::const_iterator it = a_subscriberIdTestCaseMap.find(subscriberId);
+  if (it != a_subscriberIdTestCaseMap.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);
+    }
+  }
+  else {
+    a_subscriberIdTestCaseMap[subscriberId] = const_cast<TestCase*>(testCase);
+    LOGDEBUG(anna::Logger::debug(anna::functions::asString("TestManager::registerSubscriberId for test case (id = %llu): %s)", testCase->getId(), subscriberId.c_str()), ANNA_FILE_LOCATION));
   }
 }
 
@@ -76,9 +146,9 @@ throw(anna::RuntimeException) {
   result->setTimeout(timeout);
 
   LOGDEBUG(
-    std::string msg("TestManager::createTimer | ");
-    msg += result->asString();
-    anna::Logger::debug(msg, ANNA_FILE_LOCATION);
+      std::string msg("TestManager::createTimer | ");
+  msg += result->asString();
+  anna::Logger::debug(msg, ANNA_FILE_LOCATION);
   );
 
   a_timeController->activate(result);
@@ -91,9 +161,9 @@ throw() {
     return;
 
   LOGDEBUG(
-    std::string msg("TestManager::cancel | ");
-    msg += timer->asString();
-    anna::Logger::debug(msg, ANNA_FILE_LOCATION);
+      std::string msg("TestManager::cancel | ");
+  msg += timer->asString();
+  anna::Logger::debug(msg, ANNA_FILE_LOCATION);
   );
 
   try {
@@ -153,15 +223,15 @@ bool TestManager::configureTTPS(int testTicksPerSecond) throw() {
 
   if (a_synchronousAmount > 1) {
     LOGWARNING(
-      std::string msg = anna::functions::asString("Desired testing time trigger rate (%d events per second) requires more than one sending per event (%d every %lld milliseconds). Consider launch more instances with lower rate (for example %d ADML processes with %d ttps), or configure %d or more sockets to the remote endpoints to avoid burst sendings",
-                        testTicksPerSecond,
-                        a_synchronousAmount,
-                        admlTimeInterval.getValue(),
-                        a_synchronousAmount,
-                        1000/admlTimeInterval,
-                        a_synchronousAmount);
-
-      anna::Logger::warning(msg, ANNA_FILE_LOCATION);
+        std::string msg = anna::functions::asString("Desired testing time trigger rate (%d events per second) requires more than one sending per event (%d every %lld milliseconds). Consider launch more instances with lower rate (for example %d ADML processes with %d ttps), or configure %d or more sockets to the remote endpoints to avoid burst sendings",
+            testTicksPerSecond,
+            a_synchronousAmount,
+            admlTimeInterval.getValue(),
+            a_synchronousAmount,
+            1000/admlTimeInterval,
+            a_synchronousAmount);
+
+    anna::Logger::warning(msg, ANNA_FILE_LOCATION);
     );
   }
 
@@ -208,11 +278,15 @@ TestCase *TestManager::getTestCase(unsigned int id) throw() {
 bool TestManager::clearPool() throw() {
   if (!tests()) return false;
   for (test_pool_it it = a_testPool.begin(); it != a_testPool.end(); it++) delete it->second;
+  // TODO: stop the possible command threads or there will be a core dump
+
   a_testPool.clear();
   a_sessionIdTestCaseMap.clear();
+  a_subscriberIdTestCaseMap.clear();
   a_currentTestIt = a_testPool.end();
   a_poolCycle = 1;
   configureTTPS(0); // stop
+  a_statSummary.clear();
   return true;
 }
 
@@ -225,19 +299,24 @@ bool TestManager::resetPool(bool hard) throw() {
       result = true;
   }
   //a_sessionIdTestCaseMap.clear();
+  //a_subscriberIdTestCaseMap.clear();
   return result;
 }
 
 bool TestManager::tick() throw() {
   LOGDEBUG(anna::Logger::debug("New test clock tick !", ANNA_FILE_LOCATION));
+  return execTestCases(a_synchronousAmount);
+}
+
+bool TestManager::execTestCases(int sync_amount) throw() {
 
   if (!tests()) {
-    LOGWARNING(anna::Logger::warning("Testing pool is empty. You need programming. Stopping test clock", ANNA_FILE_LOCATION));
+    LOGWARNING(anna::Logger::warning("Testing pool is empty. You need programming", ANNA_FILE_LOCATION));
     return false;
   }
 
   // Synchronous sendings per tick:
-  int count = a_synchronousAmount;
+  int count = sync_amount;
   while (count > 0) {
     if (!nextTestCase()) return false; // stop the clock
     count--;
@@ -251,7 +330,7 @@ bool TestManager::nextTestCase() throw() {
   while (true) {
 
     // Limit for in-progress test cases:
-    if (a_inProgressCount >= a_inProgressLimit) {
+    if (getInProgressCount() >= a_inProgressLimit) {
       LOGDEBUG(anna::Logger::debug(anna::functions::asString("TestManager next case ignored (over in-progress count limit: %llu)", a_inProgressLimit), ANNA_FILE_LOCATION));
       return true; // wait next tick to release OTA test cases
     }
@@ -271,13 +350,13 @@ bool TestManager::nextTestCase() throw() {
       }
       else {
         LOGWARNING(
-          std::string nolimit = (a_poolRepeats != -1) ? "":" [no limit]";
-          anna::Logger::warning(anna::functions::asString("Testing pool cycle %d completed (repeats configured: %d%s). Restarting for the %s cycle", a_poolCycle, a_poolRepeats, nolimit.c_str(), (a_poolRepeats == a_poolCycle) ? "last":"next"), ANNA_FILE_LOCATION);
+            std::string nolimit = (a_poolRepeats != -1) ? "":" [no limit]";
+        anna::Logger::warning(anna::functions::asString("Testing pool cycle %d completed (repeats configured: %d%s). Restarting for the %s cycle", a_poolCycle, a_poolRepeats, nolimit.c_str(), (a_poolRepeats == a_poolCycle) ? "last":"next"), ANNA_FILE_LOCATION);
         );
         a_poolCycle++;
         //a_currentTestIt = a_testPool.begin();
         return true; // avoids infinite loop: if the cycle takes less time than test cases completion, below reset never will turns state
-                     // into Initialized and this while will be infinite. It is preferable to wait one tick when the cycle is completed.
+        // into Initialized and this while will be infinite. It is preferable to wait one tick when the cycle is completed.
       }
     }
 
@@ -289,8 +368,8 @@ bool TestManager::nextTestCase() throw() {
     if (a_currentTestIt->second->getState() != TestCase::State::InProgress) {
       a_currentTestIt->second->process();
       return true; // is not probably to reach still In-Progress test cases from previous cycles due to the whole
-                   //  time for complete the test cases pool regarding the single test case lifetime. You shouldn't
-                   //  forget to programm a test case timeout with a reasonable value
+      //  time for complete the test cases pool regarding the single test case lifetime. You shouldn't
+      //  forget to programm a test case timeout with a reasonable value
     }
   }
 }
@@ -301,26 +380,51 @@ TestCase *TestManager::getTestCaseFromSessionId(const anna::DataBlock &message,
   }
   catch (anna::RuntimeException &ex) {
     //ex.trace();
-    LOGWARNING(anna::Logger::warning("Cannot get the Session-Id from received DataBlock in order to identify the Test Case", ANNA_FILE_LOCATION));
+    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<std::string /* session id's */, TestCase*>::const_iterator sessionIdIt = a_sessionIdTestCaseMap.find(sessionId);
   if (sessionIdIt != a_sessionIdTestCaseMap.end())
     return sessionIdIt->second;
 
-  LOGWARNING(anna::Logger::warning(anna::functions::asString("Cannot identify the Test Case for received Session-Id: %s", sessionId.c_str()), ANNA_FILE_LOCATION));
+  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;
 }
 
-void TestManager::receiveMessage(const anna::DataBlock &message, RealmNode *realm, const anna::diameter::comm::ClientSession *clientSession) throw(anna::RuntimeException) {
+TestCase *TestManager::getTestCaseFromSubscriberId(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
+      subscriberId = anna::diameter::helpers::dcca::functions::getSubscriptionIdData(message, anna::diameter::helpers::dcca::AVPVALUES__Subscription_Id_Type::END_USER_IMSI);
+  }
+  catch (anna::RuntimeException &ex) {
+    //ex.trace();
+    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<std::string /* subscriber id's */, TestCase*>::const_iterator subscriberIdIt = a_subscriberIdTestCaseMap.find(subscriberId);
+  if (subscriberIdIt != a_subscriberIdTestCaseMap.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, OriginHost *host, const anna::diameter::comm::ClientSession *clientSession) throw(anna::RuntimeException) {
 
   // Testing disabled:
   if (!tests()) return;
 
   // Identify the test case:
-  std::string sessionId;
-  TestCase *tc = getTestCaseFromSessionId(message, sessionId);
-  if (!tc) return;
+  std::string sessionId, subscriberId;
+  TestCase *tc;
+  tc = getTestCaseFromSessionId(message, sessionId);
+  if (!tc)
+    tc = getTestCaseFromSubscriberId(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;
+  }
 
   // Work with Test case:
   TestStepWait *tsw = tc->searchNextWaitConditionFulfilled(message, true /* comes from entity */);
@@ -332,10 +436,10 @@ void TestManager::receiveMessage(const anna::DataBlock &message, RealmNode *real
       codecMsg.decode(message);
       hint += "\n"; hint += codecMsg.asXMLString();
 
-      //      // Realm checking:
-      //      std::string messageOR = message.getAvp(anna::diameter::helpers::base::AVPID__Origin_Realm)->getDiameterIdentity()->getValue();
-      //      if (messageOR != realm->getName()) {
-      //        LOGWARNING(anna::Logger::warning(anna::functions::asString("Received message from realm '%s', with different Origin-Realm: %s", realm->getName().c_str(), messageOR.c_str()), ANNA_FILE_LOCATION));
+      //      // Host checking:
+      //      std::string messageOH = message.getAvp(anna::diameter::helpers::base::AVPID__Origin_Host)->getDiameterIdentity()->getValue();
+      //      if (messageOH != host->getName()) {
+      //        LOGWARNING(anna::Logger::warning(anna::functions::asString("Received message from host '%s', with different Origin-Host: %s", host->getName().c_str(), messageOH.c_str()), ANNA_FILE_LOCATION));
       //      }
     }
     catch (anna::RuntimeException &ex) {
@@ -352,15 +456,21 @@ void TestManager::receiveMessage(const anna::DataBlock &message, RealmNode *real
   }
 }
 
-void TestManager::receiveMessage(const anna::DataBlock &message, RealmNode *realm, const anna::diameter::comm::ServerSession *serverSession) throw(anna::RuntimeException) {
+void TestManager::receiveMessage(const anna::DataBlock &message, OriginHost *host, const anna::diameter::comm::ServerSession *serverSession) throw(anna::RuntimeException) {
 
   // Testing disabled:
   if (!tests()) return;
 
   // Identify the test case:
-  std::string sessionId;
-  TestCase *tc = getTestCaseFromSessionId(message, sessionId);
-  if (!tc) return;
+  std::string sessionId, subscriberId;
+  TestCase *tc;
+  tc = getTestCaseFromSessionId(message, sessionId);
+  if (!tc)
+    tc = getTestCaseFromSubscriberId(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;
+  }
 
   // Work with Test case:
   TestStepWait *tsw = tc->searchNextWaitConditionFulfilled(message, false /* comes from client */);
@@ -372,10 +482,10 @@ void TestManager::receiveMessage(const anna::DataBlock &message, RealmNode *real
       codecMsg.decode(message);
       hint += "\n"; hint += codecMsg.asXMLString();
 
-      //      // Realm checking:
-      //      std::string messageOR = message.getAvp(anna::diameter::helpers::base::AVPID__Origin_Realm)->getDiameterIdentity()->getValue();
-      //      if (messageOR != realm->getName()) {
-      //        LOGWARNING(anna::Logger::warning(anna::functions::asString("Received message from realm '%s', with different Origin-Realm: %s", realm->getName().c_str(), messageOR.c_str()), ANNA_FILE_LOCATION));
+      //      // Host checking:
+      //      std::string messageOH = message.getAvp(anna::diameter::helpers::base::AVPID__Origin_Host)->getDiameterIdentity()->getValue();
+      //      if (messageOH != host->getName()) {
+      //        LOGWARNING(anna::Logger::warning(anna::functions::asString("Received message from host '%s', with different Origin-Host: %s", host->getName().c_str(), messageOH.c_str()), ANNA_FILE_LOCATION));
       //      }
     }
     catch (anna::RuntimeException &ex) {
@@ -401,12 +511,15 @@ throw() {
   if (a_poolRepeats) result->createAttribute("PoolRepeats", a_poolRepeats);
   else result->createAttribute("PoolRepeats", "disabled");
   result->createAttribute("PoolCycle", a_poolCycle);
-  result->createAttribute("InProgressCount", a_inProgressCount);
+  a_statSummary.asXML(result);
   if (a_inProgressLimit == UINT_MAX)
     result->createAttribute("InProgressLimit", "<no limit>");
   else
     result->createAttribute("InProgressLimit", a_inProgressLimit);
-  result->createAttribute("DumpReports", (a_dumpReports ? "yes":"no"));
+  result->createAttribute("DumpInitializedReports", (a_dumpInitializedReports ? "yes":"no"));
+  result->createAttribute("DumpInProgressReports", (a_dumpInProgressReports ? "yes":"no"));
+  result->createAttribute("DumpFailedReports", (a_dumpFailedReports ? "yes":"no"));
+  result->createAttribute("DumpSuccessReports", (a_dumpSuccessReports ? "yes":"no"));
   result->createAttribute("DumpHexMessages", (a_dumpHexMessages ? "yes":"no"));
   result->createAttribute("ReportsDirectory", a_reportsDirectory);
   if (a_clock) {
@@ -417,9 +530,13 @@ throw() {
   if (a_currentTestIt != a_testPool.end()) {
     result->createAttribute("CurrentTestCaseId", (*a_currentTestIt).first);
   }
-  if (a_dumpReports && poolSize != 0) {
+  if (poolSize != 0) {
     anna::xml::Node* testCases = result->createChild("TestCases");
     for (test_pool_it it = a_testPool.begin(); it != a_testPool.end(); it++) {
+      if (((*it).second->getState() == TestCase::State::Initialized) && (!getDumpInitializedReports())) continue;
+      if (((*it).second->getState() == TestCase::State::InProgress) && (!getDumpInProgressReports())) continue;
+      if (((*it).second->getState() == TestCase::State::Failed) && (!getDumpFailedReports())) continue;
+      if (((*it).second->getState() == TestCase::State::Success) && (!getDumpSuccessReports())) continue;
       (*it).second->asXML(testCases);
     }
   }