Function TestWriteThenWriterContextTakeover

compress_test.go:141–224

TestWriteThenWriterContextTakeover verifies that using Conn.Write followed by Conn.Writer works correctly with context takeover enabled. This tests that the flateWriter destination is properly restored after Conn.Write redirects it to a temporary buffer.

(t *testing.T)

// the flateWriter destination is properly restored after Conn.Write redirects
// it to a temporary buffer.
func TestWriteThenWriterContextTakeover(t *testing.T) {
	t.Parallel()

	clientConn, serverConn := net.Pipe()
	defer clientConn.Close()
	defer serverConn.Close()

	client := newConn(connConfig{
		rwc:            clientConn,
		client:         true,
		copts:          CompressionContextTakeover.opts(),
		flateThreshold: 64,
		br:             bufio.NewReader(clientConn),
		bw:             bufio.NewWriterSize(clientConn, 4096),
	})

	server := newConn(connConfig{
		rwc:            serverConn,
		client:         false,
		copts:          CompressionContextTakeover.opts(),
		flateThreshold: 64,
		br:             bufio.NewReader(serverConn),
		bw:             bufio.NewWriterSize(serverConn, 4096),
	})

	ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond*500)
	defer cancel()

	msg1 := []byte(strings.Repeat("first message ", 100))
	msg2 := []byte(strings.Repeat("second message ", 100))

	type readResult struct {
		typ MessageType
		p   []byte
		err error
	}
	readCh := make(chan readResult, 3)
	go func() {
		for range 3 {
			typ, p, err := server.Read(ctx)
			readCh <- readResult{typ, p, err}
		}
	}()

	// We want to verify mixing `Write` and `Writer` usages still work.
	//
	// To this end, we call them in this order:
	// - `Write`
	// - `Writer`
	// - `Write`
	//
	// This verifies that it works for a `Write` followed by a `Writer`
	// as well as a `Writer` followed by a `Write`.

	// 1. `Write` API
	err := client.Write(ctx, MessageText, msg1)
	assert.Success(t, err)