X-Git-Url: https://git.teslayout.com/public/public/public/?a=blobdiff_plain;f=example%2Fdiameter%2Flauncher%2FMyDiameterEntity.cpp;h=6c1244b55710a92cbb8583ad595ce1dac5fcdad4;hb=603b76b5dc02acdfd87c58c51b6d48ae1858f58d;hp=50a96636f99077273348114b3e426c7a89a54ab9;hpb=4c3f0a4d7e4db76996404d80c6f939548fca656f;p=anna.git

diff --git a/example/diameter/launcher/MyDiameterEntity.cpp b/example/diameter/launcher/MyDiameterEntity.cpp
index 50a9663..6c1244b 100644
--- a/example/diameter/launcher/MyDiameterEntity.cpp
+++ b/example/diameter/launcher/MyDiameterEntity.cpp
@@ -101,9 +101,6 @@ throw(anna::RuntimeException) {
   Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
   RealmNode * my_node = my_app.getRealmNode(getEngine()->getRealm());
 
-  // Testing:
-  TestManager::instantiate().receiveMessage(message, clientSession);
-
   // CommandId:
   anna::diameter::CommandId cid = anna::diameter::codec::functions::getCommandId(message);
   LOGDEBUG
@@ -177,6 +174,9 @@ throw(anna::RuntimeException) {
       ex.trace();
     }
   }
+
+  // Testing:
+  TestManager::instantiate().receiveMessage(message, my_node, clientSession);
 }
 
 void MyDiameterEntity::eventResponse(const anna::diameter::comm::Response &response)
@@ -195,9 +195,6 @@ throw(anna::RuntimeException) {
   bool isUnavailable = (result == anna::diameter::comm::Response::ResultCode::DiameterUnavailable);
   bool isOK = (result == anna::diameter::comm::Response::ResultCode::Success);
 
-  // Testing:
-  TestManager::instantiate().receiveMessage(*message, clientSession);
-
   // CommandId:
   anna::diameter::CommandId request_cid = request->getCommandId();
   LOGDEBUG
@@ -272,6 +269,9 @@ throw(anna::RuntimeException) {
 
   // Triggering burst:
   if(isOK || contextExpired) my_node->sendBurstMessage();
+
+  // Testing:
+  TestManager::instantiate().receiveMessage(*message, my_node, clientSession);
 }
 
 void MyDiameterEntity::eventUnknownResponse(anna::diameter::comm::ClientSession *clientSession, const anna::DataBlock &message)