Method TestClientSendsRSTStream_InHeaders

internal/transport/transport_test.go:3711–3740

Tests the scenario where the server sets the END_STREAM flag in the HEADERS frame and verifies that the client responds with a RST stream.

(t *testing.T)

// Tests the scenario where the server sets the END_STREAM flag in the HEADERS
// frame and verifies that the client responds with a RST stream.
func (s) TestClientSendsRSTStream_InHeaders(t *testing.T) {
	serverFrames := func(t *testing.T, framer *http2.Framer, streamID uint32) {
		var buf bytes.Buffer
		henc := hpack.NewEncoder(&buf)
		henc.WriteField(hpack.HeaderField{Name: "content-type", Value: "application/grpc"})
		if err := framer.WriteHeaders(http2.HeadersFrameParam{
			StreamID:      streamID,
			BlockFragment: buf.Bytes(),
			EndHeaders:    true,
			EndStream:     true,
		}); err != nil {
			t.Errorf("Server failed to write headers: %v", err)
		}
	}

	ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
	defer cancel()
	stream, waitForServer := setupRSTStreamOnEOSTest(ctx, t, serverFrames)
	defer waitForServer()

	if _, err := stream.readTo(make([]byte, 1)); !errors.Is(err, io.EOF) {
		t.Fatalf("stream.readTo() got %v, want %v", err, io.EOF)
	}

	// Ensure the stream is done before checking status.
	<-stream.Done()
	if code := stream.Status().Code(); code != codes.Unknown {
		t.Fatalf("stream.Status().Code() got %s, want %s", code, codes.Unknown)
	}
}

// Tests the scenario where the server sets the END_STREAM flag in the Trailers
// (HEADERS frame) and verifies that the client responds with a RST stream.