feat: Add ability to use instrument lib with test runners.

dropecho · Jan 24, 2025 · 7839351 · 7839351
1 parent 4f058ec
commit 7839351
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 1 deletion.
diff --git a/.vim/coc-settings.json b/.vim/coc-settings.json
@@ -0,0 +1,3 @@
+{
+  "haxe.hxml": "build.hxml"
+}
diff --git a/src/dropecho/testing/BuddyIncludeMacros.hx b/src/dropecho/testing/BuddyIncludeMacros.hx
@@ -46,6 +46,10 @@ class BuddyIncludeMacros {
 							var suites = $a{suites};
 							var runner = new buddy.SuitesRunner(suites, null);
 							runner.run().then(_ -> {
+								#if instrument
+								instrument.coverage.Coverage.endCoverage();
+								#end
+
 								Sys.exit(runner.statusCode());
 							});
 						},

diff --git a/src/dropecho/testing/UTestIncludeMacros.hx b/src/dropecho/testing/UTestIncludeMacros.hx
@@ -48,7 +48,15 @@ class UTestIncludeMacros {
 								runner.addCase(s);
 							}
 
-							utest.ui.Report.create(runner);
+							#if instrument
+							runner.onComplete.add(_ -> {
+								instrument.coverage.Coverage.endCoverage();
+							});
+							#end
+
+							utest.ui.Report.create(runner, utest.ui.common.HeaderDisplayMode.SuccessResultsDisplayMode.NeverShowSuccessResults,
+								utest.ui.common.HeaderDisplayMode.NeverShowHeader);
+
 							runner.run();
 						}
 					}),