fix ut

Signed-off-by: jiefenghuang <[email protected]>

Author: jiefenghuang

pkg/chunk/prefetch_test.go

@@ -1,6 +1,7 @@
 package chunk
 
 import (
+	"context"
 	"sync/atomic"
 	"testing"
 	"time"
@@ -15,7 +16,7 @@ func TestPrefetcher(t *testing.T) {
 			chRes <- k + "Done"
 		})
 		for _, k := range keys {
-			f.fetch(k)
+			f.fetch(context.TODO(), k)
 		}
 		res := make(map[string]bool, len(keys))
 		for range keys {
@@ -39,7 +40,7 @@ func TestPrefetcher(t *testing.T) {
 			atomic.AddInt32(&counter, 1)
 		})
 		for i := 0; i < 5; i++ {
-			f.fetch("a")
+			f.fetch(context.TODO(), "a")
 		}
 		if atomic.LoadInt32(&counter) > 1 {
 			t.Errorf("Duplicate keys  fetched")
@@ -56,7 +57,7 @@ func TestPrefetcher(t *testing.T) {
 		})
 
 		for i := 0; i < maxPending+1; i++ {
-			f.fetch(string(rune('a' + i)))
+			f.fetch(context.TODO(), string(rune('a'+i)))
 		}
 
 		time.Sleep(50 * time.Millisecond)