From 6a32ee22095daae92f9c392a58cb46c92d95793c Mon Sep 17 00:00:00 2001
From: Borovkova Elena <borovkovaes@iitp.ru>
Date: Mon, 6 Jul 2009 22:58:31 +0400
Subject: [PATCH] RREP_ACK unit test

---
 src/routing/aodv/aodv-packet.cc | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/src/routing/aodv/aodv-packet.cc b/src/routing/aodv/aodv-packet.cc
index f4143e333..ae352dc79 100644
--- a/src/routing/aodv/aodv-packet.cc
+++ b/src/routing/aodv/aodv-packet.cc
@@ -379,5 +379,30 @@ std::ostream & operator<<(std::ostream & os, RrepAckHader const & h)
 	return os;
 }
 
+#ifdef RUN_SELF_TESTS
+/// Unit test for RREP-ACK
+struct RrepAckHeaderTest : public Test
+{
+	RrepAckHeaderTest () : Test ("AODV/RREP-ACK") {}
+  virtual bool RunTests();
+};
+
+/// Test instance
+static RrepAckHeaderTest g_RrepAckHeaderTest;
+
+bool RrepAckHeaderTest::RunTests ()
+{
+  bool result(true);
+
+  RrepAckHader h;
+  Ptr<Packet> p = Create<Packet> ();
+  p->AddHeader (h);
+  RrepAckHader h2;
+  uint32_t bytes = p->RemoveHeader(h2);
+  NS_TEST_ASSERT_EQUAL (bytes, 2);
+  NS_TEST_ASSERT_EQUAL (h, h2);
+  return result;
+}
+#endif
 
 }}