Narrow type on assertCount and assertNotCount

Author: michalbundyra

src/Type/PHPUnit/Assert/AssertTypeSpecifyingExtensionHelper.php

@@ -136,6 +136,20 @@ private static function getExpressionResolvers(): array
 	{
 if (self::$resolvers === null) {
 self::$resolvers = [
+'Count' => staticfunction (Scope$scope, Arg$expected, Arg$actual): Identical {
+returnnewIdentical(
+$expected->value,
+newFuncCall(newName('count'), [$actual])
+ );
+ },
+'NotCount' => staticfunction (Scope$scope, Arg$expected, Arg$actual): BooleanNot {
+returnnewBooleanNot(
+newIdentical(
+$expected->value,
+newFuncCall(newName('count'), [$actual])
+ )
+ );
+ },
 'InstanceOf' => staticfunction (Scope$scope, Arg$class, Arg$object): Instanceof_ {
 returnnewInstanceof_(
 $object->value,

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

@@ -7,6 +7,8 @@
 usefunctionPHPUnit\Framework\assertContains;
 usefunctionPHPUnit\Framework\assertContainsEquals;
 usefunctionPHPUnit\Framework\assertContainsOnlyInstancesOf;
+usefunctionPHPUnit\Framework\assertCount;
+usefunctionPHPUnit\Framework\assertNotCount;
 usefunctionPHPUnit\Framework\assertEmpty;
 usefunctionPHPUnit\Framework\assertInstanceOf;
 usefunctionPHPUnit\Framework\assertObjectHasAttribute;
@@ -90,4 +92,23 @@ public function containsOnlyInstancesOf(array $a, \Traversable $b): void
 assertType('Traversable', $b);
 	}
 
+publicfunctioncount(array$a, \Countable$b): void
+	{
+assertCount(3, $a);
+assertType('non-empty-array', $a);
+
+assertCount(7, $b);
+assertType('Countable', $b);
+	}
+
+publicfunctionnotCount(array$a, array$b): void
+	{
+assertNotCount(0, $a);
+assertType('non-empty-array', $a);
+
+// still might be empty
+assertNotCount(1, $b);
+assertType('array', $b);
+	}
+
 }