fix TestPinGC localstore test data race on test hook (#207)

1685e1ec · Janoš Guljaš · GitHub · 9c84e3ff · 1685e1ec
Commit 1685e1ec authored May 25, 2020 by Janoš Guljaš Committed by GitHub May 25, 2020
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

gc_test.go pkg/localstore/gc_test.go +2 -2

No files found.
--- a/pkg/localstore/gc_test.go
+++ b/pkg/localstore/gc_test.go
@@ -149,12 +149,12 @@ func TestPinGC(t *testing.T) {

 	var closed chan struct{}
 	testHookCollectGarbageChan := make(chan uint64)
-	defer setTestHookCollectGarbage(func(collectedCount uint64) {
+	t.Cleanup(setTestHookCollectGarbage(func(collectedCount uint64) {
 		select {
 		case testHookCollectGarbageChan <- collectedCount:
 		case <-closed:
 		}
-	})()
+	}))

 	db := newTestDB(t, &Options{
 		Capacity: dbCapacity,