test(mpsc): tests for draining after tx closes (#12)

Signed-off-by: Eliza Weisman <[email protected]>

src/mpsc/tests/mpsc_async.rs

@@ -256,3 +256,27 @@ fn tx_close_wakes() {
         consumer.join().unwrap();
     });
 }
+
+#[test]
+fn tx_close_drains_queue() {
+    const LEN: usize = 4;
+    loom::model(|| {
+        let (tx, rx) = channel(ThingBuf::new(LEN));
+        let producer = thread::spawn(move || {
+            future::block_on(async move {
+                for i in 0..LEN {
+                    tx.send(i).await.unwrap();
+                }
+            })
+        });
+
+        future::block_on(async move {
+            for i in 0..LEN {
+                assert_eq!(rx.recv().await, Some(i))
+            }
+        });
+
+
+        producer.join().unwrap();
+    });
+}

src/mpsc/tests/mpsc_sync.rs

@@ -258,3 +258,22 @@ fn tx_close_wakes() {
         consumer.join().unwrap();
     });
 }
+
+#[test]
+fn tx_close_drains_queue() {
+    const LEN: usize = 4;
+    loom::model(|| {
+        let (tx, rx) = sync::channel(ThingBuf::new(LEN));
+        let producer = thread::spawn(move || {
+            for i in 0..LEN {
+                tx.send(i).unwrap();
+            }
+        });
+
+        for i in 0..LEN {
+            assert_eq!(rx.recv(), Some(i))
+        }
+
+        producer.join().unwrap();
+    });
+}

tests/mpsc_sync.rs

@@ -49,3 +49,24 @@ fn basically_works() {
         }
     }
 }
+
+#[test]
+fn tx_close_drains_queue() {
+    const LEN: usize = 4;
+    for i in 0..10000 {
+        println!("\n\n--- iteration {} ---\n\n", i);
+
+        let (tx, rx) = sync::channel(ThingBuf::new(LEN));
+        let producer = thread::spawn(move || {
+            for i in 0..LEN {
+                tx.send(i).unwrap();
+            }
+        });
+
+        for i in 0..LEN {
+            assert_eq!(rx.recv(), Some(i))
+        }
+
+        producer.join().unwrap();
+    }
+}