2024-09-18 08:15:53 +00:00
|
|
|
package racer
|
|
|
|
|
2024-09-18 10:45:58 +00:00
|
|
|
import (
|
|
|
|
"net/http"
|
|
|
|
"net/http/httptest"
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
)
|
2024-09-18 08:15:53 +00:00
|
|
|
|
|
|
|
func TestRacer(t *testing.T) {
|
2024-09-18 10:45:58 +00:00
|
|
|
slowServer := httptest.NewServer(http.HandlerFunc(
|
|
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
time.Sleep(20 * time.Millisecond)
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
|
|
}))
|
|
|
|
defer slowServer.Close()
|
|
|
|
|
|
|
|
fastServer := httptest.NewServer(http.HandlerFunc(
|
|
|
|
func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
w.WriteHeader(http.StatusOK)
|
|
|
|
}))
|
|
|
|
defer fastServer.Close()
|
|
|
|
|
|
|
|
slowURL := slowServer.URL
|
|
|
|
fastURL := fastServer.URL
|
2024-09-18 08:15:53 +00:00
|
|
|
|
|
|
|
want := fastURL
|
|
|
|
got := Racer(slowURL, fastURL)
|
|
|
|
|
|
|
|
if got != want {
|
|
|
|
t.Errorf("got %q, want %q", got, want)
|
|
|
|
}
|
|
|
|
}
|