Merge pull request #191 from CDC-IITDH/upgrade-selenium-test

added auto clean test files and test objects

CDC_Backend/APIs/companyViews.py

@@ -235,14 +235,14 @@ def addPlacement(request):
 
     except ValueError as e:
         store_all_files(request)
-        exception_email(data)
+        #exception_email(data)
         logger.warning("ValueError in addPlacement: " + str(e))
         logger.warning(traceback.format_exc())
         return Response({'action': "Add Placement", 'message': str(e)},
                         status=status.HTTP_400_BAD_REQUEST)
     except:
         store_all_files(request)
-        exception_email(data)
+        #exception_email(data)
         logger.warning("Add New Placement: " + str(sys.exc_info()))
         logger.warning(traceback.format_exc())
         return Response({'action': "Add Placement", 'message': "Something went wrong"},

CDC_Backend/APIs/cron.py

@@ -0,0 +1,63 @@
+from .models import *
+from .utils import *
+import shutil
+import logging
+import traceback
+
+logger=logging.getLogger('db')
+def clean_up_tests():
+    # Delete all the test internships and placements created
+    try:
+        internships= Internship.objects.filter(company_name="test company",email="notifications@cdc-iitdh.tech")
+        hits_internship=len(internships)
+        for internship in internships:
+            #count number of file
+            files = os.listdir(STORAGE_DESTINATION_COMPANY_ATTACHMENTS+internship.id)
+            if len(files) == 4:
+                print("working fine")
+            else:
+                print("not working fine")
+                logger.error("files submitted in inf are not getting stored for test case"+internship.description)
+
+            #remove folder from the server
+            print("removing folder ",STORAGE_DESTINATION_COMPANY_ATTACHMENTS+internship.id)
+            shutil.rmtree(STORAGE_DESTINATION_COMPANY_ATTACHMENTS+internship.id)
+            
+            
+            internship.delete()
+        
+        placements= Placement.objects.filter(company_name="test company",email="notifications@cdc-iitdh.tech")
+        hits_placement=len(placements)
+        for placement in placements:
+            #count number of file
+            files = os.listdir(STORAGE_DESTINATION_COMPANY_ATTACHMENTS+placement.id)
+            if len(files) == 4:
+                print("working fine")
+            else:
+                print("not working fine")
+                logger.error("files submitted in inf are not getting stored for test case"+placement.description)
+
+            #remove folder from the server
+            print("removing folder ",STORAGE_DESTINATION_COMPANY_ATTACHMENTS+internship.id)
+            shutil.rmtree(STORAGE_DESTINATION_COMPANY_ATTACHMENTS+placement.id)
+            
+            
+            placement.delete()
+        
+        if hits_internship >= 6:
+            print("all hits are working fine")
+        else:
+            print("some hits are not working fine")
+            logger.error("some test hits are not working fine for internship")
+        
+        if hits_placement >= 6:
+            print("all hits are working fine")
+        else:
+            print("some hits are not working fine")
+            logger.error("some test hits are not working fine for placement")
+
+    except :
+        logger.error("error in clean up function")
+        logger.error(traceback.format_exc())
+
+    

CDC_Backend/CDC_Backend/settings.py

@@ -47,7 +47,8 @@ INSTALLED_APPS = [
     'background_task',
     'simple_history',
     'import_export',
-    'django_extensions'
+    'django_extensions',
+    'django_crontab',
 ]
 
 MIDDLEWARE = [
@@ -203,3 +204,7 @@ LOGGING = {
 }
 
 # django_heroku.settings(locals())
+
+CRONJOBS = [
+    ('0 8,20 * * *', 'APIs.cron.clean_up_tests')
+]

requirements.txt

@@ -13,6 +13,7 @@ Django==3.2.13
 django-background-tasks==1.2.5
 django-compat==1.0.15
 django-cors-headers==3.11.0
+django-crontab==0.7.1
 django-db-logger==0.1.12
 django-extensions==3.2.1
 django-import-export==2.8.0

setup.sh

@@ -8,7 +8,7 @@ python3 manage.py migrate
 echo "Migrations complete"
 
 python3 manage.py collectstatic --noinput
-
+python3 manage.py crontab add
 DIR="./Storage"
 if [ -d "$DIR" ]; then
   ### Take action if $DIR exists ###