def test_check_blocks(self): self.mock_stream.send.return_value = self._make_future( message_type=Message.CONSENSUS_CHECK_BLOCKS_RESPONSE, content=consensus_pb2.ConsensusCheckBlocksResponse( status=consensus_pb2.ConsensusCheckBlocksResponse.OK)) self.service.check_blocks(priority=[b'test1', b'test2']) self.mock_stream.send.assert_called_with( message_type=Message.CONSENSUS_CHECK_BLOCKS_REQUEST, content=consensus_pb2.ConsensusCheckBlocksRequest( block_ids=[b'test1', b'test2']).SerializeToString())
def check_blocks(self, priority): request = consensus_pb2.ConsensusCheckBlocksRequest(block_ids=priority) response_type = consensus_pb2.ConsensusCheckBlocksResponse response = self._send( request=request, message_type=Message.CONSENSUS_CHECK_BLOCKS_REQUEST, response_type=response_type) status = response.status if status == response_type.UNKNOWN_BLOCK: raise exceptions.UnknownBlock() if status != response_type.OK: raise exceptions.ReceiveError( 'Failed with status {}'.format(status))