fix: testLoadFixturesCheckReferences

liip · Nov 12, 2023 · 00a5a97 · 00a5a97
1 parent c13efc1
commit 00a5a97
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/tests/Test/ConfigMysqlCacheDbTest.php b/tests/Test/ConfigMysqlCacheDbTest.php
@@ -124,14 +124,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