1 files changed, 40 insertions, 14 deletions

diff --git a/tests/common/mod.rs b/tests/common/mod.rs
index 18918bf..9572c5c 100644
--- a/tests/common/mod.rs
+++ b/tests/common/mod.rs
@@ -16,12 +16,13 @@ thread_local! {
     pub static RNG: RefCell<rand::rngs::ThreadRng> = RefCell::new(rand::thread_rng());
 }
 
+/*
 pub fn gen_rand_letters(i: usize) -> String {
-    //let mut rng = rand::thread_rng();
     RNG.with(|rng| {
         (0..i).map(|_| (rng.borrow_mut().gen_range(0, 26) + 'a' as u8) as char).collect()
     })
 }
+*/
 
 struct FileContentEmul(RefCell<Vec<u8>>);
 
@@ -209,6 +210,21 @@ impl PaintStrokes {
         }
         PaintStrokes(res)
     }
+
+    pub fn gen_rand<R: rand::Rng>(min_pos: u32, max_pos: u32, max_col: u32, n: usize, rng: &mut R) -> PaintStrokes {
+        let mut strokes = Self::new();
+        for _ in 0..n {
+            let mut s = 0;
+            let mut e = 0;
+            while s == e {
+                s = rng.gen_range(min_pos, max_pos);
+                e = rng.gen_range(min_pos, max_pos);
+            }
+            if s > e { std::mem::swap(&mut s, &mut e) }
+            strokes.stroke(s, e, rng.gen_range(0, max_col))
+        }
+        strokes
+    }
     
     pub fn stroke(&mut self, start: u32, end: u32, color: u32) {
         self.0.push((start, end, color))
@@ -280,12 +296,12 @@ impl Canvas {
     // TODO: allow customized scheduler
     /// Schedule to paint one position, randomly. It optionally returns a finished batch write
     /// identified by its start position of WALRingId.
-    pub fn rand_paint<R: rand::Rng>(&mut self, rng: &mut R) -> (Option<WALPos>, u32) {
-        println!("{}", self.queue.len());
+    pub fn rand_paint<R: rand::Rng>(&mut self, rng: &mut R) -> Option<(Option<WALPos>, u32)> {
+        if self.queue.is_empty() { return None }
         let idx = rng.gen_range(0, self.queue.len());
         let (pos, _) = self.queue.get_index_mut(idx).unwrap();
         let pos = *pos;
-        (self.paint(pos), pos)
+        Some((self.paint(pos), pos))
     }
 
     pub fn paint(&mut self, pos: u32) -> Option<WALPos> {
@@ -303,17 +319,27 @@ impl Canvas {
     pub fn is_same(&self, other: &Canvas) -> bool {
         self.canvas.cmp(&other.canvas) == std::cmp::Ordering::Equal
     }
+
+    pub fn print(&self, max_col: usize) {
+        for r in self.canvas.chunks(max_col) {
+            for c in r.iter() {
+                print!("{:02x} ", c & 0xff);
+            }
+            println!("");
+        }
+    }
 }
 
 #[test]
 fn test_canvas() {
-    let mut canvas1 = Canvas::new(10);
-    let mut canvas2 = Canvas::new(10);
-    let canvas3 = Canvas::new(11);
-    let mut s1 = PaintStrokes::new();
-    s1.stroke(0, 1, 1);
-    let mut s2 = PaintStrokes::new();
-    s2.stroke(1, 2, 2);
+    let mut canvas1 = Canvas::new(100);
+    let mut canvas2 = Canvas::new(100);
+    let canvas3 = Canvas::new(101);
+    let (s1, s2) = RNG.with(|rng| {
+        let rng = &mut *rng.borrow_mut();
+        (PaintStrokes::gen_rand(0, 100, 256, 2, rng),
+        PaintStrokes::gen_rand(0, 100, 256, 2, rng))
+    });
     assert!(canvas1.is_same(&canvas2));
     assert!(!canvas2.is_same(&canvas3));
     canvas1.prepaint(&s1, &0);
@@ -323,8 +349,8 @@ fn test_canvas() {
     assert!(canvas1.is_same(&canvas2));
     RNG.with(|rng| canvas1.rand_paint(&mut *rng.borrow_mut()));
     assert!(!canvas1.is_same(&canvas2));
-    RNG.with(|rng| canvas1.rand_paint(&mut *rng.borrow_mut()));
-    RNG.with(|rng| canvas2.rand_paint(&mut *rng.borrow_mut()));
-    RNG.with(|rng| canvas2.rand_paint(&mut *rng.borrow_mut()));
+    RNG.with(|rng| while let Some(_) = canvas1.rand_paint(&mut *rng.borrow_mut()) {});
+    RNG.with(|rng| while let Some(_) = canvas2.rand_paint(&mut *rng.borrow_mut()) {});
     assert!(canvas1.is_same(&canvas2));
+    canvas1.print(10);
 }