tests for threads search, including fulltext search

misago/threads/tests/test_search.py

@@ -0,0 +1,161 @@
+from django.urls import reverse
+
+from misago.categories.models import Category
+from misago.users.testutils import AuthenticatedUserTestCase
+
+from .. import testutils
+
+
+class SearchApiTests(AuthenticatedUserTestCase):
+    def setUp(self):
+        super(SearchApiTests, self).setUp()
+
+        self.category = Category.objects.get(slug='first-category')
+
+        self.api_link = reverse('misago:api:search')
+
+    def test_no_query(self):
+        """api handles no search query"""
+        response = self.client.get(self.api_link)
+        self.assertEqual(response.status_code, 200)
+
+        reponse_json = response.json()
+        self.assertIn('threads', [p['id'] for p in reponse_json])
+
+        for provider in reponse_json:
+            if provider['id'] == 'threads':
+                self.assertEqual(provider['results']['results'], [])
+
+    def test_empty_query(self):
+        """api handles empty search query"""
+        response = self.client.get('%s?q=' % self.api_link)
+        self.assertEqual(response.status_code, 200)
+
+        reponse_json = response.json()
+        self.assertIn('threads', [p['id'] for p in reponse_json])
+
+        for provider in reponse_json:
+            if provider['id'] == 'threads':
+                self.assertEqual(provider['results']['results'], [])
+
+    def test_short_query(self):
+        """api handles short search query"""
+        thread = testutils.post_thread(self.category)
+        post = testutils.reply_thread(thread, message="Lorem ipsum dolor.")
+
+        response = self.client.get('%s?q=ip' % self.api_link)
+        self.assertEqual(response.status_code, 200)
+
+        reponse_json = response.json()
+        self.assertIn('threads', [p['id'] for p in reponse_json])
+
+        for provider in reponse_json:
+            if provider['id'] == 'threads':
+                self.assertEqual(provider['results']['results'], [])
+
+    def test_wrong_query(self):
+        """api handles query miss"""
+        thread = testutils.post_thread(self.category)
+        post = testutils.reply_thread(thread, message="Lorem ipsum dolor.")
+
+        response = self.client.get('%s?q=elit' % self.api_link)
+        self.assertEqual(response.status_code, 200)
+
+        reponse_json = response.json()
+        self.assertIn('threads', [p['id'] for p in reponse_json])
+
+        for provider in reponse_json:
+            if provider['id'] == 'threads':
+                self.assertEqual(provider['results']['results'], [])
+
+    def test_hidden_post(self):
+        """hidden posts are extempt from search"""
+        thread = testutils.post_thread(self.category)
+        post = testutils.reply_thread(
+            thread, message="Lorem ipsum dolor.", is_hidden=True)
+
+        response = self.client.get('%s?q=ipsum' % self.api_link)
+        self.assertEqual(response.status_code, 200)
+
+        reponse_json = response.json()
+        self.assertIn('threads', [p['id'] for p in reponse_json])
+
+        for provider in reponse_json:
+            if provider['id'] == 'threads':
+                self.assertEqual(provider['results']['results'], [])
+
+    def test_unapproved_post(self):
+        """unapproves posts are extempt from search"""
+        thread = testutils.post_thread(self.category)
+        post = testutils.reply_thread(
+            thread, message="Lorem ipsum dolor.", is_unapproved=True)
+
+        response = self.client.get('%s?q=ipsum' % self.api_link)
+        self.assertEqual(response.status_code, 200)
+
+        reponse_json = response.json()
+        self.assertIn('threads', [p['id'] for p in reponse_json])
+
+        for provider in reponse_json:
+            if provider['id'] == 'threads':
+                self.assertEqual(provider['results']['results'], [])
+
+    def test_query(self):
+        """api handles search query"""
+        thread = testutils.post_thread(self.category)
+        post = testutils.reply_thread(thread, message="Lorem ipsum dolor.")
+
+        response = self.client.get('%s?q=ipsum' % self.api_link)
+        self.assertEqual(response.status_code, 200)
+
+        reponse_json = response.json()
+        self.assertIn('threads', [p['id'] for p in reponse_json])
+
+        for provider in reponse_json:
+            if provider['id'] == 'threads':
+                results = provider['results']['results']
+                self.assertEqual(len(results), 1)
+                self.assertEqual(results[0]['id'], post.id)
+
+    def test_thread_title_search(self):
+        """api searches threads by title"""
+        thread = testutils.post_thread(self.category, title="Atmosphere of mars")
+        testutils.reply_thread(thread, message="Lorem ipsum dolor.")
+
+        response = self.client.get('%s?q=mars atmosphere' % self.api_link)
+        self.assertEqual(response.status_code, 200)
+
+        reponse_json = response.json()
+        self.assertIn('threads', [p['id'] for p in reponse_json])
+
+        for provider in reponse_json:
+            if provider['id'] == 'threads':
+                results = provider['results']['results']
+                self.assertEqual(len(results), 1)
+                self.assertEqual(results[0]['id'], thread.first_post_id)
+
+    def test_complex_query(self):
+        """api handles complex query that uses fulltext search facilities"""
+        thread = testutils.post_thread(self.category)
+        post = testutils.reply_thread(thread, message="Atmosphere of Mars")
+
+        response = self.client.get('%s?q=Mars atmosphere' % self.api_link)
+        self.assertEqual(response.status_code, 200)
+
+        reponse_json = response.json()
+        self.assertIn('threads', [p['id'] for p in reponse_json])
+
+        for provider in reponse_json:
+            if provider['id'] == 'threads':
+                results = provider['results']['results']
+                self.assertEqual(len(results), 1)
+                self.assertEqual(results[0]['id'], post.id)
+
+
+class SearchProviderApiTests(SearchApiTests):
+    def setUp(self):
+        super(SearchProviderApiTests, self).setUp()
+
+        self.api_link = reverse('misago:api:search', kwargs={
+            'search_provider': 'threads'
+        })