System test feature

[anna.git] / example / diameter / launcher / MyLocalServer.hpp

diff --git a/example/diameter/launcher/MyLocalServer.hpp b/example/diameter/launcher/MyLocalServer.hpp
index 65b74e2..da4ed97 100644 (file)
--- a/example/diameter/launcher/MyLocalServer.hpp
+++ b/example/diameter/launcher/MyLocalServer.hpp
@@ -13,11 +13,20 @@
 #include <anna/diameter.comm/LocalServer.hpp>
 
 // Process
-#include "ProgrammedAnswers.hpp"
+#include <ProgrammedAnswers.hpp>
 
+namespace anna {
+  namespace diameter {
+    namespace codec {
+      class Engine;
+    }
+  }
+}
 
 class MyLocalServer : public anna::diameter::comm::LocalServer {
 
+  anna::diameter::codec::Engine * a_codecEngine; // for automatic answers (failed-avp), write logs, etc.
+
   void eventResponse(const anna::diameter::comm::Response&) throw(anna::RuntimeException);
   void eventRequest(anna::diameter::comm::ServerSession *, const anna::DataBlock&) throw(anna::RuntimeException);
   void eventUnknownResponse(anna::diameter::comm::ServerSession *, const anna::DataBlock&) throw(anna::RuntimeException);
@@ -26,6 +35,7 @@ class MyLocalServer : public anna::diameter::comm::LocalServer {
 public:
 
   ProgrammedAnswers a_reactingAnswers;
+  void setCodecEngine(anna::diameter::codec::Engine *codecEngine) throw() { a_codecEngine = codecEngine; a_reactingAnswers.setCodecEngine(codecEngine); }
   ProgrammedAnswers *getReactingAnswers() throw() { return (ProgrammedAnswers *)&a_reactingAnswers; }
 };