103 lines
2.6 KiB
Python
103 lines
2.6 KiB
Python
# standard imports
|
|
import unittest
|
|
import logging
|
|
import os
|
|
import json
|
|
|
|
# local imports
|
|
from cic.contract.processor import ContractProcessor
|
|
from cic.contract.components.proof import Proof
|
|
from cic.contract.components.attachment import Attachment
|
|
from cic.contract.components.meta import Meta
|
|
|
|
# test imports
|
|
from tests.base_cic import (
|
|
TestCICBase,
|
|
test_data_dir,
|
|
root_merged_hash,
|
|
root_unmerged_hash,
|
|
)
|
|
|
|
logg = logging.getLogger()
|
|
logg.setLevel(logging.DEBUG)
|
|
|
|
|
|
class MockExt:
|
|
def __init__(self, address):
|
|
self.address = address
|
|
|
|
def process(self):
|
|
return (self.address, "foo")
|
|
|
|
|
|
class TestCICProcessor(TestCICBase):
|
|
def test_processor_meta(self):
|
|
fp = os.path.join(test_data_dir, "proof")
|
|
m = Meta(fp)
|
|
m.load()
|
|
|
|
mock_ext = MockExt(self.token_address)
|
|
p = ContractProcessor(
|
|
metadata=m, outputs_writer=self.outputs_writer, extensions=[mock_ext]
|
|
)
|
|
p.token_address = self.token_address
|
|
p.process()
|
|
|
|
meta_reference = m.reference(self.token_address)
|
|
|
|
fp = os.path.join(self.outputs_dir, meta_reference)
|
|
with open(fp, "r", encoding="utf-8") as f:
|
|
o = json.load(f)
|
|
self.assertEqual(m.asdict(), o)
|
|
|
|
def test_processor_attachment(self):
|
|
fp = os.path.join(test_data_dir, "proof")
|
|
m = Attachment(fp)
|
|
m.load()
|
|
|
|
mock_ext = MockExt(self.token_address)
|
|
p = ContractProcessor(
|
|
attachment=m, outputs_writer=self.outputs_writer, extensions=[mock_ext]
|
|
)
|
|
p.process()
|
|
|
|
for _k in list(m.contents.keys()):
|
|
os.stat(fp)
|
|
|
|
def test_processor_proof_noattachment(self):
|
|
fp = os.path.join(test_data_dir, "proof")
|
|
m = Proof(fp)
|
|
|
|
ap = os.path.join(test_data_dir, "proof_empty")
|
|
m.extra_attachments = Attachment(ap)
|
|
m.load()
|
|
|
|
mock_ext = MockExt(self.token_address)
|
|
p = ContractProcessor(
|
|
proof=m, outputs_writer=self.outputs_writer, extensions=[mock_ext]
|
|
)
|
|
p.process()
|
|
|
|
self.assertEqual(p.outputs[0], root_unmerged_hash)
|
|
|
|
def test_processor_proof_attachment(self):
|
|
fp = os.path.join(test_data_dir, "proof")
|
|
|
|
ma = Attachment(fp)
|
|
ma.load()
|
|
|
|
mp = Proof(fp, attachments=ma)
|
|
mp.load()
|
|
|
|
mock_ext = MockExt(self.token_address)
|
|
p = ContractProcessor(
|
|
proof=mp, outputs_writer=self.outputs_writer, extensions=[mock_ext]
|
|
)
|
|
p.process()
|
|
|
|
self.assertEqual(p.outputs[0], root_merged_hash)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|