Add SearchQueryString tests

Author: WaVEV

tests/atlas_search_/test_search.py

@@ -20,6 +20,7 @@
     SearchIn,
     SearchMoreLikeThis,
     SearchPhrase,
+    SearchQueryString,
     SearchRange,
     SearchRegex,
     SearchScoreOption,
@@ -359,6 +360,68 @@ def test_str_returns_expected_format(self):
         )
 
 
+@skipUnlessDBFeature("supports_atlas_search")
+class SearchQueryStringTests(SearchUtilsMixin):
+    @classmethod
+    def setUpClass(cls):
+        cls.create_search_index(
+            Article,
+            "query_string_index",
+            {
+                "mappings": {
+                    "dynamic": False,
+                    "fields": {
+                        "headline": {"type": "string"},
+                        "body": {"type": "string"},
+                    },
+                }
+            },
+        )
+
+    def setUp(self):
+        self.mars_mission = Article.objects.create(
+            headline="space exploration",
+            body="NASA launches a new mission to Mars",
+            number=1,
+        )
+        self.exoplanet = Article.objects.create(
+            headline="space exploration",
+            body="Astronomers discover exoplanets orbiting distant stars",
+            number=2,
+        )
+        Article.objects.create(
+            headline="other news",
+            body="Local team wins championship",
+            number=3,
+        )
+
+    def test_search_query_string_basic(self):
+        qs = Article.objects.annotate(
+            score=SearchQueryString(path="headline", query="space AND body:Mars")
+        )
+        self.assertCountEqual(qs, [self.mars_mission])
+
+    def test_search_query_string_with_score(self):
+        constant_score = SearchScoreOption({"constant": {"value": 10}})
+        qs = Article.objects.annotate(
+            score=SearchQueryString(
+                path="headline",
+                query="space AND body:exoplanets",
+                score=constant_score,
+            )
+        )
+        self.assertCountEqual(qs, [self.exoplanet])
+        self.assertAlmostEqual(qs.first().score, 10.0, places=2)
+
+    def test_str_returns_expected_format(self):
+        score = SearchScoreOption({"constant": {"value": 10}})
+        se = SearchQueryString(path="body", query="space AND body:exoplanets", score=score)
+        self.assertEqual(
+            str(se),
+            f"SearchQueryString(path='body', query='space AND body:exoplanets, score={score})>",
+        )
+
+
 class SearchRangeTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):