Rehabilitate balance api

apps/cic-eth/tests/conftest.py

@@ -3,10 +3,14 @@ import os
 import sys
 import logging
 
+# local imports
+from cic_eth.api import Api
+
 script_dir = os.path.dirname(os.path.realpath(__file__))
 root_dir = os.path.dirname(script_dir)
 sys.path.insert(0, root_dir)
 
+# assemble fixtures
 from tests.fixtures_config import *
 from tests.fixtures_database import *
 from tests.fixtures_celery import *
@@ -15,3 +19,12 @@ from chainlib.eth.pytest import *
 from contract_registry.pytest import *
 from cic_eth_registry.pytest.fixtures_contracts import *
 from cic_eth_registry.pytest.fixtures_tokens import *
+
+
+@pytest.fixture(scope='function')
+def api(
+    default_chain_spec,
+    custodial_roles,
+    ):
+    chain_str = str(default_chain_spec)
+    return Api(chain_str, queue=None, callback_param='foo')

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

@@ -0,0 +1,55 @@
+# standard imports
+import os
+import logging
+
+# external imports
+from chainlib.eth.address import to_checksum_address
+
+# local imports
+from cic_eth.api.api_task import Api
+
+logg = logging.getLogger()
+
+def test_balance_simple_api(
+        default_chain_spec,
+        init_database,
+        cic_registry,
+        foo_token,
+        register_tokens,
+        api,
+        celery_session_worker,
+    ):
+
+    chain_str = str(default_chain_spec)
+
+    a = to_checksum_address('0x' + os.urandom(20).hex())
+    t = api.balance(a, 'FOO', include_pending=False)
+    r = t.get_leaf()
+    assert t.successful()
+    logg.debug(r)
+
+    assert r[0].get('balance_network') != None
+
+
+def test_balance_complex_api(
+        default_chain_spec,
+        init_database,
+        cic_registry,
+        foo_token,
+        register_tokens,
+        api,
+        celery_session_worker,
+    ):
+
+    chain_str = str(default_chain_spec)
+
+    a = to_checksum_address('0x' + os.urandom(20).hex())
+    t = api.balance(a, 'FOO', include_pending=True)
+    r = t.get_leaf()
+    assert t.successful()
+    logg.debug(r)
+
+    assert r[0].get('balance_incoming') != None
+    assert r[0].get('balance_outgoing') != None
+    assert r[0].get('balance_network') != None
+