Rehabilitate address translate test

apps/cic-eth/cic_eth/ext/address.py

@@ -3,25 +3,36 @@ import logging
 
 # third-party imports
 import celery
-from cic_registry.chain import ChainSpec
+from chainlib.chain import ChainSpec
+from chainlib.connection import RPCConnection
+from chainlib.eth.constant import ZERO_ADDRESS
+from cic_eth_registry import CICRegistry
+from eth_address_declarator import AddressDeclarator
 
 # local imports
 from cic_eth.eth import RpcClient
 from cic_eth.registry import safe_registry
+from cic_eth.task import BaseTask
 
 celery_app = celery.current_app
 
 logg = logging.getLogger()
 
 
-def translate_address(address, trusted_addresses, chain_spec):
-    c = RpcClient(chain_spec)
-    registry = safe_registry(c.w3)
+def translate_address(address, trusted_addresses, chain_spec, sender_address=ZERO_ADDRESS):
+
+    rpc = RPCConnection.connect(chain_spec, 'default')
+    registry = CICRegistry(chain_spec, rpc)
+    
+    declarator_address = registry.by_name('AddressDeclarator', sender_address=sender_address)
+    c = AddressDeclarator()
+
     for trusted_address in trusted_addresses:
-        o = registry.get_contract(chain_spec, 'AddressDeclarator', 'Declarator')
-        fn = o.function('declaration')
-        declaration_hex = fn(trusted_address, address).call()
-        declaration_bytes = declaration_hex[0].rstrip(b'\x00')
+        o = c.declaration(declarator_address, trusted_address, address, sender_address=sender_address)
+        r = rpc.do(o)
+        declaration_hex = AddressDeclarator.parse_declaration(r)
+        declaration_hex = declaration_hex[0].rstrip('0')
+        declaration_bytes = bytes.fromhex(declaration_hex)
         declaration = None
         try:
             declaration = declaration_bytes.decode('utf-8', errors='strict')
@@ -30,19 +41,19 @@ def translate_address(address, trusted_addresses, chain_spec):
         return declaration
 
 
-@celery_app.task()
-def translate_tx_addresses(tx, trusted_addresses, chain_str):
+@celery_app.task(bind=True, base=BaseTask)
+def translate_tx_addresses(self, tx, trusted_addresses, chain_spec_dict):
 
-    chain_spec = ChainSpec.from_chain_str(chain_str)
+    chain_spec = ChainSpec.from_dict(chain_spec_dict)
 
     declaration = None
     if tx['sender_label'] == None:
-        declaration = translate_address(tx['sender'], trusted_addresses, chain_spec)
+        declaration = translate_address(tx['sender'], trusted_addresses, chain_spec, self.call_address)
     tx['sender_label'] = declaration
 
     declaration = None
     if tx['recipient_label'] == None:
-        declaration = translate_address(tx['recipient'], trusted_addresses, chain_spec)
+        declaration = translate_address(tx['recipient'], trusted_addresses, chain_spec, self.call_address)
     tx['recipient_label'] = declaration
 
     return tx

apps/cic-eth/tests/fixtures_celery.py

@@ -1,16 +1,28 @@
-# third-party imports
+# external imports
 import pytest
 import tempfile
 import logging
 import shutil
 
+# local impors
+from cic_eth.task import BaseTask
+
 #logg = logging.getLogger(__name__)
 logg = logging.getLogger()
 
 
+@pytest.fixture(scope='function')
+def init_celery_tasks(
+    contract_roles, 
+        ):
+    BaseTask.call_address = contract_roles['DEFAULT']
+
+
 # celery fixtures
 @pytest.fixture(scope='session')
-def celery_includes():
+def celery_includes(
+    contract_roles,
+        ):
     return [
 #        'cic_eth.eth.bancor',
         'cic_eth.eth.erc20',

apps/cic-eth/tests/task/api/test_list.py

@@ -29,7 +29,7 @@ def test_list_tx(
         foo_token,
         register_tokens,
         init_eth_tester,
-        celery_worker,
+        celery_session_worker,
         ):
 
     chain_id = default_chain_spec.chain_id()

apps/cic-eth/tests/task/test_task_account.py

@@ -69,7 +69,7 @@ def test_register_account(
         eth_empty_accounts,
         custodial_roles,
         call_sender,
-        celery_worker,
+        celery_session_worker,
         ):
 
     s_nonce = celery.signature(
@@ -123,9 +123,9 @@ def test_register_account(
 
 @pytest.mark.skip()
 def test_role_task(
+    default_chain_spec,
     init_database,
     celery_session_worker,
-    default_chain_spec,
         ):
 
     address = '0x' + os.urandom(20).hex()