21 lines
433 B
Go
21 lines
433 B
Go
package slow_test
|
|
|
|
import "show-rss/src/slow"
|
|
import "testing"
|
|
import "context"
|
|
import "bytes"
|
|
import "io"
|
|
|
|
func TestReader(t *testing.T) {
|
|
junk := bytes.NewReader(bytes.Repeat([]byte("1"), 256_000))
|
|
|
|
slowReader := slow.NewReader(context.Background(), 300_000, junk)
|
|
|
|
buff := bytes.NewBuffer(nil)
|
|
if n, err := io.Copy(buff, slowReader); err != nil {
|
|
t.Fatal(err)
|
|
} else if n != 256_000 {
|
|
t.Fatal(n)
|
|
}
|
|
}
|