compatible range might be nil

The-K-R-O-K · Oct 9, 2024 · 2953c1f · 2953c1f
1 parent e5a9996
commit 2953c1f
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/access/grpc/grpc.go b/access/grpc/grpc.go
@@ -166,10 +166,13 @@ func (c *BaseClient) GetNodeVersionInfo(ctx context.Context, opts ...grpc.CallOp
 		return nil, newRPCError(err)
 	}
 
+	var compRange *flow.CompatibleRange
 	info := res.GetInfo()
-	compRange := flow.CompatibleRange{
-		StartHeight: info.CompatibleRange.GetStartHeight(),
-		EndHeight:   info.CompatibleRange.GetEndHeight(),
+	if info != nil {
+		compRange = &flow.CompatibleRange{
+			StartHeight: info.CompatibleRange.GetStartHeight(),
+			EndHeight:   info.CompatibleRange.GetEndHeight(),
+		}
 	}
 
 	return &flow.NodeVersionInfo{

diff --git a/access/grpc/grpc_test.go b/access/grpc/grpc_test.go
@@ -177,7 +177,7 @@ func TestClient_GetNodeInfo(t *testing.T) {
 			ProtocolVersion:      ver,
 			SporkRootBlockHeight: spork,
 			NodeRootBlockHeight:  root,
-			CompatibleRange:      flow.CompatibleRange{StartHeight: compRange.StartHeight, EndHeight: compRange.EndHeight},
+			CompatibleRange:      &flow.CompatibleRange{StartHeight: compRange.StartHeight, EndHeight: compRange.EndHeight},
 		}
 
 		rpc.On("GetNodeVersionInfo", ctx, mock.Anything).Return(response, nil)

diff --git a/flow.go b/flow.go
@@ -134,7 +134,7 @@ type NodeVersionInfo struct {
 	ProtocolVersion      uint64
 	SporkRootBlockHeight uint64
 	NodeRootBlockHeight  uint64
-	CompatibleRange      CompatibleRange
+	CompatibleRange      *CompatibleRange
 }
 
 type CompatibleRange struct {