From 941fec3fe519b86c2943a575f765514fc755d8a1 Mon Sep 17 00:00:00 2001
From: Will Stott <willstott101@gmail.com>
Date: Tue, 2 Apr 2019 20:42:09 +0000
Subject: [PATCH] Add some tests to Sink

---
 src/sink.rs | 73 ++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 69 insertions(+), 4 deletions(-)

diff --git a/src/sink.rs b/src/sink.rs
index 1c394eb..365c449 100644
--- a/src/sink.rs
+++ b/src/sink.rs
@@ -33,13 +33,20 @@ struct Controls {
 }
 
 impl Sink {
-    /// Builds a new `Sink`.
+    /// Builds a new `Sink`, beginning playback on a Device.
     #[inline]
     pub fn new(device: &Device) -> Sink {
-        let (queue_tx, queue_rx) = queue::queue(true);
+        let (sink, queue_rx) = Sink::new_idle();
         play_raw(device, queue_rx);
+        sink
+    }
 
-        Sink {
+    /// Builds a new `Sink`.
+    #[inline]
+    pub fn new_idle() -> (Sink, queue::SourcesQueueOutput<f32>) {
+        let (queue_tx, queue_rx) = queue::queue(true);
+
+        let sink = Sink {
             queue_tx: queue_tx,
             sleep_until_end: Mutex::new(None),
             controls: Arc::new(Controls {
@@ -49,7 +56,8 @@ impl Sink {
             }),
             sound_count: Arc::new(AtomicUsize::new(0)),
             detached: false,
-        }
+        };
+        (sink, queue_rx)
     }
 
     /// Appends a sound to the queue of sounds to play.
@@ -168,3 +176,60 @@ impl Drop for Sink {
         }
     }
 }
+
+
+#[cfg(test)]
+mod tests {
+    use buffer::SamplesBuffer;
+    use source::Source;
+    use sink::Sink;
+
+    #[test]
+    fn test_pause_and_stop() {
+        let (sink, mut queue_rx) = Sink::new_idle();
+
+        // assert_eq!(queue_rx.next(), Some(0.0));
+
+        let v = vec![10i16, -10, 20, -20, 30, -30];
+
+        // Low rate to ensure immediate control.
+        sink.append(SamplesBuffer::new(1, 1, v.clone()));
+        let mut src = SamplesBuffer::new(1, 1, v.clone()).convert_samples();
+
+        assert_eq!(queue_rx.next(), src.next());
+        assert_eq!(queue_rx.next(), src.next());
+
+        sink.pause();
+
+        assert_eq!(queue_rx.next(), Some(0.0));
+
+        sink.play();
+
+        assert_eq!(queue_rx.next(), src.next());
+        assert_eq!(queue_rx.next(), src.next());
+
+        sink.stop();
+
+        assert_eq!(queue_rx.next(), Some(0.0));
+
+        assert_eq!(sink.empty(), true);
+    }
+
+    #[test]
+    fn test_volume() {
+        let (sink, mut queue_rx) = Sink::new_idle();
+
+        let v = vec![10i16, -10, 20, -20, 30, -30];
+
+        // High rate to avoid immediate control.
+        sink.append(SamplesBuffer::new(2, 44100, v.clone()));
+        let src = SamplesBuffer::new(2, 44100, v.clone()).convert_samples();
+
+        let mut src = src.amplify(0.5);
+        sink.set_volume(0.5);
+
+        for _ in 0..v.len() {
+            assert_eq!(queue_rx.next(), src.next());
+        }
+    }
+}