Add backup flag to test get_index_data_from_database

khaledk2 · Feb 7, 2023 · c3f345c · c3f345c
1 parent de129a6
commit c3f345c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -49,7 +49,7 @@ jobs:
           #run  restore omero database
           python manage.py restore_postgresql_database
           #run indexing indexing
-          python manage.py get_index_data_from_database
+          python manage.py get_index_data_from_database -n True
           # run tests
           python -m unittest discover -s unit_tests
   upload:

diff --git a/manage.py b/manage.py
@@ -127,7 +127,12 @@ def restore_postgresql_database():
     "--resource",
     help="resource name, creating all the indexes for all the resources is the default",  # noqa
 )
-def get_index_data_from_database(resource="all"):
+@manager.option(
+    "-n",
+    "--nobackup",
+    help="if True, backup will be called ",  # noqa
+)
+def get_index_data_from_database(resource="all", nobackup=None):
     """
     insert data in Elasticsearch index for each resource
     It gets the data from postgres database server
@@ -155,7 +160,8 @@ def get_index_data_from_database(resource="all"):
         test_indexing_search_query(deep_check=False, check_studies=True)
 
     # backup the index data
-    backup_elasticsearch_data()
+    if not nobackup:
+        backup_elasticsearch_data()
 
 
 # set configurations