@@ -47,11 +47,12 @@ def test_store_events_as_entities(self):
4747 self .scenario_runner .TEST_SPEC = spec
4848 self .scenario_runner .setUp ()
4949 self .scenario_runner .run_scenario (spec ["tests" ][0 ])
50+ self .scenario_runner .entity_map ["client0" ].close ()
5051 final_entity_map = self .scenario_runner .entity_map
5152 self .assertIn ("events1" , final_entity_map )
5253 self .assertGreater (len (final_entity_map ["events1" ]), 0 )
5354 for event in final_entity_map ["events1" ]:
54- self .assertIn ("PoolCreatedEvent" , event )
55+ self .assertIn ("PoolCreatedEvent" , event [ "name" ] )
5556
5657 def test_store_all_others_as_entities (self ):
5758 self .scenario_runner = UnifiedSpecTestMixinV1 ()
@@ -130,6 +131,7 @@ def test_store_all_others_as_entities(self):
130131 self .scenario_runner .TEST_SPEC = spec
131132 self .scenario_runner .setUp ()
132133 self .scenario_runner .run_scenario (spec ["tests" ][0 ])
134+ self .scenario_runner .entity_map ["client0" ].close ()
133135 final_entity_map = self .scenario_runner .entity_map
134136 for entity in ["errors" , "failures" ]:
135137 self .assertIn (entity , final_entity_map )
0 commit comments