From 050a699bc8a13b769cc742b63db15a0789d2a6b0 Mon Sep 17 00:00:00 2001
From: Hector Sanjuan <hector@protocol.ai>
Date: Tue, 27 Feb 2018 18:40:02 +0100
Subject: [PATCH] merkledag_test.go: Handle short reads in makeTestDAG

License: MIT
Signed-off-by: Hector Sanjuan <hector@protocol.ai>
---
 merkledag/merkledag_test.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/merkledag/merkledag_test.go b/merkledag/merkledag_test.go
index ec7465162..cae3566fc 100644
--- a/merkledag/merkledag_test.go
+++ b/merkledag/merkledag_test.go
@@ -134,7 +134,10 @@ func makeTestDAG(t *testing.T, read io.Reader, ds ipld.DAGService) ipld.Node {
 	p := make([]byte, 512)
 	nodes := []*ProtoNode{}
 	var err error
-	_, err = io.ReadFull(read, p)
+	n, err = io.ReadFull(read, p)
+	if n != len(p) {
+		t.Fatal("should have read 512 bytes from the reader")
+	}
 	for err == nil {
 		protoNode := NodeWithData(p)
 		nodes = append(nodes, protoNode)