Merge pull request #53 from yfery/FinalRedrawWhenMaxRefreshRateSet

Fix No final redraw when max_refresh_rate is set
a8m · Sep 21, 2019 · 9fa7b0e · 9fa7b0e
2 parents 7714f4e + ffb5d50
commit 9fa7b0e
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/src/pb.rs b/src/pb.rs
@@ -313,7 +313,7 @@ impl<T: Write> ProgressBar<T> {
     fn draw(&mut self) {
         let now = SteadyTime::now();
         if let Some(mrr) = self.max_refresh_rate {
-            if now - self.last_refresh_time < mrr {
+            if now - self.last_refresh_time < mrr && self.current < self.total {
                 return;
             }
         }

diff --git a/tests/lib.rs b/tests/lib.rs
@@ -106,3 +106,18 @@ fn npm_bar() {
     }
     pb.finish_println("done!");
 }
+
+#[test]
+// see: issue 45#
+fn final_redraw_max_refresh_rate() {
+    let count = 500;
+    let mut pb = ProgressBar::new(count);
+    pb.format("╢▌▌░╟");
+    pb.set_max_refresh_rate(Some(Duration::from_millis(100)));
+    for _ in 0..count {
+        pb.inc();
+        thread::sleep(Duration::from_millis(5));
+    }
+    pb.finish_println("done!");
+}
+