From d4799b94e4d99135a153f6304e0696256360353a Mon Sep 17 00:00:00 2001
From: "Hoa V. Dinh" <dinh.viet.hoa@gmail.com>
Date: Wed, 5 Nov 2014 18:24:12 -0800
Subject: Updated unit tests

---
 unittest/unittest.mm | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'unittest/unittest.mm')

diff --git a/unittest/unittest.mm b/unittest/unittest.mm
index fa1798d1..ed5f94d5 100644
--- a/unittest/unittest.mm
+++ b/unittest/unittest.mm
@@ -26,11 +26,11 @@
 
 - (void) prepareForUnitTest
 {
-    if ([[self date] timeIntervalSinceNow] <= 2) {
+    if (fabs([[self date] timeIntervalSinceNow]) <= 2) {
         // Date might be generated, set to known date.
         [self setDate:[NSDate dateWithTimeIntervalSinceReferenceDate:0]];
     }
-    if ([[self receivedDate] timeIntervalSinceNow] <= 2) {
+    if (fabs([[self receivedDate] timeIntervalSinceNow]) <= 2) {
         // Date might be generated, set to known date.
         [self setReceivedDate:[NSDate dateWithTimeIntervalSinceReferenceDate:0]];
     }
@@ -224,6 +224,7 @@
         }
         NSData * data = [NSData dataWithContentsOfFile:path];
         MCOMessageParser * parser = [MCOMessageParser messageParserWithData:data];
+        [[parser header] prepareForUnitTest];
         NSString * str = [parser plainTextRendering];
 
 //        NSString * outputPath = [@"/Users/hoa/mc2-results/summary" stringByAppendingPathComponent:name];
-- 
cgit v1.2.3