Function TestRequestBatchBasics

modules/frontend/v1/request_batch_test.go:18–46

(t *testing.T)

)

func TestRequestBatchBasics(t *testing.T) {
	rb := &requestBatch{}

	const totalRequests = 3

	for i := byte(0); i < totalRequests; i++ {
		req := httptest.NewRequest("GET", "http://example.com", bytes.NewReader([]byte{i}))
		_ = rb.add(&request{
			request: pipeline.NewHTTPRequest(req),
		})
	}

	// confirm len
	require.Equal(t, totalRequests, rb.len())

	// confirm grpc requests are ordered
	grpcRequests := rb.httpGrpcRequests()
	require.Equal(t, totalRequests, len(grpcRequests))
	for i := byte(0); i < byte(len(grpcRequests)); i++ {
		require.Equal(t, []byte{i}, grpcRequests[i].Body)
	}

	// clear and confirm len
	rb.clear()
	require.Equal(t, 0, rb.len())

	grpcRequests = rb.httpGrpcRequests()
	require.Equal(t, 0, len(grpcRequests))
}

func TestRequestBatchContextError(t *testing.T) {
	rb := &requestBatch{}