Add support for assertContains and assertContainsEquals

Author: villfa

src/Type/PHPUnit/Assert/AssertTypeSpecifyingExtensionHelper.php

@@ -276,6 +276,21 @@ private static function getExpressionResolvers(): array
 'ObjectHasAttribute' => staticfunction (Scope$scope, Arg$property, Arg$object): FuncCall {
 returnnewFuncCall(newName('property_exists'), [$object, $property]);
  },
+'Contains' => staticfunction (Scope$scope, Arg$needle, Arg$haystack): Expr {
+returnnewExpr\BinaryOp\BooleanOr(
+newExpr\Instanceof_($haystack->value, newName('Traversable')),
+newFuncCall(newName('in_array'), [$needle, $haystack, newArg(newConstFetch(newName('true')))])
+ );
+ },
+'ContainsEquals' => staticfunction (Scope$scope, Arg$needle, Arg$haystack): Expr {
+returnnewExpr\BinaryOp\BooleanOr(
+newExpr\Instanceof_($haystack->value, newName('Traversable')),
+newExpr\BinaryOp\BooleanAnd(
+newExpr\BooleanNot(newExpr\Empty_($haystack->value)),
+newFuncCall(newName('in_array'), [$needle, $haystack, newArg(newConstFetch(newName('false')))])
+ )
+ );
+ },
 'ContainsOnlyInstancesOf' => staticfunction (Scope$scope, Arg$className, Arg$haystack): Expr {
 returnnewExpr\BinaryOp\BooleanOr(
 newExpr\Instanceof_($haystack->value, newName('Traversable')),

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

@@ -4,6 +4,8 @@
 
 usefunctionPHPStan\Testing\assertType;
 usefunctionPHPUnit\Framework\assertArrayHasKey;
+usefunctionPHPUnit\Framework\assertContains;
+usefunctionPHPUnit\Framework\assertContainsEquals;
 usefunctionPHPUnit\Framework\assertContainsOnlyInstancesOf;
 usefunctionPHPUnit\Framework\assertEmpty;
 usefunctionPHPUnit\Framework\assertInstanceOf;
@@ -61,6 +63,24 @@ public function testEmpty($a): void
 assertType("0|0.0|''|'0'|array{}|Countable|EmptyIterator|false|null", $a);
 	}
 
+publicfunctioncontains(array$a, \Traversable$b): void
+	{
+assertContains('foo', $a);
+assertType('non-empty-array', $a);
+
+assertContains('foo', $b);
+assertType('Traversable', $b);
+	}
+
+publicfunctioncontainsEquals(array$a, \Traversable$b): void
+	{
+assertContainsEquals('foo', $a);
+assertType('non-empty-array', $a);
+
+assertContainsEquals('foo', $b);
+assertType('Traversable', $b);
+	}
+
 publicfunctioncontainsOnlyInstancesOf(array$a, \Traversable$b): void
 	{
 assertContainsOnlyInstancesOf(\stdClass::class, $a);