rand_test.go 869 B

12345678910111213141516171819202122232425262728293031323334353637383940414243
  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. package rand
  5. import (
  6. "bytes"
  7. "compress/flate"
  8. "io"
  9. "testing"
  10. )
  11. func TestRead(t *testing.T) {
  12. var n int = 4e6
  13. if testing.Short() {
  14. n = 1e5
  15. }
  16. b := make([]byte, n)
  17. n, err := io.ReadFull(Reader, b)
  18. if n != len(b) || err != nil {
  19. t.Fatalf("ReadFull(buf) = %d, %s", n, err)
  20. }
  21. var z bytes.Buffer
  22. f, _ := flate.NewWriter(&z, 5)
  23. f.Write(b)
  24. f.Close()
  25. if z.Len() < len(b)*99/100 {
  26. t.Fatalf("Compressed %d -> %d", len(b), z.Len())
  27. }
  28. }
  29. func TestReadEmpty(t *testing.T) {
  30. n, err := Reader.Read(make([]byte, 0))
  31. if n != 0 || err != nil {
  32. t.Fatalf("Read(make([]byte, 0)) = %d, %v", n, err)
  33. }
  34. n, err = Reader.Read(nil)
  35. if n != 0 || err != nil {
  36. t.Fatalf("Read(nil) = %d, %v", n, err)
  37. }
  38. }