Add partial support for assertContainsOnlyInstancesOf

Author: villfa

src/Type/PHPUnit/Assert/AssertTypeSpecifyingExtensionHelper.php

@@ -13,7 +13,9 @@
 usePhpParser\Node\Expr\FuncCall;
 usePhpParser\Node\Expr\Instanceof_;
 usePhpParser\Node\Name;
+usePhpParser\Node\Param;
 usePhpParser\Node\Scalar\LNumber;
+usePhpParser\Node\Stmt;
 usePHPStan\Analyser\Scope;
 usePHPStan\Analyser\SpecifiedTypes;
 usePHPStan\Analyser\TypeSpecifier;
@@ -274,6 +276,28 @@ private static function getExpressionResolvers(): array
 'ObjectHasAttribute' => staticfunction (Scope$scope, Arg$property, Arg$object): FuncCall {
 returnnewFuncCall(newName('property_exists'), [$object, $property]);
  },
+'ContainsOnlyInstancesOf' => staticfunction (Scope$scope, Arg$className, Arg$haystack): Expr {
+returnnewExpr\BinaryOp\BooleanOr(
+newExpr\Instanceof_($haystack->value, newName('Traversable')),
+newIdentical(
+$haystack->value,
+newFuncCall(newName('array_filter'), [
+$haystack,
+newArg(newExpr\Closure([
+'static' => true,
+'params' => [
+newParam(newExpr\Variable('_')),
+ ],
+'stmts' => [
+newStmt\Return_(
+newFuncCall(newName('is_a'), [newArg(newExpr\Variable('_')), $className])
+ ),
+ ],
+ ])),
+ ])
+ )
+ );
+ },
  ];
  }
 

tests/Type/PHPUnit/data/assert-function.php

@@ -4,6 +4,7 @@
 
 usefunctionPHPStan\Testing\assertType;
 usefunctionPHPUnit\Framework\assertArrayHasKey;
+usefunctionPHPUnit\Framework\assertContainsOnlyInstancesOf;
 usefunctionPHPUnit\Framework\assertEmpty;
 usefunctionPHPUnit\Framework\assertInstanceOf;
 usefunctionPHPUnit\Framework\assertObjectHasAttribute;
@@ -60,4 +61,13 @@ public function testEmpty($a): void
 assertType("0|0.0|''|'0'|array{}|Countable|EmptyIterator|false|null", $a);
 	}
 
+publicfunctioncontainsOnlyInstancesOf(array$a, \Traversable$b): void
+	{
+assertContainsOnlyInstancesOf(\stdClass::class, $a);
+assertType('array<stdClass>', $a);
+
+assertContainsOnlyInstancesOf(\stdClass::class, $b);
+assertType('Traversable', $b);
+	}
+
 }