context.TODO() a few places that now require a context

impl/graphsync_test.go

@@ -847,7 +847,7 @@ func TestUnixFSFetch(t *testing.T) {
 				return nil, errors.New("Incorrect Link Type")
 			}
 			// read block from one store
-			block, err := bs.Get(c.Cid)
+			block, err := bs.Get(context.TODO(), c.Cid)
 			if err != nil {
 				return nil, err
 			}
@@ -867,7 +867,7 @@ func TestUnixFSFetch(t *testing.T) {
 				if err != nil {
 					return err
 				}
-				return bs.Put(block)
+				return bs.Put(context.TODO(), block)
 			}
 			return &buf, committer, nil
 		}

storeutil/storeutil.go

@@ -2,6 +2,7 @@ package storeutil
 
 import (
 	"bytes"
+	"context"
 	"fmt"
 	"io"
 
@@ -19,7 +20,7 @@ func LoaderForBlockstore(bs bstore.Blockstore) ipld.Loader {
 		if !ok {
 			return nil, fmt.Errorf("Unsupported Link Type")
 		}
-		block, err := bs.Get(asCidLink.Cid)
+		block, err := bs.Get(context.TODO(), asCidLink.Cid)
 		if err != nil {
 			return nil, err
 		}
@@ -41,7 +42,7 @@ func StorerForBlockstore(bs bstore.Blockstore) ipld.Storer {
 			if err != nil {
 				return err
 			}
-			return bs.Put(block)
+			return bs.Put(context.TODO(), block)
 		}
 		return &buffer, committer, nil
 	}

storeutil/storeutil_test.go

@@ -1,6 +1,7 @@
 package storeutil
 
 import (
+	"context"
 	"io/ioutil"
 	"testing"
 
@@ -16,9 +17,10 @@ import (
 )
 
 func TestLoader(t *testing.T) {
+	ctx := context.Background()
 	store := bstore.NewBlockstore(dss.MutexWrap(datastore.NewMapDatastore()))
 	blk := testutil.GenerateBlocksOfSize(1, 1000)[0]
-	err := store.Put(blk)
+	err := store.Put(ctx, blk)
 	require.NoError(t, err, "Unable to put block to store")
 	loader := LoaderForBlockstore(store)
 	data, err := loader(cidlink.Link{Cid: blk.Cid()}, ipld.LinkContext{})
@@ -30,6 +32,7 @@ func TestLoader(t *testing.T) {
 }
 
 func TestStorer(t *testing.T) {
+	ctx := context.Background()
 	store := bstore.NewBlockstore(dss.MutexWrap(datastore.NewMapDatastore()))
 	blk := testutil.GenerateBlocksOfSize(1, 1000)[0]
 	storer := StorerForBlockstore(store)
@@ -39,6 +42,6 @@ func TestStorer(t *testing.T) {
 	require.NoError(t, err, "Unable to write data to buffer")
 	err = commit(cidlink.Link{Cid: blk.Cid()})
 	require.NoError(t, err, "Unable to commit with storer function")
-	_, err = store.Get(blk.Cid())
+	_, err = store.Get(ctx, blk.Cid())
 	require.NoError(t, err, "Block not written to store")
 }