httptestでエラーを返したい時 #Go

テスト用のモックサーバーを作るために
httptest.NewServer()作ったときのエラーの作り方。

  ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    w.WriteHeader(http.StatusBadRequest)
    fmt.Fprint(w, "Bad Request")
  }))
  defer ts.Close()

Headerへの書き込みより先に、Bodyの書き込みをしてはいけない。

ダメ

  ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    fmt.Fprint(w, "Bad Request")
    w.WriteHeader(http.StatusBadRequest)
}