cic-cli/tests/base_cic.py

# standard imports
import os
import tempfile
import logging
import unittest
import random

# external imports
from hexathon import add_0x

# local imports
from cic.output import KVWriter
from cic.processor import Processor
from cic.attachment import Attachment
from cic import Proof

logg = logging.getLogger(__name__)

test_base_dir = os.path.dirname(os.path.realpath(__file__))
test_data_dir = os.path.join(test_base_dir, 'testdata')

proof_hash = '0f6fc017f29caf512c0feaaf83bc10614b488311cace2973dc248dc24b01e04f'
foo_hash = '2c26b46b68ffc68ff99b453c1d30413413422d706483bfa0f98a5e886266e7ae'
bar_hash = 'fcde2b2edba56bf408601fb721fe9b5c338d10ee429ea04fae5511b68fbf8fb9'
root_merged_hash = '1b38b0e9a2cc0895c72e5f460a613bb0f3d88c4a87c1ee2b39a41c64f1dbcd1e'
root_unmerged_hash = '86d0091b98635a45da472388a1204104d7d10ea1199c1bd9c1d235de73a2c285'


class TestCICBase(unittest.TestCase):

    def setUp(self):
        super(TestCICBase, self).setUp()
        random.seed(42)

        addresses = [
            add_0x(random.randbytes(20).hex()),
            add_0x(random.randbytes(20).hex()),
            add_0x(random.randbytes(20).hex()),
                ]
        self.token_address = add_0x(random.randbytes(32).hex())
        self.token_index_address = add_0x(random.randbytes(32).hex())
        self.address_declarator_address = add_0x(random.randbytes(32).hex())

        self.outputs_dir = tempfile.mkdtemp()
        self.outputs_writer = KVWriter(self.outputs_dir)
        self.core_processor = Processor(outputs_writer=self.outputs_writer)

        self.resources = {
                'token': {
                    'reference': self.token_address,
                    'key_address': addresses[0],
                    },
                'token_index': {
                    'reference': self.token_index_address,
                    'key_address': addresses[1],
                    },
                'address_declarator': {
                    'reference': self.address_declarator_address,
                    'key_address': addresses[2],
                    },
            }
        proof_dir = os.path.join(test_data_dir, 'proof')
        attach = Attachment(path=proof_dir)
        attach.load()
        self.proofs = Proof(proof_dir, attachments=attach)
        self.proofs.load()