Function testCompress

pkg/util/grpcencoding/encoding_test.go:40–85

(name string, t *testing.T)

}

func testCompress(name string, t *testing.T) {
	c := encoding.GetCompressor(name)
	assert.Equal(t, name, c.Name())

	tests := []struct {
		test  string
		input string
	}{
		{"empty", ""},
		{"short", "hello world"},
		{"long", strings.Repeat("123456789", 2024)},
	}
	for _, test := range tests {
		t.Run(test.test, func(t *testing.T) {
			buf := &bytes.Buffer{}
			// Compress
			w, err := c.Compress(buf)
			require.NoError(t, err)
			n, err := w.Write([]byte(test.input))
			require.NoError(t, err)
			assert.Len(t, test.input, n)
			err = w.Close()
			require.NoError(t, err)
			compressedBytes := buf.Bytes()
			buf = bytes.NewBuffer(compressedBytes)

			// Decompress
			r, err := c.Decompress(buf)
			require.NoError(t, err)
			out, err := io.ReadAll(r)
			require.NoError(t, err)
			assert.Equal(t, test.input, string(out))

			if sizer, ok := c.(interface {
				DecompressedSize(compressedBytes []byte) int
			}); ok {
				buf = bytes.NewBuffer(compressedBytes)
				r, err := c.Decompress(buf)
				require.NoError(t, err)
				out, err := io.ReadAll(r)
				require.NoError(t, err)
				assert.Equal(t, len(out), sizer.DecompressedSize(compressedBytes))
			}
		})
	}
}

func BenchmarkCompress(b *testing.B) {
	data := []byte(strings.Repeat("123456789", 1024))