[PhpUnitBridge] Added a CoverageListener to enhance the code coverage report

lyrixx · lyrixx · commit 449e406e178b · 2017-06-12T16:39:22.000+02:00
diff --git a/src/Symfony/Bridge/PhpUnit/CoverageListener.php b/src/Symfony/Bridge/PhpUnit/CoverageListener.php
@@ -0,0 +1,47 @@
+<?php
+
+namespace App;
+
+use PHPUnit\Framework\AssertionFailedError;
+use PHPUnit\Framework\BaseTestListener;
+use PHPUnit\Framework\Test;
+use PHPUnit\Framework\TestSuite;
+use PHPUnit\Framework\Warning;
+
+/**
+ * CoverageListener adds `@covers <className>` on each test suite when possible
+ * to make the code coverage more accurate.
+ *
+ * @author Grégoire Pineau <lyrixx@lyrixx.info>
+ */
+class CoverageListener extends BaseTestListener
+{
+    public function startTest(Test $test)
+    {
+        $annotations = $test->getAnnotations();
+
+        if (isset($annotations['class']['covers']) || isset($annotations['method']['covers'])) {
+            return;
+        }
+
+        $class = get_class($test);
+
+        $coverClass = str_replace('Tests\\', '', $class);
+        $coverClass = preg_replace('{Test$}', '', $coverClass);
+
+        if (!class_exists($coverClass)) {
+            return;
+        }
+
+        $r = new \ReflectionProperty(\PHPUnit\Util\Test::class, 'annotationCache');
+        $r->setAccessible(true);
+
+        $cache = $r->getValue();
+        $cache = array_replace_recursive($cache, array(
+            $class => array(
+                'covers' => array($coverClass),
+            )
+        ));
+        $r->setValue(\PHPUnit\Util\Test::class, $cache);
+    }
+}
