use of org.aion.zero.impl.sync.msg.ResponseBlocks in project aion by aionnetwork.
the class RequestBlocksHandlerTest method testReceive_correctMessage_nullValue_withHash.
@Test
public void testReceive_correctMessage_nullValue_withHash() {
Block first = consecutiveBlocks.get(0);
byte[] hash = first.getHash();
Block last = consecutiveBlocks.get(3);
Logger log = mock(Logger.class);
when(log.isDebugEnabled()).thenReturn(true);
IAionBlockchain chain = mock(AionBlockchainImpl.class);
when(chain.getBlockByHash(hash)).thenReturn(first);
when(chain.getBlocksByRange(first.getNumber(), last.getNumber())).thenReturn(null);
IP2pMgr p2p = mock(P2pMgr.class);
RequestBlocksHandler handler = new RequestBlocksHandler(log, chain, p2p);
// receive correct message
byte[] encoding = RLP.encodeList(RLP.encodeByte(isFalse), RLP.encode(hash), RLP.encodeInt(4), RLP.encodeByte(isFalse));
handler.receive(peerId, displayId, encoding);
verify(log, times(1)).debug("<request-blocks from-block={} count={} order={}>", Hex.toHexString(hash), 4, "ASC");
verify(chain, times(1)).getBlockByHash(hash);
verify(chain, times(1)).getBlocksByRange(first.getNumber(), last.getNumber());
ResponseBlocks expectedResponse = new ResponseBlocks(List.of(first));
verify(p2p, times(1)).send(peerId, displayId, expectedResponse);
}
use of org.aion.zero.impl.sync.msg.ResponseBlocks in project aion by aionnetwork.
the class RequestBlocksHandlerTest method testReceive_correctMessage_withHashOnSideChain.
@Test
public void testReceive_correctMessage_withHashOnSideChain() {
Block first = consecutiveBlocks.get(0);
byte[] hash = first.getHash();
Block last = consecutiveBlocks.get(3);
Logger log = mock(Logger.class);
when(log.isDebugEnabled()).thenReturn(true);
IAionBlockchain chain = mock(AionBlockchainImpl.class);
when(chain.getBlockByHash(hash)).thenReturn(first);
when(chain.getBlocksByRange(first.getNumber(), last.getNumber())).thenReturn(List.of(last));
IP2pMgr p2p = mock(P2pMgr.class);
RequestBlocksHandler handler = new RequestBlocksHandler(log, chain, p2p);
// receive correct message
RequestBlocks request = new RequestBlocks(hash, 4, false);
handler.receive(peerId, displayId, request.encode());
verify(log, times(1)).debug("<request-blocks from-block={} count={} order={}>", Hex.toHexString(hash), 4, "ASC");
verify(chain, times(1)).getBlockByHash(hash);
verify(chain, times(1)).getBlocksByRange(first.getNumber(), last.getNumber());
ResponseBlocks expectedResponse = new ResponseBlocks(List.of(first));
verify(p2p, times(1)).send(peerId, displayId, expectedResponse);
}
use of org.aion.zero.impl.sync.msg.ResponseBlocks in project aion by aionnetwork.
the class RequestBlocksHandlerTest method testReceive_correctMessage_ascending_withHeight.
@Test
public void testReceive_correctMessage_ascending_withHeight() {
Block first = consecutiveBlocks.get(0);
Block last = consecutiveBlocks.get(3);
Logger log = mock(Logger.class);
when(log.isDebugEnabled()).thenReturn(true);
IAionBlockchain chain = mock(AionBlockchainImpl.class);
when(chain.getBlocksByRange(first.getNumber(), last.getNumber())).thenReturn(consecutiveBlocks);
IP2pMgr p2p = mock(P2pMgr.class);
RequestBlocksHandler handler = new RequestBlocksHandler(log, chain, p2p);
// receive correct message
RequestBlocks request = new RequestBlocks(first.getNumber(), 4, false);
handler.receive(peerId, displayId, request.encode());
verify(log, times(1)).debug("<request-blocks from-block={} count={} order={}>", first.getNumber(), 4, "ASC");
verify(chain, times(1)).getBlocksByRange(first.getNumber(), last.getNumber());
ResponseBlocks expectedResponse = new ResponseBlocks(consecutiveBlocks);
verify(p2p, times(1)).send(peerId, displayId, expectedResponse);
}
use of org.aion.zero.impl.sync.msg.ResponseBlocks in project aion by aionnetwork.
the class RequestBlocksHandlerTest method testReceive_correctMessage_descending_withHash.
@Test
public void testReceive_correctMessage_descending_withHash() {
Block first = consecutiveBlocks.get(3);
byte[] hash = first.getHash();
Block last = consecutiveBlocks.get(0);
// reverse the list order
LinkedList<Block> reverse = new LinkedList<>();
for (Block b : consecutiveBlocks) {
reverse.addFirst(b);
}
Logger log = mock(Logger.class);
when(log.isDebugEnabled()).thenReturn(true);
IAionBlockchain chain = mock(AionBlockchainImpl.class);
when(chain.getBlockByHash(hash)).thenReturn(first);
when(chain.getBlocksByRange(first.getNumber(), last.getNumber())).thenReturn(reverse);
IP2pMgr p2p = mock(P2pMgr.class);
RequestBlocksHandler handler = new RequestBlocksHandler(log, chain, p2p);
// receive correct message
RequestBlocks request = new RequestBlocks(hash, 4, true);
handler.receive(peerId, displayId, request.encode());
verify(log, times(1)).debug("<request-blocks from-block={} count={} order={}>", Hex.toHexString(hash), 4, "DESC");
verify(chain, times(1)).getBlockByHash(hash);
verify(chain, times(1)).getBlocksByRange(first.getNumber(), last.getNumber());
ResponseBlocks expectedResponse = new ResponseBlocks(reverse);
verify(p2p, times(1)).send(peerId, displayId, expectedResponse);
}
Aggregations