20 lines
450 B
Go
20 lines
450 B
Go
package api
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestValidatedAPIWriteTimeoutProtectsSynchronousAIRequests(t *testing.T) {
|
|
if _, err := validatedAPIWriteTimeout(2); err == nil {
|
|
t.Fatal("two-second write timeout must be rejected")
|
|
}
|
|
got, err := validatedAPIWriteTimeout(620)
|
|
if err != nil {
|
|
t.Fatalf("620-second write timeout should be accepted: %v", err)
|
|
}
|
|
if got != 620*time.Second {
|
|
t.Fatalf("write timeout = %s, want 620s", got)
|
|
}
|
|
}
|