diff --git a/merkledag/merkledag_test.go b/merkledag/merkledag_test.go index 2739e2195..430e31f7a 100644 --- a/merkledag/merkledag_test.go +++ b/merkledag/merkledag_test.go @@ -32,13 +32,6 @@ type dagservAndPinner struct { mp pin.Pinner } -func getDagserv(t *testing.T) DAGService { - db := dssync.MutexWrap(ds.NewMapDatastore()) - bs := bstore.NewBlockstore(db) - blockserv := bserv.New(bs, offline.Exchange(bs)) - return NewDAGService(blockserv) -} - func getDagservAndPinner(t *testing.T) dagservAndPinner { db := dssync.MutexWrap(ds.NewMapDatastore()) bs := bstore.NewBlockstore(db) @@ -253,7 +246,7 @@ func assertCanGet(t *testing.T, ds DAGService, n *Node) { } func TestEmptyKey(t *testing.T) { - ds := getDagserv(t) + ds := dstest.Mock() _, err := ds.Get(context.Background(), key.Key("")) if err != ErrNotFound { t.Error("dag service should error when key is nil", err)