From 7b2df93a758a383a0c455d29757cf5c78061559f Mon Sep 17 00:00:00 2001 From: Lluis Gifre <lluis.gifre@cttc.es> Date: Wed, 5 Jan 2022 22:14:22 +0100 Subject: [PATCH] Solve minor issue in unitary test of Context that prevented tests to finish correctly in random cases. --- src/context/tests/test_unitary.py | 26 ++++++++++++++++---------- 1 file changed, 16 insertions(+), 10 deletions(-) diff --git a/src/context/tests/test_unitary.py b/src/context/tests/test_unitary.py index 750fa2a31..bf8857f70 100644 --- a/src/context/tests/test_unitary.py +++ b/src/context/tests/test_unitary.py @@ -318,16 +318,22 @@ def test_grpc_topology( assert response.context_uuid.uuid == DEFAULT_CONTEXT_UUID # ----- Check create event ----------------------------------------------------------------------------------------- - event = events_collector.get_event(block=True) - assert isinstance(event, TopologyEvent) - assert event.event.event_type == EventTypeEnum.EVENTTYPE_CREATE - assert event.topology_id.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID - assert event.topology_id.topology_uuid.uuid == DEFAULT_TOPOLOGY_UUID - - event = events_collector.get_event(block=True) - assert isinstance(event, ContextEvent) - assert event.event.event_type == EventTypeEnum.EVENTTYPE_UPDATE - assert event.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID + context_event = None + topology_event = None + for _ in range(2): + event = events_collector.get_event(block=True) + assert isinstance(event, ContextEvent, TopologyEvent) + if isinstance(event, ContextEvent): context_event = event + if isinstance(event, TopologyEvent): topology_event = event + + assert context_event is not None + assert context_event.event.event_type == EventTypeEnum.EVENTTYPE_UPDATE + assert context_event.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID + + assert topology_event is not None + assert topology_event.event.event_type == EventTypeEnum.EVENTTYPE_CREATE + assert topology_event.topology_id.context_id.context_uuid.uuid == DEFAULT_CONTEXT_UUID + assert topology_event.topology_id.topology_uuid.uuid == DEFAULT_TOPOLOGY_UUID # ----- Update the object ------------------------------------------------------------------------------------------ response = context_client_grpc.SetTopology(Topology(**TOPOLOGY)) -- GitLab