Fix build

phpstan · Sep 1, 2024 · 7414a7d · 7414a7d
1 parent 00439d5
commit 7414a7d
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/tests/Type/Doctrine/DBAL/PDOResultRowCountReturnTypeTest.php b/tests/Type/Doctrine/DBAL/PDOResultRowCountReturnTypeTest.php
@@ -2,6 +2,8 @@
 
 namespace PHPStan\Type\Doctrine\DBAL;
 
+use Composer\InstalledVersions;
+use Composer\Semver\VersionParser;
 use PHPStan\Testing\TypeInferenceTestCase;
 
 class PDOResultRowCountReturnTypeTest extends TypeInferenceTestCase
@@ -10,7 +12,12 @@ class PDOResultRowCountReturnTypeTest extends TypeInferenceTestCase
 	/** @return iterable<mixed> */
 	public function dataFileAsserts(): iterable
 	{
-		yield from $this->gatherAssertTypes(__DIR__ . '/data/pdo-result-row-count.php');
+		$versionParser = new VersionParser();
+		if (InstalledVersions::satisfies($versionParser, 'doctrine/dbal', '<3')) {
+			yield from $this->gatherAssertTypes(__DIR__ . '/data/pdo-result-row-count-dbal-2.php');
+		} else {
+			yield from $this->gatherAssertTypes(__DIR__ . '/data/pdo-result-row-count.php');
+		}
 	}
 
 	/**

diff --git a/tests/Type/Doctrine/DBAL/data/pdo-result-row-count-dbal-2.php b/tests/Type/Doctrine/DBAL/data/pdo-result-row-count-dbal-2.php
@@ -0,0 +1,15 @@
+<?php
+
+namespace PDOResultRowCountDbal2;
+
+use Doctrine\DBAL\Result;
+use Doctrine\DBAL\Driver\Result as DriverResult;
+use function PHPStan\Testing\assertType;
+
+function (Result $r): void {
+	assertType('int|string', $r->rowCount());
+};
+
+function (DriverResult $r): void {
+	assertType('int|string', $r->rowCount());
+};