refactoring proxyreader_test
Vladimir Bauer
4 years ago
| 27 | 27 | return r.Reader.Read(p) |
| 28 | 28 | } |
| 29 | 29 | |
| 30 | type testWriterTo struct { | |
| 31 | *testReader | |
| 32 | wt io.WriterTo | |
| 33 | } | |
| 34 | ||
| 35 | func (wt testWriterTo) WriteTo(w io.Writer) (n int64, err error) { | |
| 36 | wt.called = true | |
| 37 | return wt.wt.WriteTo(w) | |
| 38 | } | |
| 39 | ||
| 30 | 40 | func TestProxyReader(t *testing.T) { |
| 31 | 41 | p := mpb.New(mpb.WithOutput(ioutil.Discard)) |
| 32 | 42 | |
| 33 | 43 | tReader := &testReader{strings.NewReader(content), false} |
| 34 | 44 | |
| 35 | bar := p.AddBar(int64(len(content)), mpb.BarFillerTrim()) | |
| 45 | bar := p.AddBar(int64(len(content))) | |
| 36 | 46 | |
| 37 | 47 | var buf bytes.Buffer |
| 38 | 48 | _, err := io.Copy(&buf, bar.ProxyReader(tReader)) |
| 51 | 61 | } |
| 52 | 62 | } |
| 53 | 63 | |
| 54 | type testWriterTo struct { | |
| 55 | io.Reader | |
| 56 | wt io.WriterTo | |
| 57 | called bool | |
| 58 | } | |
| 59 | ||
| 60 | func (wt *testWriterTo) WriteTo(w io.Writer) (n int64, err error) { | |
| 61 | wt.called = true | |
| 62 | return wt.wt.WriteTo(w) | |
| 63 | } | |
| 64 | ||
| 65 | 64 | func TestProxyWriterTo(t *testing.T) { |
| 66 | 65 | p := mpb.New(mpb.WithOutput(ioutil.Discard)) |
| 67 | 66 | |
| 68 | 67 | var reader io.Reader = strings.NewReader(content) |
| 69 | tReader := &testWriterTo{reader, reader.(io.WriterTo), false} | |
| 68 | tWriterTo := testWriterTo{&testReader{reader, false}, reader.(io.WriterTo)} | |
| 70 | 69 | |
| 71 | bar := p.AddBar(int64(len(content)), mpb.BarFillerTrim()) | |
| 70 | bar := p.AddBar(int64(len(content))) | |
| 72 | 71 | |
| 73 | 72 | var buf bytes.Buffer |
| 74 | _, err := io.Copy(&buf, bar.ProxyReader(tReader)) | |
| 73 | _, err := io.Copy(&buf, bar.ProxyReader(tWriterTo)) | |
| 75 | 74 | if err != nil { |
| 76 | 75 | t.Errorf("Error copying from reader: %+v\n", err) |
| 77 | 76 | } |
| 78 | 77 | |
| 79 | 78 | p.Wait() |
| 80 | 79 | |
| 81 | if !tReader.called { | |
| 80 | if !tWriterTo.called { | |
| 82 | 81 | t.Error("WriteTo not called") |
| 83 | 82 | } |
| 84 | 83 | |