1 package sdk
2
3 import (
4 "context"
5 "strings"
6 "testing"
7 "time"
8 )
9
10 func TestSleepWithContext(t *testing.T) {
11 ctx, cancelFn := context.WithCancel(context.Background())
12 defer cancelFn()
13
14 err := sleepWithContext(ctx, 1*time.Millisecond)
15 if err != nil {
16 t.Errorf("expect context to not be canceled, got %v", err)
17 }
18 }
19
20 func TestSleepWithContext_Canceled(t *testing.T) {
21 ctx, cancelFn := context.WithCancel(context.Background())
22 cancelFn()
23
24 err := sleepWithContext(ctx, 10*time.Second)
25 if err == nil {
26 t.Fatalf("expect error, did not get one")
27 }
28
29 if e, a := "context canceled", err.Error(); !strings.Contains(a, e) {
30 t.Errorf("expect %v error, got %v", e, a)
31 }
32 }
33
View as plain text