예제 #1
0
 def initiate_conversions(self):
     return [DefaultConversion(self), MultiChainConversion(self)]
class TestConversion(MultiChainTestCase):
    def __init__(self, *args, **kwargs):
        super(TestConversion, self).__init__(*args, **kwargs)
        self.community = TestCommunity()

    @blocking_call_on_reactor_thread
    @inlineCallbacks
    def setUp(self):
        yield super(TestConversion, self).setUp()
        self.converter = MultiChainConversion(self.community)
        self.block = TestBlock()

    def test_encoding_decoding_half_block(self):
        """
        Test encoding of a signed message
        """
        # Arrange
        meta = self.community.get_meta_message(HALF_BLOCK)
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=(self.block,))
        # Act
        encoded_message = self.converter._encode_half_block(message)[0]
        result = self.converter._decode_half_block(TestPlaceholder(meta), 0, encoded_message)[1]

        # Assert
        self.assertEqual_block(self.block, result.block)

    def test_encoding_decoding_half_block_big_number(self):
        """
        Test if a responder can send a signature message with big total_up and down.
        """
        # Arrange
        meta = self.community.get_meta_message(HALF_BLOCK)
        block = TestBlock()
        block.total_up_requester = pow(2, 63)
        block.total_down_requester = pow(2, 62)
        block.total_up_responder = pow(2, 61)
        block.total_down_responder = pow(2, 60)

        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=(self.block,))
        # Act
        encoded_message = self.converter._encode_half_block(message)[0]
        result = self.converter._decode_half_block(TestPlaceholder(meta), 0, encoded_message)[1]

        # Assert
        self.assertEqual_block(self.block, result.block)

    def test_decoding_half_block_wrong_size(self):
        """
        Test decoding a signature message with wrong size
        """
        # Arrange
        meta = self.community.get_meta_message(HALF_BLOCK)
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=(self.block,))
        # Act
        encoded_message = self.converter._encode_half_block(message)[0]
        # Act & Assert
        with self.assertRaises(DropPacket):
            # Remove a bit of message.
            self.converter._decode_half_block(TestPlaceholder(meta), 0, encoded_message[:-10])

    def test_encoding_decoding_crawl_request(self):
        """
        Test if a requester can send a crawl request message.
        """
        # Arrange
        meta = self.community.get_meta_message(CRAWL)
        requested_sequence_number = 500

        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=(requested_sequence_number,))
        # Act
        encoded_message = self.converter._encode_crawl_request(message)[0]

        result = self.converter._decode_crawl_request(TestPlaceholder(meta), 0, encoded_message)[1]
        # Assert
        self.assertEqual(requested_sequence_number, result.requested_sequence_number)

    def test_decoding_crawl_request_wrong_size(self):
        """
        Test if a DropPacket is raised when the crawl request size is wrong.
        """
        # Arrange
        meta = self.community.get_meta_message(CRAWL)
        requested_sequence_number = 500
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=(requested_sequence_number,))
        encoded_message = self.converter._encode_crawl_request(message)[0]

        # Act & Assert
        with self.assertRaises(DropPacket):
            # Remove a bit of message.
            self.converter._decode_crawl_request(TestPlaceholder(meta), 0, encoded_message[:-10])
 def setUp(self):
     yield super(TestConversion, self).setUp()
     self.converter = MultiChainConversion(self.community)
     self.block = TestBlock()
