test(gateapi): gateapi generated code expects JSON header to be sent

Author: kevinawoo

cmd/account/get_test.go

@@ -142,6 +142,7 @@ func TestAccountGet_fail(t *testing.T) {
 func testGateAccountGetSuccess() *httptest.Server {
 	mux := util.TestGateMuxWithVersionHandler()
 	mux.Handle("/credentials/"+ACCOUNT, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Add("content-type", "application/json")
 		fmt.Fprintln(w, strings.TrimSpace(accountJson))
 	}))
 	return httptest.NewServer(mux)

cmd/account/list_test.go

@@ -74,6 +74,7 @@ func TestAccountList_fail(t *testing.T) {
 func testGateAccountListSuccess() *httptest.Server {
 	mux := util.TestGateMuxWithVersionHandler()
 	mux.Handle("/credentials/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Add("content-type", "application/json")
 		fmt.Fprintln(w, strings.TrimSpace(accountListJson))
 	}))
 	return httptest.NewServer(mux)

cmd/application/delete_test.go

@@ -78,20 +78,23 @@ func testGateApplicationDeleteSuccess() *httptest.Server {
 	mux.Handle("/applications/"+APP, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		payload := map[string]string{} // We don't use the payload, we are just checking if the target app exists.
 		b, _ := json.Marshal(&payload)
+		w.Header().Add("content-type", "application/json")
 		fmt.Fprintln(w, string(b))
 	}))
 	mux.Handle("/tasks", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		payload := map[string]string{
 			"ref": "/tasks/id",
 		}
 		b, _ := json.Marshal(&payload)
+		w.Header().Add("content-type", "application/json")
 		fmt.Fprintln(w, string(b))
 	}))
 	mux.Handle("/tasks/id", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		payload := map[string]string{
 			"status": "SUCCEEDED",
 		}
 		b, _ := json.Marshal(&payload)
+		w.Header().Add("content-type", "application/json")
 		fmt.Fprintln(w, string(b))
 	}))
 	return httptest.NewServer(mux)

cmd/canary/canary-config/get_test.go

@@ -157,6 +157,7 @@ func testGateCanaryConfigGetSuccess() *httptest.Server {
 	mux.Handle(
 		"/v2/canaryConfig/",
 		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			w.Header().Add("content-type", "application/json")
 			fmt.Fprintln(w, strings.TrimSpace(canaryConfigGetJson))
 		}))
 	return httptest.NewServer(mux)

cmd/canary/canary-config/list_test.go

@@ -85,6 +85,7 @@ func testGateCanaryConfigListSuccess() *httptest.Server {
 	mux.Handle(
 		"/v2/canaryConfig",
 		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			w.Header().Add("content-type", "application/json")
 			fmt.Fprintln(w, strings.TrimSpace(canaryConfigListJson))
 		}))
 	return httptest.NewServer(mux)
@@ -96,6 +97,7 @@ func testGateCanaryConfigListMalformed() *httptest.Server {
 	mux.Handle(
 		"/v2/canaryConfig",
 		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			w.Header().Add("content-type", "application/json")
 			fmt.Fprintln(w, strings.TrimSpace(malformedCanaryConfigListJson))
 		}))
 	return httptest.NewServer(mux)
@@ -105,6 +107,7 @@ func testGateCanaryConfigListMalformed() *httptest.Server {
 // to direct requests to. Responds with a 500 InternalServerError.
 func testGateFail() *httptest.Server {
 	return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Add("content-type", "application/json")
 		http.Error(w, "Internal Server Error", http.StatusInternalServerError)
 	}))
 }

cmd/canary/canary-config/retro_test.go

@@ -169,10 +169,12 @@ func TestCanaryConfigRetro_timeout(t *testing.T) {
 func gateServerRetroPass() *httptest.Server {
 	mux := util.TestGateMuxWithVersionHandler()
 	mux.Handle("/v2/canaries/canary", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Add("content-type", "application/json")
 		w.Write([]byte(canaryExecRespJson))
 	}))
 
 	mux.Handle("/v2/canaries/canary/executionId", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Add("content-type", "application/json")
 		w.Write([]byte(canaryFinishedPassJson))
 	}))
 	return httptest.NewServer(mux)
@@ -181,10 +183,12 @@ func gateServerRetroPass() *httptest.Server {
 func gateServerExecHang() *httptest.Server {
 	mux := util.TestGateMuxWithVersionHandler()
 	mux.Handle("/v2/canaries/canary", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Add("content-type", "application/json")
 		w.Write([]byte(canaryExecRespJson))
 	}))
 
 	mux.Handle("/v2/canaries/canary/executionId", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Add("content-type", "application/json")
 		w.Write([]byte(canaryUnfinishedJson))
 	}))
 	return httptest.NewServer(mux)

cmd/pipeline/get_test.go

@@ -144,6 +144,7 @@ func TestPipelineGet_notfound(t *testing.T) {
 func testGatePipelineGetSuccess() *httptest.Server {
 	mux := util.TestGateMuxWithVersionHandler()
 	mux.Handle("/applications/app/pipelineConfigs/one", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Add("content-type", "application/json")
 		fmt.Fprintln(w, strings.TrimSpace(pipelineGetJson))
 	}))
 	return httptest.NewServer(mux)

cmd/project/save_test.go

@@ -198,6 +198,7 @@ func testGateProjectSaveSuccess(buffer io.Writer) *httptest.Server {
 		util.NewTestBufferHandlerFunc(http.MethodPost, buffer, http.StatusOK, strings.TrimSpace(testAppTaskRefJsonStr)),
 	)
 	mux.Handle("/tasks/id", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Add("content-type", "application/json")
 		fmt.Fprintln(w, strings.TrimSpace(testProjectTaskStatusJsonStr))
 	}))
 	return httptest.NewServer(mux)

util/test_helpers.go

@@ -19,6 +19,7 @@ func TestGateMuxWithVersionHandler() *http.ServeMux {
 			"version": "Unknown",
 		}
 		b, _ := json.Marshal(&payload)
+		w.Header().Add("content-type", "application/json")
 		fmt.Fprintln(w, string(b))
 	}))
 
@@ -58,6 +59,7 @@ func NewTestBufferHandlerFunc(method string, requestBuffer io.Writer, responseHe
 		}
 
 		// Empty response body. Status: 200 Success
+		w.Header().Add("content-type", "application/json")
 		w.WriteHeader(responseHeader)
 		fmt.Fprintln(w, responseBody)
 	})