def add(
dispatcher,
thread_pool,
consensus_proxy,
consensus_notifier,
):
handler = handlers.ConsensusRegisterHandler(consensus_proxy,
consensus_notifier)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusRegisterBlockNewSyncHandler(
consensus_proxy, consensus_notifier)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusSendToHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusBroadcastHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusInitializeBlockHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusSummarizeBlockHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusFinalizeBlockHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusCancelBlockHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusCheckBlocksHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusCheckBlocksNotifier(consensus_proxy,
consensus_notifier)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusCommitBlockHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusIgnoreBlockHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusFailBlockHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusBlocksGetHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusChainHeadGetHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusSettingsGetHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
handler = handlers.ConsensusStateGetHandler(consensus_proxy)
dispatcher.add_handler(handler.request_type, handler, thread_pool)
def test_consensus_register_handler(self):
handler = handlers.ConsensusRegisterHandler(self.mock_proxy)
request_class = handler.request_class
request = request_class()
result = handler.handle('mock-id', request.SerializeToString())
response = result.message_out
self.assertEqual(response.status, handler.response_class.OK)
self.mock_proxy.register.assert_called_with('', '', [], 'mock-id')
def test_consensus_register_handler(self):
mock_chain_head = Mock()
mock_chain_head.identifier = "dead"
mock_chain_head.previous_block_id = "beef"
mock_chain_head.signer_public_key = "abcd"
mock_chain_head.block_num = 12
mock_chain_head.consensus = b"deadbeef"
self.mock_proxy.register.return_value = mock_chain_head, []
handler = handlers.ConsensusRegisterHandler(self.mock_proxy)
request_class = handler.request_class
request = request_class()
request.name = "test"
request.version = "test"
result = handler.handle(None, request.SerializeToString())
response = result.message_out
self.assertEqual(response.status, handler.response_class.OK)
self.mock_proxy.register.assert_called_with()
def test_consensus_register_handler(self):
header = BlockHeader(previous_block_id="beef",
signer_public_key="abcd",
block_num=12,
consensus=b'deadbeef')
mock_chain_head = Mock(header_signature="dead",
header=header.SerializeToString())
mock_startup_info = StartupInfo(chain_head=mock_chain_head,
peers=['dead', 'beef'],
local_peer_info=b'abc')
self.mock_proxy.register.return_value = mock_startup_info
handler = handlers.ConsensusRegisterHandler(self.mock_proxy)
request_class = handler.request_class
request = request_class()
result = handler.handle(None, request.SerializeToString())
response = result.message_out
self.assertEqual(response.status, handler.response_class.OK)
self.assertEqual(response.chain_head.block_id, bytes.fromhex("dead"))
self.assertEqual(response.peers[0].peer_id, bytes.fromhex("dead"))
self.assertEqual(response.local_peer_info.peer_id, b'abc')
self.mock_proxy.register.assert_called_with('', '', None)
def test_consensus_register_handler(self):
mock_chain_head = Mock()
mock_chain_head.identifier = "dead"
mock_chain_head.previous_block_id = "beef"
mock_chain_head.signer_public_key = "abcd"
mock_chain_head.block_num = 12
mock_chain_head.consensus = b"deadbeef"
mock_startup_info = StartupInfo(chain_head=mock_chain_head,
peers=['dead', 'beef'],
local_peer_info=b'abc')
self.mock_proxy.register.return_value = mock_startup_info
handler = handlers.ConsensusRegisterHandler(
self.mock_proxy, self.mock_consensus_notifier)
request_class = handler.request_class
request = request_class()
request.name = "test"
request.version = "test"
result = handler.handle(None, request.SerializeToString())
response = result.message_out
self.assertEqual(response.status, handler.response_class.OK)
self.assertEqual(response.chain_head.block_id, bytes.fromhex("dead"))
self.assertEqual(response.peers[0].peer_id, bytes.fromhex("dead"))
self.assertEqual(response.local_peer_info.peer_id, b'abc')
self.mock_proxy.register.assert_called_with()
