Skip to content

Commit 6d1dd6d

Browse files
authored
PYTHON-3014 Update how events are added to entity map to match specification (#785)
1 parent e1884b4 commit 6d1dd6d

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

test/test_create_entities.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -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)

test/unified_format.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,11 @@ def add_event(self, event):
226226
if event_name in self._event_types:
227227
super(EventListenerUtil, self).add_event(event)
228228
for id in self._event_mapping[event_name]:
229-
self.entity_map[id].append(str(event))
229+
self.entity_map[id].append({
230+
"name": type(event).__name__,
231+
"observedAt": time.time(),
232+
"description": repr(event)
233+
})
230234

231235
def _command_event(self, event):
232236
if event.command_name.lower() not in self._ignore_commands:

0 commit comments

Comments
 (0)