@@ -47,11 +47,12 @@ def test_store_events_as_entities(self):
47
47
self .scenario_runner .TEST_SPEC = spec
48
48
self .scenario_runner .setUp ()
49
49
self .scenario_runner .run_scenario (spec ["tests" ][0 ])
50
+ self .scenario_runner .entity_map ["client0" ].close ()
50
51
final_entity_map = self .scenario_runner .entity_map
51
52
self .assertIn ("events1" , final_entity_map )
52
53
self .assertGreater (len (final_entity_map ["events1" ]), 0 )
53
54
for event in final_entity_map ["events1" ]:
54
- self .assertIn ("PoolCreatedEvent" , event )
55
+ self .assertIn ("PoolCreatedEvent" , event [ "name" ] )
55
56
56
57
def test_store_all_others_as_entities (self ):
57
58
self .scenario_runner = UnifiedSpecTestMixinV1 ()
@@ -130,6 +131,7 @@ def test_store_all_others_as_entities(self):
130
131
self .scenario_runner .TEST_SPEC = spec
131
132
self .scenario_runner .setUp ()
132
133
self .scenario_runner .run_scenario (spec ["tests" ][0 ])
134
+ self .scenario_runner .entity_map ["client0" ].close ()
133
135
final_entity_map = self .scenario_runner .entity_map
134
136
for entity in ["errors" , "failures" ]:
135
137
self .assertIn (entity , final_entity_map )
0 commit comments