diff --git a/src/common/packet-metadata-test.cc b/src/common/packet-metadata-test.cc index 4e44c4205..e4d0a32b4 100644 --- a/src/common/packet-metadata-test.cc +++ b/src/common/packet-metadata-test.cc @@ -17,8 +17,6 @@ * * Author: Mathieu Lacage */ -#ifdef RUN_SELF_TESTS - #include #include #include @@ -250,20 +248,18 @@ HistoryTrailer::Deserialize (Buffer::Iterator start) namespace ns3 { - - -class PacketMetadataTest : public Test { +class PacketMetadataTest : public TestCase { public: PacketMetadataTest (); virtual ~PacketMetadataTest (); bool CheckHistory (Ptr p, const char *file, int line, uint32_t n, ...); - virtual bool RunTests (void); + virtual bool DoRun (void); private: Ptr DoAddHeader (Ptr p); }; PacketMetadataTest::PacketMetadataTest () - : Test ("PacketMetadata") + : TestCase ("Packet metadata") {} PacketMetadataTest::~PacketMetadataTest () @@ -339,20 +335,21 @@ PacketMetadataTest::CheckHistory (Ptr p, const char *file, int line, uin } return true; error: - Failure () << "PacketMetadata error. file="<< file - << ", line=" << line << ", got:\""; + std::ostringstream failure; + failure << "PacketMetadata error. Got:\""; for (std::list::iterator i = got.begin (); i != got.end (); i++) { - Failure () << *i << ", "; + failure << *i << ", "; } - Failure () << "\", expected: \""; + failure << "\", expected: \""; for (std::list::iterator j = expected.begin (); j != expected.end (); j++) { - Failure () << *j << ", "; + failure << *j << ", "; } - Failure () << "\"" << std::endl; + failure << "\"" << std::endl; + ReportTestFailure ("", "", "", failure.str(), file, line); return false; } @@ -403,7 +400,7 @@ PacketMetadataTest::DoAddHeader (Ptr p) } bool -PacketMetadataTest::RunTests (void) +PacketMetadataTest::DoRun (void) { bool result = true; @@ -722,7 +719,7 @@ PacketMetadataTest::RunTests (void) ADD_HEADER (p1, 20); REM_HEADER (p1, 20); REM_TRAILER (p1, 5); - NS_TEST_ASSERT_EQUAL (p->GetSize (), 1015); + NS_TEST_EXPECT_MSG_EQ (p->GetSize (), 1015, "Correct size"); p = Create (1510); @@ -733,7 +730,7 @@ PacketMetadataTest::RunTests (void) p1 = p->CreateFragment (0, 1500); p2 = p1->Copy (); ADD_HEADER (p2, 24); - NS_TEST_ASSERT_EQUAL (p->GetSize (), 1519); + NS_TEST_EXPECT_MSG_EQ (p->GetSize (), 1519, "Correct size"); p = Create (1000); ADD_HEADER (p, 2); @@ -776,12 +773,20 @@ PacketMetadataTest::RunTests (void) p->RemoveAtStart (10); CHECK_HISTORY (p, 1, 490); + return !result; +} +//----------------------------------------------------------------------------- +class PacketMetadataTestSuite : public TestSuite +{ +public: + PacketMetadataTestSuite (); +}; - return result; +PacketMetadataTestSuite::PacketMetadataTestSuite () + : TestSuite ("packet-metadata", UNIT) +{ + AddTestCase (new PacketMetadataTest); } -static PacketMetadataTest g_packetHistoryTest; - +PacketMetadataTestSuite g_packetMetadataTest; }//namespace ns3 - -#endif /* RUN_SELF_TESTS */