Update the tests

Signed-off-by: Olga Bulat <obulat@gmail.com>
WordPress · Nov 26, 2023 · e8cb711 · e8cb711
1 parent 6d0c6b5
commit e8cb711
Showing 1 changed file with 22 additions and 3 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
@@ -4,7 +4,7 @@
 from elasticsearch_dsl import Q
 
 from api.controllers import search_controller
-from api.controllers.search_controller import FILTERED_PROVIDERS_CACHE_KEY
+from api.controllers.search_controller import DEFAULT_SQS_FLAGS, FILTERED_PROVIDERS_CACHE_KEY
 
 
 pytestmark = pytest.mark.django_db
@@ -63,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,
                 }
             }
         ],
@@ -72,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"}},
@@ -96,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": ".raw",
+                    "flags": DEFAULT_SQS_FLAGS,
                 }
             }
         ],
@@ -105,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"}},
@@ -144,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,
                 }
             }
         ],
@@ -153,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"}},
@@ -180,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"}},