X-Git-Url: https://git.teslayout.com/public/public/public/?a=blobdiff_plain;f=example%2Fdiameter%2Flauncher%2Ftesting%2FTestManager.hpp;h=9e007dffc79350420138049c567ed402ecbc9db0;hb=5bf5055ab79aaf464e2491cfbab173ba37ba3419;hp=2f61501fd579243c205587676cda8beb170086ef;hpb=13b6f0196c2091f9e88a0778155cdcb733cfdafd;p=anna.git

diff --git a/example/diameter/launcher/testing/TestManager.hpp b/example/diameter/launcher/testing/TestManager.hpp
index 2f61501..9e007df 100644
--- a/example/diameter/launcher/testing/TestManager.hpp
+++ b/example/diameter/launcher/testing/TestManager.hpp
@@ -40,7 +40,7 @@ namespace anna {
 class TestClock;
 class TestCase;
 class TestCaseStep;
-class RealmNode;
+class OriginHost;
 
 
 typedef std::map<unsigned int /* test case id */, TestCase*> test_pool_t;
@@ -76,7 +76,7 @@ class TestManager : public anna::timex::TimeEventObserver, public anna::Singleto
 
   // reports
   std::string a_reportsDirectory;
-  bool a_dumpReports;
+  bool a_dumpInitializedReports, a_dumpInProgressReports, a_dumpFailedReports, a_dumpSuccessReports;
   bool a_dumpHexMessages;
 
   // Pool of test cases
@@ -90,6 +90,7 @@ class TestManager : public anna::timex::TimeEventObserver, public anna::Singleto
   int a_synchronousAmount;
   TestClock *a_clock;
   bool tick() throw();
+  bool execTestCases(int sync_amount) throw();
   bool nextTestCase() throw();
 
   // Test timers
@@ -122,8 +123,23 @@ class TestManager : public anna::timex::TimeEventObserver, public anna::Singleto
 
     void setDumpHex(bool dh) throw() { a_dumpHexMessages = dh; }
     bool getDumpHex() const throw() { return a_dumpHexMessages; }
-    void setDumpReports(bool dr) throw() { a_dumpReports = dr; }
-    bool getDumpReports() const throw() { return a_dumpReports; }
+
+
+    void setDumpInitializedReports(bool enable) throw() { a_dumpInitializedReports = enable; }
+    void setDumpInProgressReports(bool enable) throw() { a_dumpInProgressReports = enable; }
+    void setDumpFailedReports(bool enable) throw() { a_dumpFailedReports = enable; }
+    void setDumpSuccessReports(bool enable) throw() { a_dumpSuccessReports = enable; }
+    void setDumpAllReports(bool enable) throw() {
+      setDumpInitializedReports(enable);
+      setDumpInProgressReports(enable);
+      setDumpFailedReports(enable);
+      setDumpSuccessReports(enable);
+    }
+
+    bool getDumpInitializedReports() const throw() { return a_dumpInitializedReports; }
+    bool getDumpInProgressReports() const throw() { return a_dumpInProgressReports; }
+    bool getDumpFailedReports() const throw() { return a_dumpFailedReports; }
+    bool getDumpSuccessReports() const throw() { return a_dumpSuccessReports; }
 
     // Helper to calculate time interval and synchronous amount of execution tests to guarantee the input rate (tests per second)
     //  through the time manager which has a minimum resolution of ADML minimum resolution. The first call to this method will
@@ -146,8 +162,8 @@ class TestManager : public anna::timex::TimeEventObserver, public anna::Singleto
 
     // Main logic
     TestCase *getTestCaseFromSessionId(const anna::DataBlock &message, std::string &sessionId) throw();
-    void receiveMessage(const anna::DataBlock &message, RealmNode *realm, const anna::diameter::comm::ClientSession *clientSession) throw(anna::RuntimeException);
-    void receiveMessage(const anna::DataBlock &message, RealmNode *realm, const anna::diameter::comm::ServerSession *serverSession) throw(anna::RuntimeException);
+    void receiveMessage(const anna::DataBlock &message, OriginHost *host, const anna::diameter::comm::ClientSession *clientSession) throw(anna::RuntimeException);
+    void receiveMessage(const anna::DataBlock &message, OriginHost *host, const anna::diameter::comm::ServerSession *serverSession) throw(anna::RuntimeException);
 
     anna::xml::Node* asXML(anna::xml::Node* parent) const throw();
     std::string asXMLString() const throw();
@@ -162,6 +178,7 @@ class TestManager : public anna::timex::TimeEventObserver, public anna::Singleto
   friend class TestStepTimeout; // createTimer
   friend class TestStepDelay; // createTimer
   friend class TestClock; // tick
+  friend class Launcher; // tick
 };
 
 #endif