Fix No final redraw when max_refresh_rate is set

a8m · Jun 29, 2017 · ffb5d50 · ffb5d50
1 parent fe5f8ba
commit ffb5d50
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
@@ -309,7 +309,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
@@ -107,3 +107,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!");
+}
+