diff --git a/bar_option.go b/bar_option.go index 660e7c4..4ba4905 100644 --- a/bar_option.go +++ b/bar_option.go @@ -89,7 +89,10 @@ return BarFillerMiddleware(func(base BarFiller) BarFiller { return BarFillerFunc(func(w io.Writer, reqWidth int, st decor.Statistics) { if st.Completed { - io.WriteString(w, message) + _, err := io.WriteString(w, message) + if err != nil { + panic(err) + } } else { base.Fill(w, reqWidth, st) } diff --git a/draw_test.go b/draw_test.go index db112ba..b6fa7a6 100644 --- a/draw_test.go +++ b/draw_test.go @@ -774,7 +774,10 @@ s.trimSpace = tc.trim s.refill = tc.refill tmpBuf.Reset() - tmpBuf.ReadFrom(s.draw(newStatistics(tw, s))) + _, err := tmpBuf.ReadFrom(s.draw(newStatistics(tw, s))) + if err != nil { + t.FailNow() + } by := tmpBuf.Bytes() got := string(by[:len(by)-1]) @@ -1221,7 +1224,10 @@ s.trimSpace = tc.trim s.refill = tc.refill tmpBuf.Reset() - tmpBuf.ReadFrom(s.draw(newStatistics(tw, s))) + _, err := tmpBuf.ReadFrom(s.draw(newStatistics(tw, s))) + if err != nil { + t.FailNow() + } by := tmpBuf.Bytes() got := string(by[:len(by)-1]) @@ -1390,7 +1396,10 @@ s.trimSpace = tc.trim s.refill = tc.refill tmpBuf.Reset() - tmpBuf.ReadFrom(s.draw(newStatistics(tw, s))) + _, err := tmpBuf.ReadFrom(s.draw(newStatistics(tw, s))) + if err != nil { + t.FailNow() + } by := tmpBuf.Bytes() got := string(by[:len(by)-1]) diff --git a/progress.go b/progress.go index a4a0d48..09b2887 100644 --- a/progress.go +++ b/progress.go @@ -320,7 +320,10 @@ for s.bHeap.Len() > 0 { b := heap.Pop(&s.bHeap).(*Bar) frame := <-b.frameCh - cw.ReadFrom(frame.reader) + _, err := cw.ReadFrom(frame.reader) + if err != nil { + return err + } if b.toShutdown { if b.recoveredPanic != nil { s.barShutdownQueue = append(s.barShutdownQueue, b)