add test

Author: dojutsu-user

readthedocs/search/tests/test_search_tasks.py

@@ -1,5 +1,6 @@
 """Tests for search tasks."""
 
+import mock
 import pytest
 
 from django.urls import reverse
@@ -43,6 +44,48 @@ def test_search_query_recorded_when_results_not_zero(self, api_client):
             SearchQuery.objects.all().count() == 1
         ), 'there should be 1 obj since a search is made which returns one result.'
 
+    def test_partial_queries_are_not_recorded(self, api_client):
+        """Test if partial queries are not recorded."""
+
+        assert (
+            SearchQuery.objects.all().count() == 0
+        ), 'no SearchQuery should be present if there is no search made.'
+
+        time = timezone.now()
+        search_params = { 'q': 'stack', 'project': 'docs', 'version': 'latest' }
+
+        with mock.patch('django.utils.timezone.now') as test_time:
+            test_time.return_value = time
+            resp = api_client.get(self.url, search_params)
+            assert resp.status_code, 200
+
+        assert (
+            SearchQuery.objects.all().count() == 1
+        ), 'one SearchQuery should be present'
+
+        # update the time and the search query and make another search request
+        time = time + timezone.timedelta(seconds=2)
+        search_params['q'] = 'stack over'
+        with mock.patch('django.utils.timezone.now') as test_time:
+            test_time.return_value = time
+            resp = api_client.get(self.url, search_params)
+            assert resp.status_code, 200
+
+        # update the time and the search query and make another search request
+        time = time + timezone.timedelta(seconds=2)
+        search_params['q'] = 'stack overflow'
+        with mock.patch('django.utils.timezone.now') as test_time:
+            test_time.return_value = time
+            resp = api_client.get(self.url, search_params)
+            assert resp.status_code, 200
+
+        assert (
+            SearchQuery.objects.all().count() == 1
+        ), 'one SearchQuery should be present'
+        assert (
+            SearchQuery.objects.all().first().query == 'stack overflow'
+        ), 'one SearchQuery should be there because partial queries gets updated'
+
     def test_search_query_not_recorded_when_results_are_zero(self, api_client):
         """Test that search queries are not recorded when they have zero results."""