From b36d34a870213e1e220f58985b1ea149600b3822 Mon Sep 17 00:00:00 2001
From: Dmitriy Pleshevskiy <dmitriy@ideascup.me>
Date: Fri, 24 Dec 2021 23:49:04 +0300
Subject: [PATCH] tests(integration/query): add multiline test

---
 tests/query_command.rs | 36 +++++++++++++++++++++++++++++++++---
 1 file changed, 33 insertions(+), 3 deletions(-)

diff --git a/tests/query_command.rs b/tests/query_command.rs
index fcecf09..2cda5b5 100644
--- a/tests/query_command.rs
+++ b/tests/query_command.rs
@@ -23,14 +23,44 @@ fn should_find_object_by_exact_match() {
 #[test]
 fn should_find_object_by_partial_match() {
     let bucket = "query_by_partial_match";
-    let title = "Sweet Teriyaki Beef Skewers";
 
     let ingest_channel = ingest_start();
-    ingest_channel.push(COLLECTION, bucket, "1", title).unwrap();
+    ingest_channel
+        .push(COLLECTION, bucket, "1", "Sweet Teriyaki Beef Skewers")
+        .unwrap();
 
     let search_channel = search_start();
 
-    for word in title.split_whitespace() {
+    let words = ["Sweet", "Teriyaki", "Beef", "Skewers"];
+    for word in words {
+        match search_channel.query(COLLECTION, bucket, word) {
+            Ok(object_ids) => assert_eq!(object_ids, vec!["1"]),
+            Err(_) => unreachable!(),
+        }
+    }
+
+    flush_bucket(COLLECTION, bucket);
+}
+
+#[test]
+fn should_find_multiline_object_by_partial_match() {
+    let bucket = "query_multiline";
+    let multiline_text = "
+Sweet
+Teriyaki
+Beef
+Skewers
+";
+
+    let ingest_channel = ingest_start();
+    ingest_channel
+        .push(COLLECTION, bucket, "1", multiline_text)
+        .unwrap();
+
+    let search_channel = search_start();
+
+    let words = ["Sweet", "Teriyaki", "Beef", "Skewers"];
+    for word in words {
         match search_channel.query(COLLECTION, bucket, word) {
             Ok(object_ids) => assert_eq!(object_ids, vec!["1"]),
             Err(_) => unreachable!(),