drop external dep and update copy function

ipfs · Sep 6, 2018 · 83a3ec9 · 83a3ec9
1 parent a8cc0b9
commit 83a3ec9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 11 deletions.
diff --git a/daghelpers.go b/daghelpers.go
@@ -107,10 +107,10 @@ func Copy(ctx context.Context, from, to DAGService, root *cid.Cid) error {
 		if err != nil {
 			return err
 		}
-		err = to.Add(ctx, node)
-		if err != nil {
-			return err
-		}
+	}
+	err = to.Add(ctx, node)
+	if err != nil {
+		return err
 	}
 	return nil
 }

diff --git a/daghelpers_test.go b/daghelpers_test.go
@@ -4,7 +4,6 @@ import (
 	"context"
 	"github.com/ipfs/go-cid"
 	mh "github.com/multiformats/go-multihash"
-	"github.com/stretchr/testify/assert"
 	"testing"
 )
 
@@ -132,13 +131,11 @@ func TestCopy(t *testing.T) {
 	}
 
 	r, err := to.Get(ctx, root.Cid())
-	if err != nil {
-		t.Error(err)
+	if err != nil || len(r.Links()) != 2 {
+		t.Error("fail to copy dag")
 	}
 	l1, err := to.Get(ctx, l11.Cid())
-	if err != nil {
-		t.Error(err)
+	if err != nil || len(l1.Links()) != 3 {
+		t.Error("fail to copy dag")
 	}
-	assert.Equal(t, len(r.Links()), 2)
-	assert.Equal(t, len(l1.Links()), 3)
 }