blob: 2589b05cd2f59283a861a5e109fedda547d7cee6 (
plain) (
blame)
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
|
package testing
import (
"io"
"sync"
)
// ByteLoop provides an io.ReadCloser that always fills the read byte slice
// with repeating value, until closed.
type ByteLoop struct {
value byte
closed bool
mu sync.RWMutex
}
// Read populates the passed in byte slice with the value the ByteLoop was
// created with. Returns the size of bytes written. If the reader is closed,
// io.EOF will be returned.
func (l *ByteLoop) Read(p []byte) (size int, err error) {
l.mu.RLock()
defer l.mu.RUnlock()
if l.closed {
return 0, io.EOF
}
for i := 0; i < len(p); i++ {
p[i] = l.value
}
return len(p), nil
}
// Close closes the ByteLoop, and prevents any further reading.
func (l *ByteLoop) Close() error {
l.mu.Lock()
defer l.mu.Unlock()
l.closed = true
return nil
}
|
