From a6b2d941194aa07f1a03b197eafb4caa0cc62dec Mon Sep 17 00:00:00 2001
From: Paul Masurel <paul.masurel@gmail.com>
Date: Fri, 29 Jul 2016 18:38:18 +0900
Subject: [PATCH] Added unit tests

---
 src/collector/chained_collector.rs |  8 ++---
 src/collector/top_collector.rs     | 51 ++++++++++++++++++++++++++++++
 src/lib.rs                         | 10 ++++++
 3 files changed, 65 insertions(+), 4 deletions(-)

diff --git a/src/collector/chained_collector.rs b/src/collector/chained_collector.rs
index f580b03ea..da7bd24ad 100644
--- a/src/collector/chained_collector.rs
+++ b/src/collector/chained_collector.rs
@@ -81,12 +81,12 @@ mod tests {
         let mut top_collector = TopCollector::with_limit(2);
         let mut count_collector = CountCollector::new();
         {
-            let mut chain = collector_chain()
+            let mut collectors = chain()
                 .add(&mut top_collector)
                 .add(&mut count_collector);
-            chained_collector.collect(ScoredDoc(0.2, 1));
-            chained_collector.collect(ScoredDoc(0.1, 2));
-            chained_collector.collect(ScoredDoc(0.5, 3));
+            collectors.collect(ScoredDoc(0.2, 1));
+            collectors.collect(ScoredDoc(0.1, 2));
+            collectors.collect(ScoredDoc(0.5, 3));
         }
         assert_eq!(count_collector.count(), 3);
         assert!(top_collector.at_capacity());
diff --git a/src/collector/top_collector.rs b/src/collector/top_collector.rs
index 388e3c684..d8e144cf4 100644
--- a/src/collector/top_collector.rs
+++ b/src/collector/top_collector.rs
@@ -103,3 +103,54 @@ impl Collector for TopCollector {
 
     }
 }
+
+
+#[cfg(test)]
+mod tests {
+
+    use super::*;
+    use ScoredDoc;
+    use DocId;
+    use Score;
+    use collector::Collector;
+
+    #[test]
+    fn test_top_collector_not_at_capacity() {
+        let mut top_collector = TopCollector::with_limit(4);
+        top_collector.collect(ScoredDoc(0.8, 1));
+        top_collector.collect(ScoredDoc(0.2, 3));
+        top_collector.collect(ScoredDoc(0.3, 5));
+        assert!(!top_collector.at_capacity());
+        let score_docs: Vec<(Score, DocId)> = top_collector.score_docs()
+            .into_iter()
+            .map(|(score, doc_address)| (score, doc_address.doc()))
+            .collect();
+        assert_eq!(score_docs, vec!(
+            (0.8, 1), (0.3, 5), (0.2, 3),
+        ));
+    }
+
+    #[test]
+    fn test_top_collector_at_capacity() {
+        let mut top_collector = TopCollector::with_limit(4);
+        top_collector.collect(ScoredDoc(0.8, 1));
+        top_collector.collect(ScoredDoc(0.2, 3));
+        top_collector.collect(ScoredDoc(0.3, 5));
+        top_collector.collect(ScoredDoc(0.9, 7));
+        top_collector.collect(ScoredDoc(-0.2, 9));
+        assert!(top_collector.at_capacity());
+        let score_docs: Vec<(Score, DocId)> = top_collector.score_docs()
+            .into_iter()
+            .map(|(score, doc_address)| (score, doc_address.doc()))
+            .collect();
+        assert_eq!(score_docs, vec!(
+            (0.9, 7), (0.8, 1), (0.3, 5), (0.2, 3)
+        ));
+    }
+
+    #[test]
+    #[should_panic]
+    fn test_top_0() {
+        TopCollector::with_limit(0);
+    }
+}
\ No newline at end of file
diff --git a/src/lib.rs b/src/lib.rs
index da298084e..2f0669ec8 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -72,6 +72,16 @@ pub type SegmentLocalId = u32;
 #[derive(Debug, Clone, Copy)]
 pub struct DocAddress(pub SegmentLocalId, pub DocId);
 
+impl DocAddress {
+    pub fn segment_ord(&self,) -> SegmentLocalId {
+        self.0
+    }
+
+    pub fn doc(&self,) -> DocId {
+        self.1
+    }
+}
+
 #[derive(Clone, Copy)]
 pub struct ScoredDoc(Score, DocId);