1// Copyright 2010 The Go Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style
3// license that can be found in the LICENSE file.
4
5package rand
6
7import (
8	"bytes"
9	"compress/flate"
10	"io"
11	"testing"
12)
13
14func TestRead(t *testing.T) {
15	var n int = 4e6
16	if testing.Short() {
17		n = 1e5
18	}
19	b := make([]byte, n)
20	n, err := io.ReadFull(Reader, b)
21	if n != len(b) || err != nil {
22		t.Fatalf("ReadFull(buf) = %d, %s", n, err)
23	}
24
25	var z bytes.Buffer
26	f, _ := flate.NewWriter(&z, 5)
27	f.Write(b)
28	f.Close()
29	if z.Len() < len(b)*99/100 {
30		t.Fatalf("Compressed %d -> %d", len(b), z.Len())
31	}
32}
33
34func TestReadEmpty(t *testing.T) {
35	n, err := Reader.Read(make([]byte, 0))
36	if n != 0 || err != nil {
37		t.Fatalf("Read(make([]byte, 0)) = %d, %v", n, err)
38	}
39	n, err = Reader.Read(nil)
40	if n != 0 || err != nil {
41		t.Fatalf("Read(nil) = %d, %v", n, err)
42	}
43}
44
45func BenchmarkRead(b *testing.B) {
46	b.Run("32", func(b *testing.B) {
47		benchmarkRead(b, 32)
48	})
49	b.Run("4K", func(b *testing.B) {
50		benchmarkRead(b, 4<<10)
51	})
52}
53
54func benchmarkRead(b *testing.B, size int) {
55	b.SetBytes(int64(size))
56	buf := make([]byte, size)
57	for i := 0; i < b.N; i++ {
58		if _, err := Read(buf); err != nil {
59			b.Fatal(err)
60		}
61	}
62}
63