예제 #4
0
class TestConversion(MultiChainTestCase):
    def __init__(self, *args, **kwargs):
        super(TestConversion, self).__init__(*args, **kwargs)
        self.community = TestCommunity()

    @blocking_call_on_reactor_thread
    @inlineCallbacks
    def setUp(self):
        yield super(TestConversion, self).setUp()
        self.converter = MultiChainConversion(self.community)
        self.block = TestBlock()

    def test_encoding_decoding_signature(self):
        """
        Test if a responder can send a signature message.
        This only contains requester and responder data.
        """
        # Arrange
        meta = self.community.get_meta_message(SIGNATURE)
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=tuple(self.block.generate_signature_payload()))
        # Act
        encoded_message = self.converter._encode_signature(message)[0]
        result = self.converter._decode_signature(TestPlaceholder(meta), 0, encoded_message)[1]
        # Assert
        self.assertEqual_signature_payload(self.block, result)

    def test_encoding_decoding_signature_big_number(self):
        """
        Test if a responder can send a signature message with big total_up and down.
        """
        # Arrange
        meta = self.community.get_meta_message(SIGNATURE)
        self.block.total_up_requester = pow(2, 63)
        self.block.total_down_requester = pow(2, 62)
        self.block.total_up_responder = pow(2, 61)
        self.block.total_down_responder = pow(2, 60)
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=tuple(self.block.generate_signature_payload()))
        # Act
        encoded_message = self.converter._encode_signature(message)[0]
        result = self.converter._decode_signature(TestPlaceholder(meta), 0, encoded_message)[1]
        # Assert
        self.assertEqual_signature_payload(self.block, result)

    def test_encoding_decoding_signature_requester(self):
        """
        Test if a requester can send a signature message.
        This only contains requester data.
        """
        # Arrange
        meta = self.community.get_meta_message(SIGNATURE)
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=tuple(self.block.generate_requester()))
        # Act
        encoded_message = self.converter._encode_signature(message)[0]
        result = self.converter._decode_signature(TestPlaceholder(meta), 0, encoded_message)[1]
        # Assert
        self.assertEqual_signature_request(self.block, result)
        self.assertEqual(0, result.total_up_responder)
        self.assertEqual(0, result.total_down_responder)
        self.assertEqual(-1, result.sequence_number_responder)
        self.assertEqual(EMPTY_HASH, result.previous_hash_responder)

    def test_decoding_signature_wrong_size(self):
        """
        Test decoding a signature message with wrong size
        """
        # Arrange
        meta = self.community.get_meta_message(SIGNATURE)
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=tuple(self.block.generate_signature_payload()))
        # Act
        encoded_message = self.converter._encode_signature(message)[0]
        # Act & Assert
        with self.assertRaises(DropPacket):
            # Remove a bit of message.
            self.converter._decode_signature(TestPlaceholder(meta), 0, encoded_message[:-10])[1]

    def test_encoding_decoding_crawl_request(self):
        """
        Test if a requester can send a crawl request message.
        """
        # Arrange
        meta = self.community.get_meta_message(CRAWL_REQUEST)
        requested_sequence_number = 500

        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=(requested_sequence_number,))
        # Act
        encoded_message = self.converter._encode_crawl_request(message)[0]

        result = self.converter._decode_crawl_request(TestPlaceholder(meta), 0, encoded_message)[1]
        # Assert
        self.assertEqual(requested_sequence_number, result.requested_sequence_number)

    def test_decoding_crawl_request_wrong_size(self):
        """
        Test if a DropPacket is raised when the crawl request size is wrong.
        """
        # Arrange
        meta = self.community.get_meta_message(CRAWL_REQUEST)
        requested_sequence_number = 500
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=(requested_sequence_number,))
        encoded_message = self.converter._encode_crawl_request(message)[0]

        # Act & Assert
        with self.assertRaises(DropPacket):
            # Remove a bit of message.
            result = self.converter._decode_crawl_request(TestPlaceholder(meta), 0, encoded_message[:-10])[1]

    def test_encoding_decoding_crawl_request_empty(self):
        """
        Test if a requester can send a crawl request message without specifying the sequence number.
        """
        # Arrange
        meta = self.community.get_meta_message(CRAWL_REQUEST)
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=())
        # Act
        encoded_message = self.converter._encode_crawl_request(message)[0]

        result = self.converter._decode_crawl_request(TestPlaceholder(meta), 0, encoded_message)[1]
        # Assert
        self.assertEqual(-1, result.requested_sequence_number)

    def test_encoding_decoding_crawl_response(self):
        """
        Test if a responder can send a crawl_response message.
        This only contains requester and responder data.
        """
        # Arrange
        meta = self.community.get_meta_message(CRAWL_RESPONSE)
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=tuple(self.block.generate_block_payload()))
        # Act
        encoded_message = self.converter._encode_crawl_response(message)[0]
        result = self.converter._decode_crawl_response(TestPlaceholder(meta), 0, encoded_message)[1]
        # Assert
        self.assertEqual(len(self.block.public_key_requester), len(result.public_key_requester))
        self.assertTrue(self.community.crypto.is_valid_public_bin(self.block.public_key_requester))
        self.assertTrue(self.community.crypto.is_valid_public_bin(self.block.public_key_responder))

        self.assertEqual_block(self.block, result)

    def test_decoding_crawl_response_wrong_size(self):
        """
        Test if a DropPacket is raised when the crawl response size is wrong.
        """
        # Arrange
        meta = self.community.get_meta_message(CRAWL_RESPONSE)
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=tuple(self.block.generate_block_payload()))
        # Act
        encoded_message = self.converter._encode_crawl_response(message)[0]
        with self.assertRaises(DropPacket):
            # Remove a bit of message.
            self.converter._decode_crawl_response(TestPlaceholder(meta), 0, encoded_message[:-10])[1]

    def test_split_function(self):
        """
        Test the MultiChain split function.
        """
        # Arrange
        meta = self.community.get_meta_message(SIGNATURE)
        message = meta.impl(distribution=(self.community.claim_global_time(),),
                            payload=tuple(self.block.generate_signature_payload()))
        # Act
        encoded_message = self.converter._encode_signature(message)[0]
        result = split_function(encoded_message)
        # Assert
        values = unpack(signature_format[:-len(append_format)], result[0])
        self.assertEqual(6, len(values))
        self.assertEqual(self.block.up, values[0])
        self.assertEqual(self.block.down, values[1])
        self.assertEqual(self.block.total_up_requester, values[2])
        self.assertEqual(self.block.total_down_requester, values[3])
        self.assertEqual(self.block.sequence_number_requester, values[4])
        self.assertEqual(self.block.previous_hash_requester, values[5])

        self.assertEqual(encoded_message, result[1])