diff --git a/solution/backend/content_search/tests/test_search.py b/solution/backend/content_search/tests/test_search.py index bced653c4..dac24a3c2 100644 --- a/solution/backend/content_search/tests/test_search.py +++ b/solution/backend/content_search/tests/test_search.py @@ -237,18 +237,27 @@ def test_rank_sorting(self): # This test ensures that reg text is included in the search results, and only if show_regulations is set to true or not set. def test_reg_text_search(self): self.login() + + # Exclude resources but this search will not match any reg text response = self.client.get("/v3/content-search/?q=fire&show_internal=false&show_public=false") data = get_paginated_data(response) self.assertEqual(data['count'], 0) + # Reg text should not show up for this search response = self.client.get("/v3/content-search/?q=fire") data = get_paginated_data(response) self.assertEqual(data['count'], 2) + # Ensure that excluding reg text works + response = self.client.get("/v3/content-search/?q=federal&show_regulations=false") + data = get_paginated_data(response) + self.assertEqual(data['count'], 1) + + # Ensure that reg text is included in the search results and the structure is correct response = self.client.get("/v3/content-search/?q=federal") data = get_paginated_data(response) self.assertEqual(data['count'], 3) - self.assertTrue(any([i["reg_text"] is not None for i in data["results"]])) + self.assertTrue(any([i["reg_text"] for i in data["results"]])) for i in [i for i in data["results"] if i["reg_text"] is not None]: self.assertEqual(i["reg_text"]["title"], 42) self.assertEqual(i["reg_text"]["date"], "2023-01-01")