Merge pull request #8 from ergebnis/feature/slow-test-collector

Enhancement: Implement SlowTestCollector

CHANGELOG.md

@@ -11,9 +11,11 @@ For a full diff see [`7afa59c...main`][7afa59c...main].
 ### Added
 
 * Added `SlowTest` ([#6]), by [@localheinz]
+* Added `SlowTestCollector` ([#8]), by [@localheinz]
 
 [7afa59c...main]: https://github.com/ergebnis/phpunit-slow-test-collector/compare/7afa59c...main
 
 [#6]: https://github.com/ergebnis/phpunit-slow-test-collector/pull/6
+[#8]: https://github.com/ergebnis/phpunit-slow-test-collector/pull/8
 
 [@localheinz]: https://github.com/localheinz

src/SlowTestCollector.php

@@ -0,0 +1,78 @@
+<?php
+
+declare(strict_types=1);
+
+/**
+ * Copyright (c) 2021 Andreas Möller
+ *
+ * For the full copyright and license information, please view
+ * the LICENSE.md file that was distributed with this source code.
+ *
+ * @see https://github.com/ergebnis/phpunit-slow-test-collector
+ */
+
+namespace Ergebnis\PHPUnit\SlowTestCollector;
+
+use PHPUnit\Event;
+
+final class SlowTestCollector
+{
+    private Event\Telemetry\Duration $maximumDuration;
+
+    /**
+     * @var array<string, SlowTest>
+     */
+    private array $slowTests = [];
+
+    public function __construct(Event\Telemetry\Duration $maximumDuration)
+    {
+        $this->maximumDuration = $maximumDuration;
+    }
+
+    public function collect(SlowTest $slowTest): void
+    {
+        $duration = $slowTest->duration();
+
+        if (!$duration->isGreaterThan($this->maximumDuration)) {
+            return;
+        }
+
+        $test = $slowTest->test();
+
+        $key = \sprintf(
+            '%s::%s',
+            $test->className(),
+            $test->methodNameWithDataSet(),
+        );
+
+        if (\array_key_exists($key, $this->slowTests)) {
+            $previousSlowTest = $this->slowTests[$key];
+
+            if (!$duration->isGreaterThan($previousSlowTest->duration())) {
+                return;
+            }
+
+            $this->slowTests[$key] = $slowTest;
+
+            return;
+        }
+
+        $this->slowTests[$key] = $slowTest;
+    }
+
+    public function maximumDuration(): Event\Telemetry\Duration
+    {
+        return $this->maximumDuration;
+    }
+
+    /**
+     * @phpstan-return list<SlowTest>
+     * @psalm-return list<SlowTest>
+     *
+     * @return array<int, SlowTest>
+     */
+    public function slowTests(): array
+    {
+        return \array_values($this->slowTests);
+    }
+}