| // Copyright 2019 The Wuffs Authors. |
| // |
| // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or |
| // https://www.apache.org/licenses/LICENSE-2.0> or the MIT license |
| // <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your |
| // option. This file may not be copied, modified, or distributed |
| // except according to those terms. |
| // |
| // SPDX-License-Identifier: Apache-2.0 OR MIT |
| |
| package cgozlib |
| |
| import ( |
| "bytes" |
| "compress/zlib" |
| "io" |
| "os" |
| "strings" |
| "testing" |
| ) |
| |
| // These examples come from the RAC spec. |
| var ( |
| compressedMore = "\x78\x9c\x01\x06\x00\xf9\xff\x4d\x6f\x72\x65\x21" + |
| "\x0a\x07\x42\x01\xbf" |
| |
| compressedSheep = "\x78\xf9\x0b\xe0\x02\x6e\x0a\x29\xcf\x87\x31\x01\x01" + |
| "\x00\x00\xff\xff\x18\x0c\x03\xa8" |
| |
| dictSheep = []byte("\x20sheep.\n") |
| ) |
| |
| type resetReadCloser interface { |
| zlib.Resetter |
| io.ReadCloser |
| } |
| |
| func makePure() resetReadCloser { |
| r, err := zlib.NewReader(strings.NewReader(compressedMore)) |
| if err != nil { |
| panic(err.Error()) |
| } |
| return r.(resetReadCloser) |
| } |
| |
| func testReadAll(tt *testing.T, r resetReadCloser, src io.Reader, dict []byte, want string) { |
| tt.Helper() |
| if !cgoEnabled { |
| tt.Skip("cgo is not enabled") |
| } |
| |
| if err := r.Reset(src, dict); err != nil { |
| tt.Fatalf("Reset: %v", err) |
| } |
| |
| got, err := io.ReadAll(r) |
| if err != nil { |
| tt.Fatalf("ReadAll: %v", err) |
| } |
| |
| if err := r.Close(); err != nil { |
| tt.Fatalf("Close: %v", err) |
| } |
| |
| if string(got) != want { |
| tt.Fatalf("got %q, want %q", got, want) |
| } |
| } |
| |
| func testReader(tt *testing.T, r resetReadCloser) { |
| testReadAll(tt, r, strings.NewReader(compressedMore), nil, "More!\n") |
| testReadAll(tt, r, strings.NewReader(compressedSheep), dictSheep, "Two sheep.\n") |
| testReadAll(tt, r, strings.NewReader(compressedMore), nil, "More!\n") |
| testReadAll(tt, r, strings.NewReader(compressedSheep), dictSheep, "Two sheep.\n") |
| } |
| |
| func TestCgo(tt *testing.T) { testReader(tt, &Reader{}) } |
| func TestPure(tt *testing.T) { testReader(tt, makePure()) } |
| |
| func benchmarkReader(b *testing.B, r resetReadCloser) { |
| if !cgoEnabled { |
| b.Skip("cgo is not enabled") |
| } |
| |
| src := bytes.NewReader(nil) |
| srcBytes, err := os.ReadFile("../../test/data/pi.txt.zlib") |
| if err != nil { |
| b.Fatalf("ReadFile: %v", err) |
| } |
| |
| b.ResetTimer() |
| for i := 0; i < b.N; i++ { |
| src.Reset(srcBytes) |
| |
| if err := r.Reset(src, nil); err != nil { |
| b.Fatalf("Reset: %v", err) |
| } |
| |
| if n, err := io.Copy(io.Discard, r); err != nil { |
| b.Fatalf("Copy: %v", err) |
| } else if n != 100003 { |
| b.Fatalf("Copy: got %d, want %d", n, 100003) |
| } |
| |
| if err := r.Close(); err != nil { |
| b.Fatalf("Close: %v", err) |
| } |
| } |
| } |
| |
| func BenchmarkCgo(b *testing.B) { benchmarkReader(b, &Reader{}) } |
| func BenchmarkPure(b *testing.B) { benchmarkReader(b, makePure()) } |