diff --git a/tests/Type/Doctrine/DBAL/MysqliResultRowCountReturnTypeTest.php b/tests/Type/Doctrine/DBAL/MysqliResultRowCountReturnTypeTest.php index 8990bda1..cdfa8490 100644 --- a/tests/Type/Doctrine/DBAL/MysqliResultRowCountReturnTypeTest.php +++ b/tests/Type/Doctrine/DBAL/MysqliResultRowCountReturnTypeTest.php @@ -12,10 +12,13 @@ class MysqliResultRowCountReturnTypeTest extends TypeInferenceTestCase /** @return iterable */ public function dataFileAsserts(): iterable { - if (InstalledVersions::satisfies(new VersionParser(), 'doctrine/dbal', '>=4.0')) { + $versionParser = new VersionParser(); + if (InstalledVersions::satisfies($versionParser, 'doctrine/dbal', '>=4.0')) { yield from $this->gatherAssertTypes(__DIR__ . '/data/mysqli-result-row-count.php'); - } else { + } elseif (InstalledVersions::satisfies($versionParser, 'doctrine/dbal', '>=3.0')) { yield from $this->gatherAssertTypes(__DIR__ . '/data/mysqli-result-row-count-dbal-3.php'); + } else { + yield from $this->gatherAssertTypes(__DIR__ . '/data/mysqli-result-row-count-dbal-2.php'); } } diff --git a/tests/Type/Doctrine/DBAL/data/mysqli-result-row-count-dbal-2.php b/tests/Type/Doctrine/DBAL/data/mysqli-result-row-count-dbal-2.php new file mode 100644 index 00000000..235e6142 --- /dev/null +++ b/tests/Type/Doctrine/DBAL/data/mysqli-result-row-count-dbal-2.php @@ -0,0 +1,15 @@ +rowCount()); +}; + +function (DriverResult $r): void { + assertType('int|string', $r->rowCount()); +};