From b44471d7ce60d06cf797f34c7204cd0a059ac982 Mon Sep 17 00:00:00 2001
From: Raghav Gulati <rargulati@users.noreply.github.com>
Date: Tue, 7 May 2019 10:10:05 -0700
Subject: [PATCH] Ensure that the received message is the correct one

---
 floodsub_test.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/floodsub_test.go b/floodsub_test.go
index 38f5ff0..52ba129 100644
--- a/floodsub_test.go
+++ b/floodsub_test.go
@@ -1057,4 +1057,11 @@ func TestImproperlySignedMessageNotRelayed(t *testing.T) {
 	if len(honestPeerMessages) != 1 {
 		t.Fatalf("got %d messages, expected 1", len(honestPeerMessages))
 	}
+	if string(honestPeerMessages[0].GetData()) != string(correctMessage) {
+		t.Fatalf(
+			"got %s, expected message %s",
+			honestPeerMessages[0].GetData(),
+			correctMessage,
+		)
+	}
 }