refactor: proxywriter_test with ewma

Author: vbauerster

proxywriter_test.go

@@ -8,6 +8,7 @@ import (
 	"testing"
 
 	"github.com/vbauerster/mpb/v8"
+	"github.com/vbauerster/mpb/v8/decor"
 )
 
 type testWriter struct {
@@ -41,6 +42,30 @@ func TestProxyWriter(t *testing.T) {
 	p.Wait()
 }
 
+func TestEwmaProxyWriter(t *testing.T) {
+	p := mpb.New(mpb.WithOutput(io.Discard))
+	bar := p.New(int64(len(content)),
+		mpb.NopStyle(),
+		mpb.AppendDecorators(decor.EwmaETA(decor.ET_STYLE_GO, 30)),
+	)
+
+	var buf bytes.Buffer
+	tw := &testWriter{&buf, false}
+	_, err := io.Copy(bar.ProxyWriter(tw), strings.NewReader(content))
+	if err != nil {
+		t.Errorf("io.Copy: %s\n", err.Error())
+	}
+
+	if !tw.called {
+		t.Error("Read not called")
+	}
+
+	if got := buf.String(); got != content {
+		t.Errorf("Expected content: %s, got: %s\n", content, got)
+	}
+	p.Wait()
+}
+
 type testWriteCloser struct {
 	io.Writer
 	called bool
@@ -69,6 +94,27 @@ func TestProxyWriteCloser(t *testing.T) {
 	p.Wait()
 }
 
+func TestEwmaProxyWriteCloser(t *testing.T) {
+	p := mpb.New(mpb.WithOutput(io.Discard))
+	bar := p.New(int64(len(content)),
+		mpb.NopStyle(),
+		mpb.AppendDecorators(decor.EwmaETA(decor.ET_STYLE_GO, 30)),
+	)
+
+	var buf bytes.Buffer
+	tw := &testWriteCloser{&buf, false}
+	wc := bar.ProxyWriter(tw)
+	_, err := io.Copy(wc, strings.NewReader(content))
+	if err != nil {
+		t.Errorf("io.Copy: %s\n", err.Error())
+	}
+	_ = wc.Close()
+	if !tw.called {
+		t.Error("Close not called")
+	}
+	p.Wait()
+}
+
 type testWriterReadFrom struct {
 	io.Writer
 	called bool
@@ -112,3 +158,28 @@ func TestProxyWriterReadFrom(t *testing.T) {
 	}
 	p.Wait()
 }
+
+func TestEwmaProxyWriterReadFrom(t *testing.T) {
+	p := mpb.New(mpb.WithOutput(io.Discard))
+	bar := p.New(int64(len(content)),
+		mpb.NopStyle(),
+		mpb.AppendDecorators(decor.EwmaETA(decor.ET_STYLE_GO, 30)),
+	)
+
+	var buf bytes.Buffer
+	tw := &testWriterReadFrom{&buf, false}
+	// To trigger ReadFrom, WriteTo of strings.NewReader needs to be hidden
+	_, err := io.Copy(bar.ProxyWriter(tw), wrapReader{strings.NewReader(content)})
+	if err != nil {
+		t.Errorf("io.Copy: %s\n", err.Error())
+	}
+
+	if !tw.called {
+		t.Error("ReadFrom not called")
+	}
+
+	if got := buf.String(); got != content {
+		t.Errorf("Expected content: %s, got: %s\n", content, got)
+	}
+	p.Wait()
+}