Method TestDecompress_ClosesReader

rpc_util_test.go:636–676

TestDecompress_ClosesReader tests that the reader returned by Decompress is closed when decompress returns.

(t *testing.T)

// TestDecompress_ClosesReader tests that the reader returned by Decompress is
// closed when decompress returns.
func (s) TestDecompress_ClosesReader(t *testing.T) {
	testCases := []struct {
		name                  string
		maxReceiveMessageSize int
		wantCode              codes.Code
	}{
		{
			name:                  "close_on_failure",
			maxReceiveMessageSize: 1,
			wantCode:              codes.ResourceExhausted,
		},
		{
			name:                  "close_on_success",
			maxReceiveMessageSize: 100,
			wantCode:              codes.OK,
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			ctx, cancel := context.WithTimeout(context.Background(), defaultTestTimeout)
			defer cancel()

			ch := make(chan struct{})
			compressor := &fakeCloseCompressor{ch: ch}
			in := mustCompress(t, []byte("some data"))
			out, err := decompress(compressor, in, nil, tc.maxReceiveMessageSize, mem.DefaultBufferPool())
			if status.Code(err) != tc.wantCode {
				t.Fatalf("decompress() failed with error code %v, want %v", status.Code(err), tc.wantCode)
			}

			out.Free()

			select {
			case <-ch:
			case <-ctx.Done():
				t.Fatalf("Timed out waiting for Close to be called")
			}
		})
	}
}