from unittest import TestCase
from bitcaviar_plus.block import deserialize_block
from bitcaviar_plus.errors import InvalidMagicBytes
from bitcaviar_plus.search import search_block_with


class TestBlockImplementation(TestCase):
    expected_genesis_block = {
        'magic_number': 'f9beb4d9',
        'size': '0000011d',
        'id': '000000000019d6689c085ae165831e934ff763ae46a2a6c172b3f1b60a8ce26f',
        'transaction_count': '01',
        'header': {
            'version': '00000001',
            'previous_block_id': '0000000000000000000000000000000000000000000000000000000000000000',
            'merkle_root': '4a5e1e4baab89f3a32518a88c31bc87f618f76673e2cc77ab2127b7afdeda33b',
            'time': '495fab29',
            'bits': '1d00ffff',
            'nonce': '7c2bac1d'
        },
        'transactions': [{
            'version': '00000001',
            'input_count': '01',
            'output_count': '01',
            'lock_time': '00000000',
            'id': '4a5e1e4baab89f3a32518a88c31bc87f618f76673e2cc77ab2127b7afdeda33b',
            'inputs': [{
                'id': '0000000000000000000000000000000000000000000000000000000000000000',
                'vout': 'ffffffff',
                'script_sig_size': '4d',
                'script_sig': '04ffff001d0104455468652054696d65732030332f4a616e'
                              '2f32303039204368616e63656c6c6f72206f6e206272696e'
                              '6b206f66207365636f6e64206261696c6f757420666f722062616e6b73',
                'sequence': 'ffffffff'
            }],
            'outputs': [{
                'value': '000000012a05f200',
                'script_pub_key_size': '43',
                'script_pub_key': '4104678afdb0fe5548271967f1a67130b7105cd6a828e03909a67962e0ea1f61deb649f'
                                  '6bc3f4cef38c4f35504e51ec112de5c384df7ba0b8d578a4c702b6bf11d5fac'
            }]
        }]
    }

    def test_parse_genesis_block(self):
        blk_path = '/bitcoin-node/.bitcoin/blocks/blk00000.dat'

        with open(blk_path, 'rb') as f:
            try:
                block = deserialize_block(f)
                self.assertEqual(
                    block, self.expected_genesis_block, 'Genesis block is not equal to expected genesis block'
                )
            except InvalidMagicBytes as e:
                self.fail(e)


class TestSearchImplementation(TestCase):
    def test_search_block(self):
        genesis_block_hash = '000000000019d6689c085ae165831e934ff763ae46a2a6c172b3f1b60a8ce26f'
        genesis_block = search_block_with(genesis_block_hash)
        print('---- Genesis block ----')
        print(genesis_block)

        first_block_hash = '00000000839a8e6886ab5951d76f411475428afc90947ee320161bbf18eb6048'
        first_block = search_block_with(first_block_hash)
        print('---- First block ----')
        print(first_block)