[tests:study] Add tests for CoCom & CoLic study implementation

Signed-off-by: inishchith <[email protected]>

Author: inishchith

grimoire_elk/enriched/cocom.py

@@ -181,7 +181,7 @@ def enrich_repo_analysis(self, ocean_backend, enrich_backend, no_incremental=Fal
                              out_index="cocom_enrich_graal_repo", interval_months=[3],
                              date_field="grimoire_creation_date"):
 
-        logger.info("Doing enrich_repository_analysis study for index {}"
+        logger.info("[cocom] Starting enrich_repository_analysis study {}"
                     .format(self.elastic.anonymize_url(self.elastic.index_url)))
 
         es_in = ES([enrich_backend.elastic_url], retry_on_timeout=True, timeout=100,
@@ -260,3 +260,5 @@ def enrich_repo_analysis(self, ocean_backend, enrich_backend, no_incremental=Fal
                     logger.error("%s/%s missing items for Graal CoCom Analysis Study", str(missing), str(num_items))
                 else:
                     logger.info("%s items inserted for Graal CoCom Analysis Study", str(num_items))
+
+        logger.info("[cocom] Ending enrich_repository_analysis study")

grimoire_elk/enriched/colic.py

@@ -263,7 +263,7 @@ def enrich_colic_analysis(self, ocean_backend, enrich_backend, no_incremental=Fa
                               out_index="colic_enrich_graal_repo", interval_months=[3],
                               date_field="grimoire_creation_date"):
 
-        logger.info("Doing enrich_colic_analysis study for index {}"
+        logger.info("[colic] Starting enrich_colic_analysis study {}"
                     .format(self.elastic.anonymize_url(self.elastic.index_url)))
 
         es_in = ES([enrich_backend.elastic_url], retry_on_timeout=True, timeout=100,
@@ -340,3 +340,5 @@ def enrich_colic_analysis(self, ocean_backend, enrich_backend, no_incremental=Fa
                     logger.error("%s/%s missing items for Graal CoLic Analysis Study", str(missing), str(num_items))
                 else:
                     logger.info("%s items inserted for Graal CoLic Analysis Study", str(num_items))
+
+        logger.info("[colic] Ending enrich_colic_analysis study")

tests/test_cocom.py

@@ -23,6 +23,7 @@
 import unittest
 
 from base import TestBaseBackend
+from grimoire_elk.enriched.enrich import logger
 
 
 HEADER_JSON = {"Content-Type": "application/json"}
@@ -121,6 +122,21 @@ def test_raw_to_enrich(self):
         self.assertEqual(eitem["blanks_per_loc"], None)
         self.assertEqual(eitem["loc_per_function"], None)
 
+    def test_cocom_analysis_study(self):
+        """ Test that the cocom analysis study works correctly """
+
+        study, ocean_backend, enrich_backend = self._test_study('enrich_repo_analysis')
+
+        with self.assertLogs(logger, level='INFO') as cm:
+
+            if study.__name__ == "enrich_repo_analysis":
+                study(ocean_backend, enrich_backend)
+                self.assertEqual(cm.output[0], 'INFO:grimoire_elk.enriched.cocom:[cocom] Starting '
+                                 'enrich_repository_analysis study %s/cocom_enrich_graal_file'
+                                 % enrich_backend.elastic.anonymize_url(self.es_con))
+                self.assertEqual(cm.output[-1], 'INFO:grimoire_elk.enriched.cocom:[cocom] Ending '
+                                 'enrich_repository_analysis study ')
+
 
 if __name__ == "__main__":
     logging.basicConfig(level=logging.INFO, format='%(asctime)s %(message)s')

tests/test_colic.py

@@ -23,6 +23,7 @@
 import unittest
 
 from base import TestBaseBackend
+from grimoire_elk.enriched.enrich import logger
 
 
 HEADER_JSON = {"Content-Type": "application/json"}
@@ -81,6 +82,21 @@ def test_raw_to_enrich(self):
         self.assertEqual(eitem['modules'], ["graal"])
         self.assertEqual(eitem['file_path'], "graal/codecomplexity.py")
 
+    def test_colic_analysis_study(self):
+        """ Test that the colic analysis study works correctly """
+
+        study, ocean_backend, enrich_backend = self._test_study('enrich_colic_analysis')
+
+        with self.assertLogs(logger, level='INFO') as cm:
+
+            if study.__name__ == "enrich_colic_analysis":
+                study(ocean_backend, enrich_backend)
+                self.assertEqual(cm.output[0], 'INFO:grimoire_elk.enriched.cocom:[colic] Starting '
+                                 'enrich_colic_analysis study %s/colic_enrich_graal_file'
+                                 % enrich_backend.elastic.anonymize_url(self.es_con))
+                self.assertEqual(cm.output[-1], 'INFO:grimoire_elk.enriched.cocom:[colic] Ending '
+                                 'enrich_colic_analysis study')
+
 
 if __name__ == "__main__":
     logging.basicConfig(level=logging.INFO, format='%(asctime)s %(message)s')