Function makeTestServer

browsertests/testutils.go:17–58

(t testing.TB, prof *profile.Profile)

)

func makeTestServer(t testing.TB, prof *profile.Profile) *httptest.Server {
	if runtime.GOOS == "nacl" || runtime.GOOS == "js" {
		t.Skip("test assumes tcp available")
	}

	// Custom http server creator
	var server *httptest.Server
	serverCreated := make(chan bool)
	creator := func(a *driver.HTTPServerArgs) error {
		server = httptest.NewServer(http.HandlerFunc(
			func(w http.ResponseWriter, r *http.Request) {
				if h := a.Handlers[r.URL.Path]; h != nil {
					h.ServeHTTP(w, r)
				}
			}))
		serverCreated <- true
		return nil
	}

	// Start server and wait for it to be initialized
	go func() {
		err := driver.PProf(&driver.Options{
			Obj:        fakeObjTool{},
			UI:         testUI{t},
			Fetch:      testFetcher{prof},
			HTTPServer: creator,
			Flagset: testFlags{
				"http":       "unused:1234",
				"no_browser": true,
			},
		})
		if err != nil {
			panic(err)
		}
	}()
	<-serverCreated

	// Close the server when the test is done.
	t.Cleanup(server.Close)

	return server
}

// Fake test implementations of types needed by pprof driver.