diff options
| -rw-r--r-- | racer.go | 22 | ||||
| -rw-r--r-- | racer_test.go | 33 |
2 files changed, 55 insertions, 0 deletions
diff --git a/racer.go b/racer.go new file mode 100644 index 0000000..b954222 --- /dev/null +++ b/racer.go @@ -0,0 +1,22 @@ +package main + +import ( + "net/http" + "time" +) + +func Racer(a, b string) (winner string) { + aDuration := measureResponseTime(a) + bDuration := measureResponseTime(b) + + if aDuration < bDuration { + return a + } + return b +} + +func measureResponseTime(url string) time.Duration { + start := time.Now() + http.Get(url) + return time.Since(start) +} diff --git a/racer_test.go b/racer_test.go new file mode 100644 index 0000000..a9c2b0e --- /dev/null +++ b/racer_test.go @@ -0,0 +1,33 @@ +package main + +import ( + "net/http" + "net/http/httptest" + "testing" + "time" +) + +func TestRacer(t *testing.T) { + slowServer := makeDelayedServer(20 * time.Millisecond) + fastServer := makeDelayedServer(0 * time.Millisecond) + + defer slowServer.Close() + defer fastServer.Close() + + slowUrl := slowServer.URL + fastUrl := fastServer.URL + + want := fastUrl + got := Racer(slowUrl, fastUrl) + + if got != want { + t.Errorf("got %q, want %q", got, want) + } +} + +func makeDelayedServer(delay time.Duration) *httptest.Server { + return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + time.Sleep(delay) + w.WriteHeader(http.StatusOK) + })) +} |
