@@ -95,7 +95,7 @@ public function testPrimeReferencesWithDBRefObjects(): void
9595 ->field ('groups ' )->prime (true );
9696
9797 foreach ($ qb ->getQuery () as $ user ) {
98- self ::assertInstanceOf (InternalProxy::class, $ user ->getAccount ());
98+ self ::assertIsLazyObject ( $ user ->getAccount ());
9999 self ::assertFalse ($ this ->uow ->isUninitializedObject ($ user ->getAccount ()));
100100
101101 self ::assertCount (2 , $ user ->getGroups ());
@@ -133,7 +133,7 @@ public function testPrimeReferencesWithSimpleReferences(): void
133133 ->field ('users ' )->prime (true );
134134
135135 foreach ($ qb ->getQuery () as $ simpleUser ) {
136- self ::assertInstanceOf (InternalProxy::class, $ simpleUser ->getUser ());
136+ self ::assertIsLazyObject ( $ simpleUser ->getUser ());
137137 self ::assertFalse ($ this ->uow ->isUninitializedObject ($ simpleUser ->getUser ()));
138138
139139 self ::assertCount (2 , $ simpleUser ->getUsers ());
@@ -196,7 +196,7 @@ public function testPrimeReferencesNestedInNamedEmbeddedReference(): void
196196 self ::assertNotInstanceOf (InternalProxy::class, $ embeddedDoc );
197197 self ::assertInstanceOf (EmbeddedWhichReferences::class, $ embeddedDoc );
198198
199- self ::assertInstanceOf (InternalProxy::class, $ embeddedDoc ->referencedDoc );
199+ self ::assertIsLazyObject ( $ embeddedDoc ->referencedDoc );
200200 self ::assertFalse ($ this ->uow ->isUninitializedObject ($ embeddedDoc ->referencedDoc ));
201201
202202 self ::assertCount (2 , $ embeddedDoc ->referencedDocs );
@@ -252,7 +252,7 @@ public function testPrimeReferencesWithDifferentStoreAsReferences(): void
252252 assert ($ referenceUser instanceof ReferenceUser);
253253 $ user = $ referenceUser ->getUser ();
254254 self ::assertInstanceOf (User::class, $ user );
255- self ::assertInstanceOf (InternalProxy::class, $ user );
255+ self ::assertIsLazyObject ( $ user );
256256 self ::assertFalse ($ this ->uow ->isUninitializedObject ($ user ));
257257
258258 self ::assertCount (1 , $ referenceUser ->getUsers ());
@@ -263,7 +263,7 @@ public function testPrimeReferencesWithDifferentStoreAsReferences(): void
263263 }
264264
265265 $ parentUser = $ referenceUser ->getParentUser ();
266- self ::assertInstanceOf (InternalProxy::class, $ parentUser );
266+ self ::assertIsLazyObject ( $ parentUser );
267267 self ::assertInstanceOf (User::class, $ parentUser );
268268 self ::assertFalse ($ this ->uow ->isUninitializedObject ($ parentUser ));
269269
@@ -276,7 +276,7 @@ public function testPrimeReferencesWithDifferentStoreAsReferences(): void
276276
277277 $ otherUser = $ referenceUser ->getOtherUser ();
278278 self ::assertInstanceOf (User::class, $ otherUser );
279- self ::assertInstanceOf (InternalProxy::class, $ otherUser );
279+ self ::assertIsLazyObject ( $ otherUser );
280280 self ::assertFalse ($ this ->uow ->isUninitializedObject ($ otherUser ));
281281
282282 self ::assertCount (1 , $ referenceUser ->getOtherUsers ());
@@ -331,7 +331,7 @@ public function testPrimeReferencesWithDiscriminatedReferenceOne(): void
331331 ->field ('server ' )->prime (true );
332332
333333 foreach ($ qb ->getQuery () as $ agent ) {
334- self ::assertInstanceOf (InternalProxy::class, $ agent ->server );
334+ self ::assertIsLazyObject ( $ agent ->server );
335335 self ::assertFalse ($ this ->uow ->isUninitializedObject ($ agent ->server ));
336336 }
337337 }
@@ -523,7 +523,7 @@ public function testPrimeEmbeddedReferenceTwoLevelsDeep(): void
523523
524524 $ currency = $ money ->getCurrency ();
525525
526- self ::assertInstanceOf (InternalProxy::class, $ currency );
526+ self ::assertIsLazyObject ( $ currency );
527527 self ::assertInstanceOf (Currency::class, $ currency );
528528 self ::assertFalse ($ this ->uow ->isUninitializedObject ($ currency ));
529529 }
@@ -551,7 +551,7 @@ public function testPrimeReferencesInReferenceMany(): void
551551 self ::assertInstanceOf (BlogPost::class, $ post );
552552
553553 $ comment = $ post ->comments ->first ();
554- self ::assertInstanceOf (InternalProxy::class, $ comment ->author );
554+ self ::assertIsLazyObject ( $ comment ->author );
555555 self ::assertFalse ($ this ->uow ->isUninitializedObject ($ comment ->author ));
556556 }
557557
@@ -578,7 +578,7 @@ public function testPrimeReferencesInReferenceManyWithRepositoryMethodEager(): v
578578 self ::assertInstanceOf (BlogPost::class, $ post );
579579
580580 $ comment = $ post ->repoCommentsWithPrimer ->first ();
581- self ::assertInstanceOf (InternalProxy::class, $ comment ->author );
581+ self ::assertIsLazyObject ( $ comment ->author );
582582 self ::assertFalse ($ this ->uow ->isUninitializedObject ($ comment ->author ));
583583 }
584584}
0 commit comments