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 p = Create (); + 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 }}