1
0
mirror of https://github.com/ipfs/kubo.git synced 2025-06-29 01:12:24 +08:00

Use ReadFull. Remove duplicated code

License: MIT
Signed-off-by: Hector Sanjuan <hector@protocol.ai>
This commit is contained in:
Hector Sanjuan
2018-03-01 14:17:22 +01:00
parent 050a699bc8
commit cd9c8c5d45

View File

@ -133,19 +133,23 @@ func TestBatchFetchDupBlock(t *testing.T) {
func makeTestDAG(t *testing.T, read io.Reader, ds ipld.DAGService) ipld.Node {
p := make([]byte, 512)
nodes := []*ProtoNode{}
var err error
n, err = io.ReadFull(read, p)
if n != len(p) {
t.Fatal("should have read 512 bytes from the reader")
}
for err == nil {
for {
n, err := io.ReadFull(read, p)
if err == io.EOF {
break
}
if err != nil {
t.Fatal(err)
}
if n != len(p) {
t.Fatal("should have read 512 bytes from the reader")
}
protoNode := NodeWithData(p)
nodes = append(nodes, protoNode)
_, err = read.Read(p)
}
if err != io.EOF {
t.Fatal(err)
}
ctx := context.Background()
@ -158,7 +162,7 @@ func makeTestDAG(t *testing.T, read io.Reader, ds ipld.DAGService) ipld.Node {
t.Fatal(err)
}
}
err = ds.Add(ctx, root)
err := ds.Add(ctx, root)
if err != nil {
t.Fatal(err)
}