fix: testLoadFixturesCheckReferences

liip · Nov 2, 2023 · 587c934 · 587c934
1 parent 099d3ce
commit 587c934
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/tests/Test/ConfigMysqlCacheDbTest.php b/tests/Test/ConfigMysqlCacheDbTest.php
@@ -123,14 +123,24 @@ public function testLoadFixturesCheckReferences(): void
             'Liip\Acme\Tests\App\DataFixtures\ORM\LoadUserData',
         ])->getReferenceRepository();
 
-        $this->assertCount(1, $referenceRepository->getReferences());
+        $references = (method_exists($referenceRepository, 'getReferencesByClass'))
+            // with getReferencesByClass(), references are grouped by class
+            ? $referenceRepository->getReferencesByClass()['Liip\Acme\Tests\App\Entity\User']
+            : $referenceRepository->getReferences();
+
+        $this->assertCount(1, $references);
 
         $referenceRepository = $this->databaseTool->loadFixtures([
             'Liip\Acme\Tests\App\DataFixtures\ORM\LoadUserData',
             'Liip\Acme\Tests\App\DataFixtures\ORM\LoadSecondUserData',
         ])->getReferenceRepository();
 
-        $this->assertCount(2, $referenceRepository->getReferences());
+        $references = (method_exists($referenceRepository, 'getReferencesByClass'))
+            // with getReferencesByClass(), references are grouped by class
+            ? $referenceRepository->getReferencesByClass()['Liip\Acme\Tests\App\Entity\User']
+            : $referenceRepository->getReferences();
+
+        $this->assertCount(2, $references);
     }
 
     protected static function getKernelClass(): string