Fix record tests

nikhiljindal · Jul 1, 2015 · 0e2be0d · 0e2be0d
1 parent 2a80376
commit 0e2be0d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/client/record/event_test.go b/pkg/client/record/event_test.go
@@ -316,7 +316,7 @@ func validateEvent(actualEvent *api.Event, expectedEvent *api.Event, t *testing.
 	if actualEvent.FirstTimestamp.IsZero() || actualEvent.LastTimestamp.IsZero() {
 		t.Errorf("timestamp wasn't set: %#v", *actualEvent)
 	}
-	if actualEvent.FirstTimestamp.Equal(actualEvent.LastTimestamp.Time) {
+	if actualEvent.FirstTimestamp.Equal(actualEvent.LastTimestamp) {
 		if expectCompression {
 			t.Errorf("FirstTimestamp (%q) and LastTimestamp (%q) must be equal to indicate only one occurance of the event, but were different. Actual Event: %#v", actualEvent.FirstTimestamp, actualEvent.LastTimestamp, *actualEvent)
 		}

diff --git a/pkg/client/record/events_cache_test.go b/pkg/client/record/events_cache_test.go
@@ -173,7 +173,7 @@ func compareEventWithHistoryEntry(expected *api.Event, actual *history, t *testi
 		t.Fatalf("There should be one existing instance of this event in the hash table.")
 	}
 
-	if !actual.FirstTimestamp.Equal(expected.FirstTimestamp.Time) {
+	if !actual.FirstTimestamp.Equal(expected.FirstTimestamp) {
 		t.Fatalf("Unexpected FirstTimestamp. Expected: <%v> Actual: <%v>", expected.FirstTimestamp, actual.FirstTimestamp)
 	}