Method TestDecompress_NoReadAfterEOF

rpc_util_test.go:581–612

Tests that the decompressor's Read method is not called after it returns EOF.

(t *testing.T)

// Tests that the decompressor's Read method is not called after it returns EOF.
func (s) TestDecompress_NoReadAfterEOF(t *testing.T) {
	ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
	defer cancel()

	ch := make(chan struct{}, 10)
	mc := &mockCompressor{ch: ch}
	in := mem.BufferSlice{mem.NewBuffer(&[]byte{1, 2, 3}, nil)}
	wg := sync.WaitGroup{}
	wg.Add(1)
	go func() {
		defer wg.Done()
		out, err := decompress(mc, in, nil, 1, mem.DefaultBufferPool())
		if err != nil {
			t.Errorf("Unexpected error from decompress: %v", err)
			return
		}
		out.Free()
	}()
	select {
	case <-ch:
	case <-ctx.Done():
		t.Fatalf("Timed out waiting for call to compressor")
	}
	ctx, cancel = context.WithTimeout(ctx, defaultTestShortTimeout)
	defer cancel()
	select {
	case <-ch:
		t.Fatalf("Unexpected second compressor.Read call detected")
	case <-ctx.Done():
	}
	wg.Wait()
}

type fakeCloseCompressor struct {
	encoding.Compressor