Support eq and notEq

Author: herndlm

README.md

@@ -57,6 +57,8 @@ This extension specifies types of values passed to:
 * `Assert::notFalse`
 * `Assert::null`
 * `Assert::notNull`
+* `Assert::eq`
+* `Assert::notEq`
 * `Assert::same`
 * `Assert::notSame`
 * `Assert::greaterThan`

src/Type/WebMozartAssert/AssertTypeSpecifyingExtension.php

@@ -483,6 +483,15 @@ private static function getExpressionResolvers(): array
 						new ConstFetch(new Name('null'))
 					);
 				},
+				'eq' => static function (Scope $scope, Arg $value, Arg $value2): Expr {
+					return new Equal(
+						$value->value,
+						$value2->value
+					);
+				},
+				'notEq' => static function (Scope $scope, Arg $value, Arg $value2): Expr {
+					return new BooleanNot(self::$resolvers['eq']($scope, $value, $value2));
+				},
 				'same' => static function (Scope $scope, Arg $value1, Arg $value2): Expr {
 					return new Identical(
 						$value1->value,

tests/Type/WebMozartAssert/data/comparison.php

@@ -43,6 +43,39 @@ public function notNull(?int $a): void
 		\PHPStan\Testing\assertType('int', $a);
 	}
 
+	/**
+	 * @param non-empty-string  $b2
+	 */
+	public function eq(?bool $a, string $b1, string $b2, $c, ?bool $d): void
+	{
+		Assert::eq($a, null);
+		\PHPStan\Testing\assertType('false|null', $a);
+
+		Assert::eq($b1, $b2);
+		\PHPStan\Testing\assertType('non-empty-string', $b1);
+
+		Assert::eq($c, false);
+		\PHPStan\Testing\assertType('0|0.0|\'\'|\'0\'|array{}|false|null', $c);
+
+		Assert::nullOrEq($d, true);
+		\PHPStan\Testing\assertType('true|null', $d);
+	}
+
+	public function notEq(?bool $a, string $b, $c, ?bool $d): void
+	{
+		Assert::notEq($a, null);
+		\PHPStan\Testing\assertType('true', $a);
+
+		Assert::notEq($b, '');
+		\PHPStan\Testing\assertType('non-empty-string', $b);
+
+		Assert::notEq($c, true);
+		\PHPStan\Testing\assertType('0|0.0|\'\'|\'0\'|array{}|false|null', $c);
+
+		Assert::nullOrNotEq($d, true);
+		\PHPStan\Testing\assertType('false|null', $d);
+	}
+
 	public function same($a, $b): void
 	{
 		Assert::same($a, 1);