vinchent/go-by-test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

racer: test with timeout

Browse Source
This commit is contained in:
vinchent 2024-09-18 13:41:55 +02:00
parent ee89092ea4
commit 3556c5b805
2 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
racer/racer.go
View File

@ -8,7 +8,9 @@ import (
var ErrRacerTimeout = errors.New("timeout") var ErrRacerTimeout = errors.New("timeout")
func Racer(a, b string, timeout time.Duration) (string, error) { const tenSecondTimeout = 10 * time.Second
func ConfigurableRacer(a, b string, timeout time.Duration) (string, error) {
select { select {
// wait until channel closed // wait until channel closed
case <-ping(a): case <-ping(a):
@ -20,6 +22,10 @@ func Racer(a, b string, timeout time.Duration) (string, error) {
} }
} }
func Racer(a, b string) (string, error) {
return ConfigurableRacer(a, b, tenSecondTimeout)
}
func ping(url string) chan struct{} { func ping(url string) chan struct{} {
ch := make(chan struct{}) ch := make(chan struct{})
go func() { go func() {

14
racer/racer_test.go
View File

@ -7,6 +7,12 @@ import (
"time" "time"
) )
type SpyRacerTimeout struct{}
func (s *SpyRacerTimeout) Timeout() <-chan time.Time {
return time.After(1 * time.Millisecond)
}
func TestRacer(t *testing.T) { func TestRacer(t *testing.T) {
t.Run("compares speeds of servers, returning the url of the fasted one", func(t *testing.T) { t.Run("compares speeds of servers, returning the url of the fasted one", func(t *testing.T) {
slowServer := makeDelayedServer(20 * time.Millisecond) slowServer := makeDelayedServer(20 * time.Millisecond)
@ -19,7 +25,7 @@ func TestRacer(t *testing.T) {
fastURL := fastServer.URL fastURL := fastServer.URL
want := fastURL want := fastURL
got, _ := Racer(slowURL, fastURL, 10*time.Second) got, _ := Racer(slowURL, fastURL)
if got != want { if got != want {
t.Errorf("got %q, want %q", got, want) t.Errorf("got %q, want %q", got, want)
@ -27,13 +33,13 @@ func TestRacer(t *testing.T) {
}) })
t.Run("returns an error if a server doesn't respond within 10s", func(t *testing.T) { t.Run("returns an error if a server doesn't respond within 10s", func(t *testing.T) {
serverA := makeDelayedServer(11 * time.Second) serverA := makeDelayedServer(25 * time.Millisecond)
defer serverA.Close() defer serverA.Close()
serverB := makeDelayedServer(11 * time.Second) serverB := makeDelayedServer(25 * time.Millisecond)
defer serverB.Close() defer serverB.Close()
_, err := Racer(serverA.URL, serverB.URL, 10*time.Second) _, err := ConfigurableRacer(serverA.URL, serverB.URL, 20*time.Millisecond)
if err == nil { if err == nil {
t.Error("expected an error but didn't got one") t.Error("expected an error but didn't got one")
} }