async def test_one_gossip_round(set_vals_by_key, monkeypatch, mocker):
monkeypatch.setattr(MockDBManager, "get_chain", lambda _, __: MockChain())
front = Frontier(((1, "val1"), ), (), ())
monkeypatch.setattr(MockChain, "frontier", front)
monkeypatch.setattr(
MockNextPeerSelection,
"get_next_gossip_peers",
lambda _, subcom, chain_id, frontier, fanout:
[p.overlay.my_peer for p in set_vals_by_key.nodes],
)
monkeypatch.setattr(
MockDBManager,
"reconcile",
lambda _, c_id, frontier, pub_key: FrontierDiff(((1, 1), ), {}),
)
monkeypatch.setattr(MockSettings, "frontier_gossip_collect_time", 0.1)
monkeypatch.setattr(MockSettings, "frontier_gossip_fanout", 5)
monkeypatch.setattr(
MockDBManager,
"get_block_blobs_by_frontier_diff",
lambda _, c_id, f_diff, __: [b"blob1"],
)
spy = mocker.spy(set_vals_by_key.nodes[0].overlay, "send_packet")
set_vals_by_key.nodes[0].overlay.frontier_gossip_sync_task(
set_vals_by_key.community_id)
spy.assert_called()
def test_blocks_by_frontier_diff_no_seq_num(self, monkeypatch, std_vals):
monkeypatch.setattr(
MockBlockStore, "get_hash_by_dot", lambda _, dot_bytes: self.test_hash
)
monkeypatch.setattr(
MockBlockStore, "get_block_by_hash", lambda _, blob_hash: self.block_blob
)
monkeypatch.setattr(MockChain, "get_dots_by_seq_num", lambda _, seq_num: list())
# init chain
chain_id = self.chain_id
self.dbms.chains[chain_id] = MockChain()
frontier_diff = FrontierDiff(Ranges(((1, 2),)), {})
set_to_request = set()
blobs = self.dbms.get_block_blobs_by_frontier_diff(
chain_id, frontier_diff, set_to_request
)
assert len(set_to_request) == 0
assert len(list(blobs)) == 0
def received_blocks_request(self, peer: Peer,
payload: BlocksRequestPayload) -> None:
f_diff = FrontierDiff.from_bytes(payload.frontier_diff)
chain_id = payload.subcom_id
vals_to_request = set()
self.logger.debug(
"Received block request %s from peer %s. Audit chain: %s",
f_diff,
peer,
chain_id.startswith(b"w"),
)
blocks = self.persistence.get_block_blobs_by_frontier_diff(
chain_id, f_diff, vals_to_request)
self.logger.debug(
"Sending %s blocks to peer %s. Audit chain %s",
len(blocks),
peer,
chain_id.startswith(b"w"),
)
for block in blocks:
self.send_packet(peer, RawBlockPayload(block))
def test_blocks_by_frontier_diff(self, monkeypatch, std_vals):
monkeypatch.setattr(
MockBlockStore, "get_hash_by_dot", lambda _, dot_bytes: bytes(dot_bytes)
)
monkeypatch.setattr(
MockBlockStore, "get_block_by_hash", lambda _, blob_hash: bytes(blob_hash)
)
monkeypatch.setattr(
MockChain, "get_dots_by_seq_num", lambda _, seq_num: ("dot1", "dot2")
)
# init chain
chain_id = self.chain_id
self.dbms.chains[chain_id] = MockChain()
frontier_diff = FrontierDiff(Ranges(((1, 2),)), {(1, ShortKey("efef")): {}})
vals_to_request = set()
blobs = self.dbms.get_block_blobs_by_frontier_diff(
chain_id, frontier_diff, vals_to_request
)
assert len(vals_to_request) == 0
assert len(blobs) == 3
def test_blocks_frontier_with_extra_request(self, monkeypatch, std_vals):
monkeypatch.setattr(
MockBlockStore, "get_hash_by_dot", lambda _, dot_bytes: self.test_hash
)
monkeypatch.setattr(
MockBlockStore, "get_block_by_hash", lambda _, blob_hash: self.block_blob
)
monkeypatch.setattr(
MockChain, "get_dots_by_seq_num", lambda _, seq_num: ("dot1", "dot2")
)
local_vers = {2: {"ef1"}, 7: {"ef1"}}
monkeypatch.setattr(
MockChain,
"get_all_short_hash_by_seq_num",
lambda _, seq_num: local_vers.get(seq_num),
)
monkeypatch.setattr(
MockChain,
"get_next_links",
lambda _, dot: ((dot[0] + 1, ShortKey("efef")),),
)
# init chain
chain_id = self.chain_id
self.dbms.chains[chain_id] = MockChain()
frontier_diff = FrontierDiff(
(), {(10, ShortKey("efef")): {2: ("ef1",), 7: ("ef2",)}}
)
set_to_request = set()
blobs = self.dbms.get_block_blobs_by_frontier_diff(
chain_id, frontier_diff, set_to_request
)
assert len(set_to_request) == 1
assert len(blobs) == 1
def test_frontier_diff_bytes_convert(val):
f = FrontierDiff(val.holes, val.conflicts)
assert FrontierDiff.from_bytes(f.to_bytes()) == f