Add index config for test

Author: thenav56

docker-compose.yml

@@ -136,7 +136,7 @@ services:
     <<: *base_server_setup
     command: python manage.py ingest_appeal_docs
 
-  ingest_appeal_docs:
+  user_registration_reminder:
     <<: *base_server_setup
     command: python manage.py user_registration_reminder
 

main/settings.py

@@ -67,6 +67,8 @@
     PYTEST_XDIST_WORKER=(str, None),
     # Elastic-Cache
     ELASTIC_SEARCH_HOST=(str, None),
+    ELASTIC_SEARCH_INDEX=(str, 'new_index'),
+    ELASTIC_SEARCH_TEST_INDEX=(str, 'new_test_index'),  # This will be used and cleared by test
     # FTP
     GO_FTPHOST=(str, None),
     GO_FTPUSER=(str, None),
@@ -477,6 +479,8 @@
 
 # Elastic search host
 ELASTIC_SEARCH_HOST = env('ELASTIC_SEARCH_HOST')
+ELASTIC_SEARCH_INDEX = env('ELASTIC_SEARCH_INDEX')
+ELASTIC_SEARCH_TEST_INDEX = env('ELASTIC_SEARCH_TEST_INDEX')
 
 # FTP
 GO_FTPHOST = env('GO_FTPHOST')
@@ -522,7 +526,7 @@
     'default': {
         'ENGINE': 'haystack.backends.elasticsearch7_backend.Elasticsearch7SearchEngine',
         'URL': ELASTIC_SEARCH_HOST,
-        'INDEX_NAME': 'new_index',
+        'INDEX_NAME': ELASTIC_SEARCH_INDEX,
     },
 }
 

main/test_case.py

@@ -1,5 +1,6 @@
 import datetime
 import pytz
+import haystack
 from unittest import mock
 
 import snapshottest.django as django_snapshottest
@@ -9,6 +10,7 @@
 from rest_framework import test, status
 
 from django.core import management
+from django.conf import settings
 from django.contrib.auth.models import Permission
 from django.contrib.contenttypes.models import ContentType
 from django.db import DEFAULT_DB_ALIAS, connections
@@ -19,16 +21,32 @@
 
 from lang.translation import AmazonTranslate
 
+# XXX: Will not support if test are run in parallel
+TEST_HAYSTACK_CONNECTIONS = {
+    'default': {
+        'ENGINE': 'haystack.backends.elasticsearch7_backend.Elasticsearch7SearchEngine',
+        'URL': settings.ELASTIC_SEARCH_HOST,
+        'INDEX_NAME': settings.ELASTIC_SEARCH_TEST_INDEX,
+    },
+}
+
 
 class GoAPITestMixin():
     """
     Base TestCase
     """
     client_class = test.APIClient
 
-    def setUp(self):
-        super().setUp()
+    def set_up_haystack(self):
+        haystack.connections.reload('default')
 
+    def tear_down_haystack(self):
+        haystack.connections.reload('default')
+        backend = haystack.connections['default'].get_backend()
+        assert backend.index_name == settings.ELASTIC_SEARCH_TEST_INDEX
+        backend.clear()
+
+    def set_up_seed(self):
         self.root_user = UserFactory.create(
             username='[email protected]',
             first_name='Root',
@@ -127,24 +145,42 @@ def capture_on_commit_callbacks(cls, *, using=DEFAULT_DB_ALIAS, execute=False):
         return CaptureOnCommitCallbacksContext(using=using, execute=execute)
 
 
-@override_settings(CELERY_TASK_ALWAYS_EAGER=True)
-@override_settings(SUSPEND_SIGNALS=True)
+@override_settings(
+    CELERY_TASK_ALWAYS_EAGER=True,
+    SUSPEND_SIGNALS=True,
+    HAYSTACK_CONNECTIONS=TEST_HAYSTACK_CONNECTIONS,
+)
 class APITestCase(GoAPITestMixin, test.APITestCase):
-    pass
+    def setUp(self):
+        self.set_up_haystack()
+        super().setUp()
+        self.set_up_seed()
+
+    def tearDown(self):
+        super().tear_down_haystack()
+        super().tearDown()
 
 
-@override_settings(SUSPEND_SIGNALS=True)
+@override_settings(
+    SUSPEND_SIGNALS=True,
+    HAYSTACK_CONNECTIONS=TEST_HAYSTACK_CONNECTIONS,
+)
 class SnapshotTestCase(GoAPITestMixin, django_snapshottest.TestCase):
     maxDiff = None
 
     def setUp(self):
+        self.set_up_haystack()
+        super().setUp()
         management.call_command("flush", "--no-input")
         factory.random.reseed_random(42)
+        self.set_up_seed()
         self.patcher = mock.patch('django.utils.timezone.now')
         self.patcher.start().return_value = datetime.datetime(2008, 1, 1, 0, 0, 0, 123456, tzinfo=pytz.UTC)
 
     def tearDown(self):
         self.patcher.stop()
+        super().tear_down_haystack()
+        super().tearDown()
 
 
 class CaptureOnCommitCallbacksContext: