multi: add tests for ProgressBar::visual_index

Author: chris-laplante

src/multi.rs

@@ -558,9 +558,28 @@ mod tests {
         let p1 = mp.add(ProgressBar::new(1));
         let p2 = mp.add(ProgressBar::new(1));
         let p3 = mp.add(ProgressBar::new(1));
+
+        // Check position of bars on screen
+        assert_eq!(
+            &[
+                p0.visual_index().unwrap(),
+                p1.visual_index().unwrap(),
+                p2.visual_index().unwrap(),
+                p3.visual_index().unwrap()
+            ],
+            &[0, 1, 2, 3]
+        );
+
         mp.remove(&p2);
         mp.remove(&p1);
+
+        assert_eq!(
+            &[p0.visual_index().unwrap(), p3.visual_index().unwrap(),],
+            &[0, 1]
+        );
+
         let p4 = mp.insert(1, ProgressBar::new(1));
+        assert_eq!(p4.visual_index().unwrap(), 1);
 
         let state = mp.state.read().unwrap();
         // the removed place for p1 is reused
@@ -586,6 +605,18 @@ mod tests {
         assert_eq!(p1.index(), None);
         assert_eq!(p2.index(), None);
         assert_eq!(p3.index().unwrap(), 3);
+
+        // Check position of bars on screen
+        assert_eq!(
+            &[
+                p0.visual_index(),
+                p1.visual_index(),
+                p2.visual_index(),
+                p3.visual_index(),
+                p4.visual_index()
+            ],
+            &[Some(0), None, None, Some(2), Some(1)]
+        );
     }
 
     #[test]
@@ -604,6 +635,18 @@ mod tests {
         assert_eq!(p2.index().unwrap(), 2);
         assert_eq!(p3.index().unwrap(), 3);
         assert_eq!(p4.index().unwrap(), 4);
+
+        // Check position of bars on screen
+        assert_eq!(
+            &[
+                p0.visual_index().unwrap(),
+                p1.visual_index().unwrap(),
+                p2.visual_index().unwrap(),
+                p3.visual_index().unwrap(),
+                p4.visual_index().unwrap()
+            ],
+            &[1, 2, 4, 3, 0]
+        );
     }
 
     #[test]
@@ -622,6 +665,18 @@ mod tests {
         assert_eq!(p2.index().unwrap(), 2);
         assert_eq!(p3.index().unwrap(), 3);
         assert_eq!(p4.index().unwrap(), 4);
+
+        // Check position of bars on screen
+        assert_eq!(
+            &[
+                p0.visual_index().unwrap(),
+                p1.visual_index().unwrap(),
+                p2.visual_index().unwrap(),
+                p3.visual_index().unwrap(),
+                p4.visual_index().unwrap()
+            ],
+            &[0, 2, 3, 4, 1]
+        );
     }
 
     #[test]
@@ -640,6 +695,18 @@ mod tests {
         assert_eq!(p2.index().unwrap(), 2);
         assert_eq!(p3.index().unwrap(), 3);
         assert_eq!(p4.index().unwrap(), 4);
+
+        // Check position of bars on screen
+        assert_eq!(
+            &[
+                p0.visual_index().unwrap(),
+                p1.visual_index().unwrap(),
+                p2.visual_index().unwrap(),
+                p3.visual_index().unwrap(),
+                p4.visual_index().unwrap()
+            ],
+            &[1, 2, 4, 0, 3]
+        );
     }
 
     #[test]
@@ -662,6 +729,20 @@ mod tests {
         assert_eq!(p4.index().unwrap(), 4);
         assert_eq!(p5.index().unwrap(), 5);
         assert_eq!(p6.index().unwrap(), 6);
+
+        // Check position of bars on screen
+        assert_eq!(
+            &[
+                p0.visual_index().unwrap(),
+                p1.visual_index().unwrap(),
+                p2.visual_index().unwrap(),
+                p3.visual_index().unwrap(),
+                p4.visual_index().unwrap(),
+                p5.visual_index().unwrap(),
+                p6.visual_index().unwrap()
+            ],
+            &[3, 5, 6, 0, 2, 1, 4]
+        );
     }
 
     #[test]