Update tests

janos · Feb 8, 2020 · 17c785a · 17c785a
1 parent ea5c1f5
commit 17c785a
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/singleflight_test.go b/singleflight_test.go
@@ -159,8 +159,11 @@ func TestDo_cancelContextSecond(t *testing.T) {
 		time.Sleep(time.Second)
 		return want, nil
 	}
-	go g.Do(context.Background(), "key", fn)
-
+	go func() {
+		if _, _, err := g.Do(context.Background(), "key", fn); err != nil {
+			panic(err)
+		}
+	}()
 	ctx, cancel := context.WithCancel(context.Background())
 	go func() {
 		time.Sleep(100 * time.Millisecond)
@@ -197,7 +200,11 @@ func TestForget(t *testing.T) {
 		return wantPrefix + strconv.FormatUint(c, 10), nil
 	}
 
-	go g.Do(context.Background(), "key", fn)
+	go func() {
+		if _, _, err := g.Do(context.Background(), "key", fn); err != nil {
+			panic(err)
+		}
+	}()
 
 	<-firstCall
 	g.Forget("key")