Update the tests

Signed-off-by: Olga Bulat <[email protected]>
WordPress · Nov 15, 2023 · 1d60c9d · 1d60c9d
1 parent e7fb1a3
commit 1d60c9d
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/api/test/unit/controllers/test_search_controller_search_query.py b/api/test/unit/controllers/test_search_controller_search_query.py
@@ -3,6 +3,7 @@
 import pytest
 
 from api.controllers import search_controller
+from api.controllers.search_controller import DEFAULT_SQS_FLAGS
 
 
 pytestmark = pytest.mark.django_db
@@ -62,6 +63,7 @@ def test_create_search_query_q_search_no_filters(media_type_config):
                     "default_operator": "AND",
                     "fields": ["title", "description", "tags.name"],
                     "query": "cat",
+                    "flags": DEFAULT_SQS_FLAGS,
                 }
             }
         ],
@@ -71,6 +73,7 @@ def test_create_search_query_q_search_no_filters(media_type_config):
                     "boost": 10000,
                     "fields": ["title"],
                     "query": "cat",
+                    "flags": DEFAULT_SQS_FLAGS,
                 }
             },
             {"rank_feature": {"boost": 10000, "field": "standardized_popularity"}},
@@ -95,6 +98,7 @@ def test_create_search_query_q_search_with_quotes_adds_exact_suffix(media_type_c
                     "fields": ["title", "description", "tags.name"],
                     "query": '"The cutest cat"',
                     "quote_field_suffix": ".exact",
+                    "flags": DEFAULT_SQS_FLAGS,
                 }
             }
         ],
@@ -104,6 +108,7 @@ def test_create_search_query_q_search_with_quotes_adds_exact_suffix(media_type_c
                     "boost": 10000,
                     "fields": ["title"],
                     "query": "The cutest cat",
+                    "flags": DEFAULT_SQS_FLAGS,
                 }
             },
             {"rank_feature": {"boost": 10000, "field": "standardized_popularity"}},
@@ -143,6 +148,7 @@ def test_create_search_query_q_search_with_filters(image_media_type_config):
                     "default_operator": "AND",
                     "fields": ["title", "description", "tags.name"],
                     "query": "cat",
+                    "flags": DEFAULT_SQS_FLAGS,
                 }
             }
         ],
@@ -152,6 +158,7 @@ def test_create_search_query_q_search_with_filters(image_media_type_config):
                     "boost": 10000,
                     "fields": ["title"],
                     "query": "cat",
+                    "flags": DEFAULT_SQS_FLAGS,
                 }
             },
             {"rank_feature": {"boost": 10000, "field": "standardized_popularity"}},
@@ -179,10 +186,23 @@ def test_create_search_query_non_q_query(image_media_type_config):
                 "simple_query_string": {
                     "fields": ["creator"],
                     "query": "Artist From Openverse",
+                    "flags": DEFAULT_SQS_FLAGS,
+                }
+            },
+            {
+                "simple_query_string": {
+                    "fields": ["title"],
+                    "query": "kitten🐱",
+                    "flags": DEFAULT_SQS_FLAGS,
+                }
+            },
+            {
+                "simple_query_string": {
+                    "fields": ["tags.name"],
+                    "query": "cute",
+                    "flags": DEFAULT_SQS_FLAGS,
                 }
             },
-            {"simple_query_string": {"fields": ["title"], "query": "kitten🐱"}},
-            {"simple_query_string": {"fields": ["tags.name"], "query": "cute"}},
         ],
         "should": [
             {"rank_feature": {"boost": 10000, "field": "standardized_popularity"}},