Function testEventResponseWrite

tools/router/event_test.go:901–959

(
	t *testing.T,
	scenario testResponseWriteScenario[T],
	writeFunc func(e *router.Event) error,
)

}

func testEventResponseWrite[T any](
	t *testing.T,
	scenario testResponseWriteScenario[T],
	writeFunc func(e *router.Event) error,
) {
	t.Run(scenario.name, func(t *testing.T) {
		req, err := http.NewRequest(http.MethodGet, "/", nil)
		if err != nil {
			t.Fatal(err)
		}

		rec := httptest.NewRecorder()
		event := &router.Event{
			Request:  req,
			Response: &router.ResponseWriter{ResponseWriter: rec},
		}

		for k, v := range scenario.headers {
			event.Response.Header().Add(k, v)
		}

		err = writeFunc(event)
		if (scenario.expectedError != nil || err != nil) && !errors.Is(err, scenario.expectedError) {
			t.Fatalf("Expected error %v, got %v", scenario.expectedError, err)
		}

		result := rec.Result()

		if result.StatusCode != scenario.expectedStatus {
			t.Fatalf("Expected status code %d, got %d", scenario.expectedStatus, result.StatusCode)
		}

		resultBody, err := io.ReadAll(result.Body)
		result.Body.Close()
		if err != nil {
			t.Fatalf("Failed to read response body: %v", err)
		}

		resultBody, err = json.Marshal(string(resultBody))
		if err != nil {
			t.Fatal(err)
		}

		expectedBody, err := json.Marshal(scenario.expectedBody)
		if err != nil {
			t.Fatal(err)
		}

		if !bytes.Equal(resultBody, expectedBody) {
			t.Fatalf("Expected body\n%s\ngot\n%s", expectedBody, resultBody)
		}

		for k, ev := range scenario.expectedHeaders {
			if v := result.Header.Get(k); v != ev {
				t.Fatalf("Expected %q header to be %q, got %q", k, ev, v)
			}
		}
	})