Fix covers function to accept traits

gehrisandro · Sep 5, 2024 · aaee0e4 · aaee0e4
1 parent 772448d
commit aaee0e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/PendingCalls/TestCall.php b/src/PendingCalls/TestCall.php
@@ -526,8 +526,8 @@ public function covers(array|string ...$classesOrFunctions): self
             $isTrait = trait_exists($classOrFunction);
             $isFunction = function_exists($classOrFunction);
 
-            if (! $isClass && ! $isFunction) {
-                throw new InvalidArgumentException(sprintf('No class or method named "%s" has been found.', $classOrFunction));
+            if (! $isClass && ! $isTrait && ! $isFunction) {
+                throw new InvalidArgumentException(sprintf('No class, trait or method named "%s" has been found.', $classOrFunction));
             }
 
             if ($isClass) {