File tree 1 file changed +2
-4
lines changed
1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -428,6 +428,7 @@ class EntityMapUtil:
428
428
"""
429
429
430
430
def __init__ (self , test_class ):
431
+ self ._cluster_time = None
431
432
self ._entities : Dict [str , Any ] = {}
432
433
self ._listeners : Dict [str , EventListenerUtil ] = {}
433
434
self ._session_lsids : Dict [str , Mapping [str , Any ]] = {}
@@ -624,15 +625,12 @@ def get_lsid_for_session(self, session_name):
624
625
# session has been closed.
625
626
return self ._session_lsids [session_name ]
626
627
627
- def entities (self ):
628
- return self ._entities
629
-
630
628
def advance_cluster_times (self , cluster_time : Optional [ClusterTime ] = None ):
631
629
if cluster_time is not None :
632
630
self ._cluster_time = cluster_time
633
631
elif getattr (self , "_cluster_time" , None ) is None :
634
632
self ._cluster_time = self .test .client .admin .command ("ping" ).get ("$clusterTime" )
635
- for entity in self .entities () :
633
+ for entity in self ._entities :
636
634
if isinstance (entity , ClientSession ):
637
635
entity .advance_cluster_time (self ._cluster_time )
638
636
You can’t perform that action at this time.
0 commit comments