src/routing/olsr/olsr-routing-protocol.h
changeset 5509 33d52e78605a
parent 4616 a84f60b6cd12
child 5524 efed7493f2c1
--- a/src/routing/olsr/olsr-routing-protocol.h	Thu Nov 12 14:08:51 2009 +0100
+++ b/src/routing/olsr/olsr-routing-protocol.h	Fri Nov 13 11:47:02 2009 +0300
@@ -25,6 +25,7 @@
 #define __OLSR_AGENT_IMPL_H__
 
 #include "olsr-header.h"
+#include "ns3/test.h"
 #include "olsr-state.h"
 #include "olsr-repositories.h"
 
@@ -58,11 +59,21 @@
     destAddr (), nextAddr (),
     interface (0), distance (0) {};
 };
+class RoutingProtocol;
+/// Testcase for MPR computation mechanism
+class OlsrMprTestCase : public TestCase {
+public:
+  OlsrMprTestCase ();
+  ~OlsrMprTestCase ();
+  virtual bool DoRun (void);
+  ;
+};
 
 
 class RoutingProtocol : public Ipv4RoutingProtocol
 {
 public:
+  friend class OlsrMprTestCase;
   static TypeId GetTypeId (void);
 
   RoutingProtocol ();