Function TestNewSyncToAsyncResponse

modules/frontend/pipeline/responses_test.go:23–72

(t *testing.T)

)

func TestNewSyncToAsyncResponse(t *testing.T) {
	expected := &http.Response{
		Header: http.Header{
			"foo": []string{"bar"},
		},
		StatusCode: http.StatusAlreadyReported,
		Status:     http.StatusText(http.StatusEarlyHints),
		Body:       nil,
	}

	asyncR := NewHTTPToAsyncResponse(expected)

	// confirm we get back what we put in
	actual, done, err := asyncR.Next(context.Background())
	require.True(t, done)
	require.NoError(t, err)
	require.Equal(t, expected, actual.HTTPResponse())

	// confirm errored context is honored
	ctx, cancel := context.WithCancel(context.Background())
	cancel()
	actual, done, err = asyncR.Next(ctx)
	require.True(t, done)
	require.Error(t, err)
	require.Nil(t, actual)

	// confirm bad request is expected
	asyncR = NewBadRequest(errors.New("foo"))
	expected = &http.Response{
		StatusCode: http.StatusBadRequest,
		Status:     http.StatusText(http.StatusBadRequest),
		Body:       io.NopCloser(strings.NewReader("foo")),
	}
	actual, done, err = asyncR.Next(context.Background())
	require.True(t, done)
	require.NoError(t, err)
	require.Equal(t, expected, actual.HTTPResponse())

	// confirm successful response is expected
	asyncR = NewSuccessfulResponse("foo")
	expected = &http.Response{
		StatusCode: http.StatusOK,
		Status:     http.StatusText(http.StatusOK),
		Body:       io.NopCloser(strings.NewReader("foo")),
	}
	actual, done, err = asyncR.Next(context.Background())
	require.True(t, done)
	require.NoError(t, err)
	require.Equal(t, expected, actual.HTTPResponse())
}

func TestAsyncResponseReturnsResponsesInOrder(t *testing.T) {
	// create a slice of responses and send them through