refactor subscribe func and stream mock object

onflow · Oct 17, 2024 · 6b6ef84 · 6b6ef84
1 parent 7eb6328
commit 6b6ef84
Show file tree

Hide file tree

Showing 3 changed files with 199 additions and 536 deletions.
diff --git a/access/grpc/convert/convert.go b/access/grpc/convert/convert.go
@@ -245,12 +245,16 @@ func MessageToBlockHeader(m *entities.BlockHeader) (flow.BlockHeader, error) {
 	}, nil
 }
 
-func MessageToBlockDigest(m *access.SubscribeBlockDigestsResponse) flow.BlockDigest {
+func MessageToBlockDigest(m *access.SubscribeBlockDigestsResponse) (flow.BlockDigest, error) {
+	if m == nil {
+		return flow.BlockDigest{}, ErrEmptyMessage
+	}
+
 	return flow.BlockDigest{
 		BlockID:   flow.BytesToID(m.GetBlockId()),
 		Height:    m.GetBlockHeight(),
 		Timestamp: m.GetBlockTimestamp().AsTime(),
-	}
+	}, nil
 }
 
 func BlockDigestToMessage(blockDigest flow.BlockDigest) *access.SubscribeBlockDigestsResponse {