Test for alternative hash function in Dag Modifier.

License: MIT Signed-off-by: Kevin Atkinson <k@kevina.org>
ipfs · Aug 13, 2017 · 321b72d · 321b72d
1 parent 5c96088
commit 321b72d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/unixfs/mod/dagmodifier_test.go b/unixfs/mod/dagmodifier_test.go
@@ -77,6 +77,7 @@ func runAllSubtests(t *testing.T, tfunc func(*testing.T, testu.NodeOpts)) {
 	t.Run("opts=ProtoBufLeaves", func(t *testing.T) { tfunc(t, testu.UseProtoBufLeaves) })
 	t.Run("opts=RawLeaves", func(t *testing.T) { tfunc(t, testu.UseRawLeaves) })
 	t.Run("opts=CidV1", func(t *testing.T) { tfunc(t, testu.UseCidV1) })
+	t.Run("opts=Blake2b256", func(t *testing.T) { tfunc(t, testu.UseBlake2b256) })
 }
 
 func TestDagModifierBasic(t *testing.T) {

diff --git a/unixfs/test/utils.go b/unixfs/test/utils.go
@@ -15,6 +15,7 @@ import (
 	mdagmock "github.com/ipfs/go-ipfs/merkledag/test"
 	ft "github.com/ipfs/go-ipfs/unixfs"
 
+	mh "gx/ipfs/QmVGtdTZdTFaLsaj2RwdVG8jcjNNcp1DE914DKZ2kHmXHw/go-multihash"
 	u "gx/ipfs/QmWbjfz3u6HkAdPh34dgPchGbQjob6LXLhAeCGii2TX69n/go-ipfs-util"
 	cid "gx/ipfs/QmYhQaCYEcaPPjxJX7YcPcVKkQfRy6sJ7B3XmGFk82XYdQ/go-cid"
 	node "gx/ipfs/Qmb3Hm9QDFmfYuET4pu7Kyg8JV78jFa1nvZx5vnCZsK4ck/go-ipld-format"
@@ -41,6 +42,13 @@ type NodeOpts struct {
 var UseProtoBufLeaves = NodeOpts{Prefix: mdag.V0CidPrefix()}
 var UseRawLeaves = NodeOpts{Prefix: mdag.V0CidPrefix(), ForceRawLeaves: true, RawLeavesUsed: true}
 var UseCidV1 = NodeOpts{Prefix: mdag.V1CidPrefix(), RawLeavesUsed: true}
+var UseBlake2b256 NodeOpts
+
+func init() {
+	UseBlake2b256 = UseCidV1
+	UseBlake2b256.Prefix.MhType = mh.Names["blake2b-256"]
+	UseBlake2b256.Prefix.MhLength = -1
+}
 
 func GetNode(t testing.TB, dserv mdag.DAGService, data []byte, opts NodeOpts) node.Node {
 	in := bytes.NewReader(data)