26d0083faf
commit a1a37d335a8e89ac89d85c00c8585d3fc02e064a
Author: Josh Baker <joshbaker77@gmail.com>
Date: Thu Oct 5 07:36:54 2017 -0700
use symlink instead of copy
commit 96399c2c92620f633611c778e5473200bfd48d41
Author: Josh Baker <joshbaker77@gmail.com>
Date: Thu Oct 5 07:19:26 2017 -0700
use dep for vendoring
66 lines
1.1 KiB
Go
66 lines
1.1 KiB
Go
package deadline
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func takesFiveMillis(stopper <-chan struct{}) error {
|
|
time.Sleep(5 * time.Millisecond)
|
|
return nil
|
|
}
|
|
|
|
func takesTwentyMillis(stopper <-chan struct{}) error {
|
|
time.Sleep(20 * time.Millisecond)
|
|
return nil
|
|
}
|
|
|
|
func returnsError(stopper <-chan struct{}) error {
|
|
return errors.New("foo")
|
|
}
|
|
|
|
func TestDeadline(t *testing.T) {
|
|
dl := New(10 * time.Millisecond)
|
|
|
|
if err := dl.Run(takesFiveMillis); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := dl.Run(takesTwentyMillis); err != ErrTimedOut {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := dl.Run(returnsError); err.Error() != "foo" {
|
|
t.Error(err)
|
|
}
|
|
|
|
done := make(chan struct{})
|
|
err := dl.Run(func(stopper <-chan struct{}) error {
|
|
<-stopper
|
|
close(done)
|
|
return nil
|
|
})
|
|
if err != ErrTimedOut {
|
|
t.Error(err)
|
|
}
|
|
<-done
|
|
}
|
|
|
|
func ExampleDeadline() {
|
|
dl := New(1 * time.Second)
|
|
|
|
err := dl.Run(func(stopper <-chan struct{}) error {
|
|
// do something possibly slow
|
|
// check stopper function and give up if timed out
|
|
return nil
|
|
})
|
|
|
|
switch err {
|
|
case ErrTimedOut:
|
|
// execution took too long, oops
|
|
default:
|
|
// some other error
|
|
}
|
|
}
|