Fix nullOr* non-empty-string assertions

src/Type/WebMozartAssert/AssertTypeSpecifyingExtension.php

@@ -67,10 +67,6 @@ public function isStaticMethodSupported(
 		TypeSpecifierContext $context
 	): bool
 	{
-		if (in_array($staticMethodReflection->getName(), self::ASSERTIONS_RESULTING_IN_NON_EMPTY_STRING, true)) {
-			return true;
-		}
-
 		if (substr($staticMethodReflection->getName(), 0, 6) === 'allNot') {
 			$methods = [
 				'allNotInstanceOf' => 2,
@@ -84,6 +80,10 @@ public function isStaticMethodSupported(
 		$trimmedName = self::trimName($staticMethodReflection->getName());
 		$resolvers = self::getExpressionResolvers();
 
+		if (in_array($trimmedName, self::ASSERTIONS_RESULTING_IN_NON_EMPTY_STRING, true)) {
+			return true;
+		}
+
 		if (!array_key_exists($trimmedName, $resolvers)) {
 			return false;
 		}
@@ -167,12 +167,13 @@ private static function createExpression(
 		$trimmedName = self::trimName($name);
 
 		if (in_array($trimmedName, self::ASSERTIONS_RESULTING_IN_NON_EMPTY_STRING, true)) {
-			return self::createIsNonEmptyStringExpression($args);
+			$expression = self::createIsNonEmptyStringExpression($args);
+		} else {
+			$resolvers = self::getExpressionResolvers();
+			$resolver = $resolvers[$trimmedName];
+			$expression = $resolver($scope, ...$args);
 		}
 
-		$resolvers = self::getExpressionResolvers();
-		$resolver = $resolvers[$trimmedName];
-		$expression = $resolver($scope, ...$args);
 		if ($expression === null) {
 			return null;
 		}

tests/Type/WebMozartAssert/data/array.php

@@ -25,13 +25,16 @@ public function keyExists(array $a): void
 		\PHPStan\Testing\assertType('array{foo: string, bar?: string}', $a);
 	}
 
-	public function validArrayKey($a, bool $b): void
+	public function validArrayKey($a, bool $b, $c): void
 	{
 		Assert::validArrayKey($a);
 		\PHPStan\Testing\assertType('int|string', $a);
 
 		Assert::validArrayKey($b);
 		\PHPStan\Testing\assertType('*NEVER*', $b);
+
+		Assert::nullOrValidArrayKey($c);
+		\PHPStan\Testing\assertType('int|string|null', $c);
 	}
 
 	/**
@@ -94,10 +97,13 @@ public function countBetween(array $a, array $b, array $c, array $d): void
 		\PHPStan\Testing\assertType('*NEVER*', $d);
 	}
 
-	public function isList($a): void
+	public function isList($a, $b): void
 	{
 		Assert::isList($a);
 		\PHPStan\Testing\assertType('array', $a);
+
+		Assert::nullOrIsList($b);
+		\PHPStan\Testing\assertType('array|null', $b);
 	}
 
 }

tests/Type/WebMozartAssert/data/comparison.php

@@ -6,19 +6,25 @@
 
 class ComparisonTest
 {
-	public function true($a): void
+	public function true($a, $b): void
 	{
 		Assert::true($a);
 		\PHPStan\Testing\assertType('true', $a);
+
+		Assert::nullOrTrue($b);
+		\PHPStan\Testing\assertType('true|null', $b);
 	}
 
-	public function false($a): void
+	public function false($a, $b): void
 	{
 		Assert::false($a);
 		\PHPStan\Testing\assertType('false', $a);
+
+		Assert::nullOrFalse($b);
+		\PHPStan\Testing\assertType('false|null', $b);
 	}
 
-	public function notFalse(int $a): void
+	public function notFalse($a, $b): void
 	{
 		/** @var int|false $a */
 		Assert::notFalse($a);
@@ -37,10 +43,13 @@ public function notNull(?int $a): void
 		\PHPStan\Testing\assertType('int', $a);
 	}
 
-	public function same($a): void
+	public function same($a, $b): void
 	{
 		Assert::same($a, 1);
 		\PHPStan\Testing\assertType('1', $a);
+
+		Assert::nullOrSame($b, 1);
+		\PHPStan\Testing\assertType('1|null', $b);
 	}
 
 	/**
@@ -61,9 +70,12 @@ public function inArray($a, $b): void
 		\PHPStan\Testing\assertType('\'bar\'|\'foo\'|null', $b);
 	}
 
-	public function oneOf($a): void
+	public function oneOf($a, $b): void
 	{
 		Assert::oneOf($a, [1, 2]);
 		\PHPStan\Testing\assertType('1|2', $a);
+
+		Assert::nullOrOneOf($b, [1, 2]);
+		\PHPStan\Testing\assertType('1|2|null', $b);
 	}
 }

tests/Type/WebMozartAssert/data/object.php

@@ -7,28 +7,40 @@
 class ObjectTest
 {
 
-	public function classExists($a): void
+	public function classExists($a, $b): void
 	{
 		Assert::classExists($a);
 		\PHPStan\Testing\assertType('class-string', $a);
+
+		Assert::nullOrClassExists($b);
+		\PHPStan\Testing\assertType('class-string|null', $b);
 	}
 
-	public function subclassOf($a): void
+	public function subclassOf($a, $b): void
 	{
 		Assert::subclassOf($a, self::class);
 		\PHPStan\Testing\assertType('class-string<PHPStan\Type\WebMozartAssert\ObjectTest>|PHPStan\Type\WebMozartAssert\ObjectTest', $a);
+
+		Assert::nullOrSubclassOf($b, self::class);
+		\PHPStan\Testing\assertType('class-string<PHPStan\Type\WebMozartAssert\ObjectTest>|PHPStan\Type\WebMozartAssert\ObjectTest|null', $b);
 	}
 
-	public function interfaceExists($a): void
+	public function interfaceExists($a, $b): void
 	{
 		Assert::interfaceExists($a);
 		\PHPStan\Testing\assertType('class-string', $a);
+
+		Assert::nullOrInterfaceExists($b);
+		\PHPStan\Testing\assertType('class-string|null', $b);
 	}
 
-	public function implementsInterface($a): void
+	public function implementsInterface($a, $b): void
 	{
 		Assert::implementsInterface($a, ObjectFoo::class);
 		\PHPStan\Testing\assertType('PHPStan\Type\WebMozartAssert\ObjectFoo', $a);
+
+		Assert::nullOrImplementsInterface($b, ObjectFoo::class);
+		\PHPStan\Testing\assertType('PHPStan\Type\WebMozartAssert\ObjectFoo|null', $b);
 	}
 
 	public function propertyExists(object $a): void

tests/Type/WebMozartAssert/data/string.php

@@ -91,16 +91,22 @@ public function lengthBetween(string $a, string $b, string $c, string $d): void
 		\PHPStan\Testing\assertType('non-empty-string', $d);
 	}
 
-	public function unicodeLetters($a): void
+	public function unicodeLetters($a, $b): void
 	{
 		Assert::unicodeLetters($a);
 		\PHPStan\Testing\assertType('non-empty-string', $a);
+
+		Assert::nullOrUnicodeLetters($b);
+		\PHPStan\Testing\assertType('non-empty-string|null', $b);
 	}
 
-	public function alpha($a): void
+	public function alpha($a, $b): void
 	{
 		Assert::alpha($a);
 		\PHPStan\Testing\assertType('non-empty-string', $a);
+
+		Assert::nullOrAlpha($b);
+		\PHPStan\Testing\assertType('non-empty-string|null', $b);
 	}
 
 	public function digits(string $a): void
@@ -133,28 +139,40 @@ public function uuid(string $a): void
 		\PHPStan\Testing\assertType('non-empty-string', $a);
 	}
 
-	public function ip($a): void
+	public function ip($a, $b): void
 	{
 		Assert::ip($a);
 		\PHPStan\Testing\assertType('non-empty-string', $a);
+
+		Assert::nullOrIp($b);
+		\PHPStan\Testing\assertType('non-empty-string|null', $b);
 	}
 
-	public function ipv4($a): void
+	public function ipv4($a, $b): void
 	{
 		Assert::ipv4($a);
 		\PHPStan\Testing\assertType('non-empty-string', $a);
+
+		Assert::nullOrIpv4($b);
+		\PHPStan\Testing\assertType('non-empty-string|null', $b);
 	}
 
-	public function ipv6($a): void
+	public function ipv6($a, $b): void
 	{
 		Assert::ipv6($a);
 		\PHPStan\Testing\assertType('non-empty-string', $a);
+
+		Assert::nullOrIpv6($b);
+		\PHPStan\Testing\assertType('non-empty-string|null', $b);
 	}
 
-	public function email($a): void
+	public function email($a, $b): void
 	{
 		Assert::email($a);
 		\PHPStan\Testing\assertType('non-empty-string', $a);
+
+		Assert::nullOrEmail($b);
+		\PHPStan\Testing\assertType('non-empty-string|null', $b);
 	}
 
 	public function notWhitespaceOnly(string $a): void