31 lines
573 B
Go
31 lines
573 B
Go
|
package policies
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestTimeoutPolicy(t *testing.T) {
|
||
|
|
policy := UnknownPurposeTimeout
|
||
|
|
result := func() error {
|
||
|
|
time.Sleep(2 * time.Second)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
err := policy.ExecuteAction(result)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Action should ahve returned nil: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTimeoutPolicyExpires(t *testing.T) {
|
||
|
|
policy := TimeoutPolicy(1 * time.Second)
|
||
|
|
result := func() error {
|
||
|
|
time.Sleep(5 * time.Second)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
err := policy.ExecuteAction(result)
|
||
|
|
if err == nil {
|
||
|
|
t.Errorf("Action should have returned err")
|
||
|
|
}
|
||
|
|
}
|