Merge pull request #194 from php-school/more-test-helpers

Add more test helpers

Author: AydinHassan

src/TestUtils/WorkshopExerciseTest.php

@@ -15,9 +15,11 @@
 use PhpSchool\PhpWorkshop\Result\Cli\CliResult;
 use PhpSchool\PhpWorkshop\Result\Failure;
 use PhpSchool\PhpWorkshop\Result\FailureInterface;
+use PhpSchool\PhpWorkshop\Result\ResultGroupInterface;
 use PhpSchool\PhpWorkshop\Result\ResultInterface;
 use PhpSchool\PhpWorkshop\ResultAggregator;
 use PhpSchool\PhpWorkshop\Utils\Collection;
+use PHPUnit\Framework\ExpectationFailedException;
 use PHPUnit\Framework\TestCase;
 use Psr\Container\ContainerInterface;
 use PhpSchool\PhpWorkshop\Input\Input;
@@ -126,11 +128,21 @@ public function assertResultsHasFailure(string $resultClass, string $reason): vo
     }
 
     public function assertOutputWasIncorrect(): void
+    {
+        $this->assertFalse($this->getOutputResult()->isSuccessful());
+    }
+
+    public function assertOutputWasCorrect(): void
+    {
+        $this->assertTrue($this->getOutputResult()->isSuccessful());
+    }
+
+    private function getOutputResult(): ResultGroupInterface
     {
         $exerciseType = $this->getExercise()->getType();
 
         if ($exerciseType->equals(ExerciseType::CLI())) {
-            $results = (new Collection($this->results->getIterator()->getArrayCopy()))
+            $results = collect($this->results->getIterator()->getArrayCopy())
                 ->filter(function (ResultInterface $result) {
                     return $result instanceof CliResult;
                 });
@@ -139,16 +151,33 @@ public function assertOutputWasIncorrect(): void
         }
 
         if ($exerciseType->equals(ExerciseType::CGI())) {
-            $results = (new Collection($this->results->getIterator()->getArrayCopy()))
+            $results = collect($this->results->getIterator()->getArrayCopy())
                 ->filter(function (ResultInterface $result) {
                     return $result instanceof CgiResult;
                 });
 
             $this->assertCount(1, $results);
         }
 
-        $outputResults = $results->values()->get(0);
+        return $results->values()->get(0);
+    }
+
+    public function assertResultsHasFailureAndMatches(string $resultClass, callable $matcher): void
+    {
+        $failures = collect($this->results->getIterator()->getArrayCopy())
+            ->filter(function (ResultInterface $result) {
+                return $result instanceof FailureInterface;
+            })
+            ->filter(function (FailureInterface $failure) use ($resultClass) {
+                return $failure instanceof $resultClass;
+            });
+
+        if ($failures->count() === 0) {
+            throw new ExpectationFailedException("No failures found for class: '$resultClass'");
+        }
 
-        $this->assertFalse($outputResults->isSuccessful());
+        $failures->each(function ($failure) use ($matcher) {
+            $this->assertTrue($matcher($failure));
+        });
     }
 }