package aws_test import ( "fmt" "testing" "time" "github.com/aws/aws-sdk-go/aws" "github.com/aws/aws-sdk-go/awstesting" ) func TestSleepWithContext(t *testing.T) { ctx := &awstesting.FakeContext{DoneCh: make(chan struct{})} err := aws.SleepWithContext(ctx, 1*time.Millisecond) if err != nil { t.Errorf("expect context to not be canceled, got %v", err) } } func TestSleepWithContext_Canceled(t *testing.T) { ctx := &awstesting.FakeContext{DoneCh: make(chan struct{})} expectErr := fmt.Errorf("context canceled") ctx.Error = expectErr close(ctx.DoneCh) err := aws.SleepWithContext(ctx, 1*time.Millisecond) if err == nil { t.Fatalf("expect error, did not get one") } if e, a := expectErr, err; e != a { t.Errorf("expect %v error, got %v", e, a) } }