| 86 | } |
| 87 | |
| 88 | func TestXerialFramedCompressionOptimized(t *testing.T) { |
| 89 | rawData := new(bytes.Buffer) |
| 90 | rawData.Grow(1024 * 1024) |
| 91 | io.CopyN(rawData, rand.Reader, 1024*1024) |
| 92 | |
| 93 | framedAndCompressedData := new(bytes.Buffer) |
| 94 | framedAndCompressedData.Grow(rawData.Len()) |
| 95 | w := &xerialWriter{writer: framedAndCompressedData, framed: true, encode: snappy.Encode} |
| 96 | r := simpleReader{bytes.NewReader(rawData.Bytes())} |
| 97 | io.Copy(w, r) |
| 98 | w.Flush() |
| 99 | |
| 100 | unframedAndDecompressedData := new(bytes.Buffer) |
| 101 | unframedAndDecompressedData.Grow(rawData.Len()) |
| 102 | io.Copy(unframedAndDecompressedData, |
| 103 | &xerialReader{reader: framedAndCompressedData, decode: snappy.Decode}) |
| 104 | |
| 105 | b0 := rawData.Bytes() |
| 106 | b1 := unframedAndDecompressedData.Bytes() |
| 107 | |
| 108 | if !bytes.Equal(b0, b1) { |
| 109 | t.Error("data mismatch") |
| 110 | } |
| 111 | } |
| 112 | |
| 113 | func TestXerialReaderAgainstGoXerialSnappy(t *testing.T) { |
| 114 | rawData := new(bytes.Buffer) |