fetchall: simplify multiple urls test.

This commit is contained in:
Arseny Balobanov 2020-02-15 17:57:04 +03:00
parent 91f9796dad
commit 901371904b

View file

@ -84,20 +84,17 @@ func TestFetchall_multipleURLs(t *testing.T) {
var fooHit, barHit int32 var fooHit, barHit int32
h := func(w http.ResponseWriter, r *http.Request) { mux := http.NewServeMux()
mux := http.NewServeMux() mux.HandleFunc("/foo", func(w http.ResponseWriter, h *http.Request) {
mux.HandleFunc("/foo", func(w http.ResponseWriter, h *http.Request) { atomic.StoreInt32(&fooHit, 1)
atomic.StoreInt32(&fooHit, 1) _, _ = w.Write([]byte("foo"))
_, _ = w.Write([]byte("foo")) })
}) mux.HandleFunc("/bar", func(w http.ResponseWriter, h *http.Request) {
mux.HandleFunc("/bar", func(w http.ResponseWriter, h *http.Request) { atomic.StoreInt32(&barHit, 1)
atomic.StoreInt32(&barHit, 1) _, _ = w.Write([]byte("bar"))
_, _ = w.Write([]byte("bar")) })
})
mux.ServeHTTP(w, r)
}
s := httptest.NewServer(http.HandlerFunc(h)) s := httptest.NewServer(mux)
defer s.Close() defer s.Close()
cmd := exec.Command(binary, s.URL+"/foo", s.URL+"/bar") cmd := exec.Command(binary, s.URL+"/foo", s.URL+"/bar")