cic-cli/tests/test_proof.py

# standard imports
import os
import unittest
import logging

# local imports
from cic import Proof
from cic.attachment import Attachment

# test imports
from tests.base_cic import (
        test_data_dir,
        TestCICBase,
        )

logging.basicConfig(level=logging.DEBUG)
logg = logging.getLogger()

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

class TestProof(TestCICBase):

    def test_proof_serialize(self):
        proof_path = os.path.join(test_data_dir, 'proof')
        c = Proof(path=proof_path, writer=self.outputs_writer)
        c.load()
        v = c.process()
        self.assertEqual(v, root_unmerged_hash)


    def test_proof_serialize_merge(self):
        proof_path = os.path.join(test_data_dir, 'proof')

        attach = Attachment(proof_path, writer=self.outputs_writer)
        attach.load()

        c = Proof(path=proof_path, attachments=attach, writer=self.outputs_writer)
        c.load()
        v = c.process()
        self.assertEqual(v, root_merged_hash)


if __name__ == '__main__':
    unittest.main()