Merge pull request #10647 from hashicorp/b-fix-nomad-event-endpoint-t…

…ests events: fix event endpoint tests to ignore heartbeats.
hashicorp · Jun 9, 2021 · 029953d · 029953d
1 parent 8b5b4c8
commit 029953d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/nomad/event_endpoint_test.go b/nomad/event_endpoint_test.go
@@ -1,6 +1,7 @@
 package nomad
 
 import (
+	"bytes"
 	"context"
 	"encoding/json"
 	"fmt"
@@ -96,7 +97,7 @@ OUTER:
 			}
 
 			// ignore heartbeat
-			if msg.Event == stream.JsonHeartbeat {
+			if bytes.Equal(msg.Event.Data, stream.JsonHeartbeat.Data) {
 				continue
 			}
 
@@ -283,7 +284,7 @@ OUTER:
 				t.Fatalf("Got error: %v", msg.Error.Error())
 			}
 
-			if msg.Event == stream.JsonHeartbeat {
+			if bytes.Equal(msg.Event.Data, stream.JsonHeartbeat.Data) {
 				continue
 			}