38 lines
779 B
Go
38 lines
779 B
Go
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)
|
|
}
|
|
}
|