diff --git a/tests/integration/network/test_node_storage.py b/tests/integration/network/test_node_storage.py index 8475f2bc8..a4d23c45c 100644 --- a/tests/integration/network/test_node_storage.py +++ b/tests/integration/network/test_node_storage.py @@ -46,6 +46,7 @@ def test_one_node_stores_a_bunch_of_others(federated_ursulas, lonely_ursula_make yield deferToThread(start_lonely_learning_loop) - assert list(newcomer.known_nodes) - assert len(list(newcomer.known_nodes)) == len(list(newcomer.node_storage.all(True))) - assert set(list(newcomer.known_nodes)) == set(list(newcomer.node_storage.all(True))) + matured_known_nodes = list(node.mature() for node in newcomer.known_nodes) + assert list(matured_known_nodes) + assert len(matured_known_nodes) == len(list(newcomer.node_storage.all(True))) + assert set(matured_known_nodes) == set(list(newcomer.node_storage.all(True)))