From 443aa1732905264161b953c33cb944be2f5142e0 Mon Sep 17 00:00:00 2001
From: Hardik Prajapati <hardikpnsp@gmail.com>
Date: Wed, 7 Apr 2021 23:10:48 +0530
Subject: [PATCH] AAdded failing test for tie scenario in topk

---
 src/collector/facet_collector.rs | 39 ++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/src/collector/facet_collector.rs b/src/collector/facet_collector.rs
index 0e6432d5f..077da6b0f 100644
--- a/src/collector/facet_collector.rs
+++ b/src/collector/facet_collector.rs
@@ -657,6 +657,45 @@ mod tests {
             );
         }
     }
+
+    #[test]
+    fn test_facet_collector_topk_tie_break() {
+        let mut schema_builder = Schema::builder();
+        let facet_field = schema_builder.add_facet_field("facet", INDEXED);
+        let schema = schema_builder.build();
+        let index = Index::create_in_ram(schema);
+
+        let docs: Vec<Document> = vec![("b", 2), ("a", 2), ("c", 4)]
+            .into_iter()
+            .flat_map(|(c, count)| {
+                let facet = Facet::from(&format!("/facet/{}", c));
+                let doc = doc!(facet_field => facet);
+                iter::repeat(doc).take(count)
+            })
+            .collect();
+
+        let mut index_writer = index.writer_for_tests().unwrap();
+        for doc in docs {
+            index_writer.add_document(doc);
+        }
+        index_writer.commit().unwrap();
+        let searcher = index.reader().unwrap().searcher();
+
+        let mut facet_collector = FacetCollector::for_field(facet_field);
+        facet_collector.add_facet("/facet");
+        let counts: FacetCounts = searcher.search(&AllQuery, &facet_collector).unwrap();
+
+        {
+            let facets: Vec<(&Facet, u64)> = counts.top_k("/facet", 2);
+            assert_eq!(
+                facets,
+                vec![
+                    (&Facet::from("/facet/c"), 4),
+                    (&Facet::from("/facet/a"), 2),
+                ]
+            );
+        }
+    }
 }
 
 #[cfg(all(test, feature = "unstable"))]