fix: nil msg

waku-org · Dec 21, 2022 · e69d6e6 · e69d6e6
1 parent 627fbba
commit e69d6e6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/waku/v2/node/wakunode2_test.go b/waku/v2/node/wakunode2_test.go
@@ -117,6 +117,9 @@ func Test5000(t *testing.T) {
 			case <-ticker.C:
 				require.Fail(t, "Timeout Sub1")
 			case msg := <-sub1.C:
+				if msg == nil {
+					return
+				}
 				if bytes.Equal(msg.Message().Payload, maxMsgBytes) {
 					return
 				}
@@ -135,6 +138,9 @@ func Test5000(t *testing.T) {
 			case <-ticker.C:
 				require.Fail(t, "Timeout Sub2")
 			case msg := <-sub2.C:
+				if msg == nil {
+					return
+				}
 				if bytes.Equal(msg.Message().Payload, maxMsgBytes) {
 					return
 				}