Add support for assertArrayHasKey (#100)

phpstan · Jul 14, 2021 · 6aaff11 · 6aaff11
1 parent efc0099
commit 6aaff11
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/Type/PHPUnit/Assert/AssertTypeSpecifyingExtensionHelper.php b/src/Type/PHPUnit/Assert/AssertTypeSpecifyingExtensionHelper.php
@@ -252,6 +252,9 @@ private static function getExpressionResolvers(): array
 						]
 					);
 				},
+				'ArrayHasKey' => function (Scope $scope, Arg $key, Arg $array): FuncCall {
+					return new \PhpParser\Node\Expr\FuncCall(new Name('array_key_exists'), [$key, $array]);
+				},
 			];
 		}
 

diff --git a/tests/Type/PHPUnit/data/assert-function.php b/tests/Type/PHPUnit/data/assert-function.php
@@ -3,6 +3,8 @@
 namespace AssertFunction;
 
 use function PHPStan\Testing\assertType;
+use function PHPUnit\Framework\assertArrayHasKey;
+use function PHPUnit\Framework\assertArrayNotHasKey;
 use function PHPUnit\Framework\assertInstanceOf;
 
 class Foo
@@ -17,4 +19,15 @@ public function doFoo($o): void
 		assertType(self::class, $o);
 	}
 
+	public function arrayHasNumericKey(array $a): void {
+		assertArrayHasKey(0, $a);
+		assertType('array&hasOffset(0)', $a);
+	}
+
+	public function arrayHasStringKey(array $a): void
+	{
+		assertArrayHasKey('key', $a);
+		assertType("array&hasOffset('key')", $a);
+	}
+
 }