src/mesh/test/flame/flame-regression.cc
changeset 7379 ff0d074f885a
parent 7256 b04ba6772f8c
child 8992 4fdf03b24af6
--- a/src/mesh/test/flame/flame-regression.cc	Wed Jul 27 10:51:39 2011 +0100
+++ b/src/mesh/test/flame/flame-regression.cc	Fri Jul 29 03:38:59 2011 -0400
@@ -28,6 +28,7 @@
 #include "ns3/internet-stack-helper.h"
 #include "ns3/ipv4-address-helper.h"
 #include "ns3/abort.h"
+#include "ns3/pcap-test.h"
 #include "ns3/udp-echo-helper.h"
 #include "ns3/mobility-model.h"
 #include <sstream>
@@ -36,8 +37,6 @@
 
 using namespace ns3;
 
-/// Set to true to rewrite reference traces, leave false to run regression test
-const bool WRITE_VECTORS = false;
 /// Unique PCAP file name prefix
 const char * const PREFIX = "flame-regression-test";
 
@@ -65,7 +64,7 @@
   Simulator::Run ();
   Simulator::Destroy ();
 
-  if (!WRITE_VECTORS) CheckResults ();
+  CheckResults ();
 
   delete m_nodes, m_nodes = 0;
 }
@@ -109,8 +108,7 @@
   address.SetBase ("10.1.1.0", "255.255.255.0");
   m_interfaces = address.Assign (meshDevices);
   // 4. write PCAP if needed
-  std::string prefix = (WRITE_VECTORS ? NS_TEST_SOURCEDIR : GetTempDir ()) + PREFIX;
-  wifiPhy.EnablePcapAll (prefix);
+  wifiPhy.EnablePcapAll (CreateTempDirFilename (PREFIX));
 
 }
 void
@@ -134,15 +132,7 @@
 {
   for (int i = 0; i < 3; ++i)
     {
-      std::ostringstream os1, os2;
-      // File naming conventions are hard-coded here.
-      os1 << NS_TEST_SOURCEDIR << PREFIX << "-" << i << "-1.pcap";
-      os2 << GetTempDir () << PREFIX << "-" << i << "-1.pcap";
-
-      uint32_t sec (0), usec (0);
-      bool diff = PcapFile::Diff (os1.str (), os2.str (), sec, usec); // TODO support default PcapWriter snap length here
-      NS_TEST_EXPECT_MSG_EQ (diff, false, "PCAP traces " << os1.str () << " and " << os2.str ()
-                                                         << " differ starting from " << sec << " s " << usec << " us");
+      NS_PCAP_TEST_EXPECT_EQ (PREFIX << "-" << i << "-1.pcap");
     }
 }