Add test coverage to test run script. (#1406)

* add coverage to run_tests * add coverage to test run script * updated libbde-python * update yara test message (rules updated)
google · Dec 15, 2023 · bfc586e · bfc586e
1 parent 90d1cf8
commit bfc586e
Show file tree

Hide file tree

Showing 3 changed files with 94 additions and 61 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/run_tests.py b/run_tests.py
@@ -15,13 +15,28 @@
 # limitations under the License.
 """Script to run the tests."""
 
+import coverage
 import unittest
 import sys
 
 if __name__ == '__main__':
   test_suite = unittest.TestLoader().discover('turbinia', pattern='*_test.py')
-  test_suite.addTests(
-      unittest.TestLoader().discover('turbinia/api/cli', pattern='*_test.py'))
-  test_results = unittest.TextTestRunner(verbosity=2).run(test_suite)
+  test_results = None
+
+  cov = coverage.Coverage(include='turbinia/**/*.py', omit='*_test.py')
+  cov.start()
+
+  try:
+    test_suite.addTests(
+        unittest.TestLoader().discover('turbinia/api/cli', pattern='*_test.py'))
+    test_results = unittest.TextTestRunner(verbosity=2).run(test_suite)
+  except Exception:
+    pass
+
+  cov.stop()
+  cov.save()
+
+  cov.report(file=sys.stdout)
+
   if not test_results.wasSuccessful():
     sys.exit(1)
diff --git a/turbinia/workers/analysis/yara_test.py b/turbinia/workers/analysis/yara_test.py
@@ -29,7 +29,7 @@
 class YaraAnalysisTaskTest(TestTurbiniaTaskBase):
   """Tests for YaraAnalysisTask Task."""
 
-  YARA_SUMMARY = 'Yara analysis found 3 alert(s)'
+  YARA_SUMMARY = 'Yara analysis found 4 alert(s)'
 
   TEST_DATA_DIR = None