nucypher/tests/network/test_node_storage.py

22 lines
850 B
Python
Raw Normal View History

from nucypher.utilities.sandbox.ursula import make_federated_ursulas
2018-07-02 12:18:31 +00:00
def test_one_node_stores_a_bunch_of_others(federated_ursulas, ursula_federated_test_config):
the_chosen_seednode = list(federated_ursulas)[2]
seed_node = the_chosen_seednode.seed_node_metadata()
newcomer = make_federated_ursulas(
ursula_config=ursula_federated_test_config,
quantity=1,
know_each_other=False,
save_metadata=True,
seed_nodes=[seed_node]).pop()
assert not newcomer.known_nodes
newcomer.start_learning_loop()
# The known_nodes are all saved in storage (and no others have been saved)
assert list(newcomer.known_nodes.values()) == list(newcomer.node_storage.all(True))
# ...and the_chosen_seednode is in there.
assert the_chosen_seednode in newcomer.node_storage.all(federated_only=True)