File: reader_test.go

package info (click to toggle)
golang-gopkg-cheggaaa-pb.v2 2.0.6-3
  • links: PTS, VCS
  • area: main
  • in suites: buster, experimental
  • size: 272 kB
  • sloc: makefile: 2
file content (60 lines) | stat: -rw-r--r-- 1,272 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package pb

import (
	"testing"
)

func TestPBProxyReader(t *testing.T) {
	bar := new(ProgressBar)
	if bar.GetBool(Bytes) {
		t.Errorf("By default bytes must be false")
	}

	testReader := new(testReaderCloser)
	proxyReader := bar.NewProxyReader(testReader)

	if !bar.GetBool(Bytes) {
		t.Errorf("Bytes must be true after call NewProxyReader")
	}

	for i := 0; i < 10; i++ {
		buf := make([]byte, 10)
		n, e := proxyReader.Read(buf)
		if e != nil {
			t.Errorf("Proxy reader return err: %v", e)
		}
		if n != len(buf) {
			t.Errorf("Proxy reader return unexpected N: %d (wand %d)", n, len(buf))
		}
		for _, b := range buf {
			if b != 'f' {
				t.Errorf("Unexpected read value: %v (want %v)", b, 'f')
			}
		}
		if want := int64((i + 1) * len(buf)); bar.Current() != want {
			t.Errorf("Unexpected bar current value: %d (want %d)", bar.Current(), want)
		}
	}
	proxyReader.Close()
	if !testReader.closed {
		t.Errorf("Reader must be closed after call ProxyReader.Close")
	}
	proxyReader.Reader = nil
	proxyReader.Close()
}

type testReaderCloser struct {
	closed bool
}

func (tr *testReaderCloser) Read(p []byte) (n int, err error) {
	for i := range p {
		p[i] = 'f'
	}
	return len(p), nil
}

func (tr *testReaderCloser) Close() (err error) {
	tr.closed = true
	return
}