Skip to content

Commit 8699b1d

Browse files
committed
Test for write ledger
Signed-off-by: Ian Costanzo <ian@anon-solutions.ca>
1 parent e89cac7 commit 8699b1d

2 files changed

Lines changed: 8 additions & 4 deletions

File tree

aries_cloudagent/ledger/multiple_ledger/indy_manager.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,9 @@ async def _get_ledger_by_did(
8989
try:
9090
indy_sdk_ledger = None
9191
if self.write_ledger_info and ledger_id == self.write_ledger_info[0]:
92-
indy_sdk_ledger = self.get_write_ledger()
92+
indy_sdk_ledger = await self.get_write_ledger()
93+
if indy_sdk_ledger:
94+
indy_sdk_ledger = indy_sdk_ledger[1]
9395
elif ledger_id in self.production_ledgers:
9496
indy_sdk_ledger = self.production_ledgers.get(ledger_id)
9597
else:

aries_cloudagent/ledger/multiple_ledger/tests/test_indy_manager.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from ....cache.base import BaseCache
1212
from ....cache.in_memory import InMemoryCache
1313
from ....core.in_memory import InMemoryProfile
14+
from ....ledger.base import BaseLedger
1415
from ....messaging.responder import BaseResponder
1516

1617
from ...error import LedgerError
@@ -36,6 +37,7 @@ async def setUp(self):
3637
IndySdkLedgerPool("test_prod_1", checked=True), self.profile
3738
)
3839
test_write_ledger = ("test_prod_1", test_prod_ledger)
40+
self.context.injector.bind_instance(BaseLedger, test_prod_ledger)
3941
self.production_ledger["test_prod_1"] = test_prod_ledger
4042
self.production_ledger["test_prod_2"] = IndySdkLedger(
4143
IndySdkLedgerPool("test_prod_2", checked=True), self.profile
@@ -385,13 +387,13 @@ async def test_lookup_did_in_configured_ledgers_prod_not_cached(
385387

386388
async def test_lookup_did_in_configured_ledgers_cached_prod_ledger(self):
387389
cache = InMemoryCache()
388-
await cache.set("did_ledger_id_resolver::Av63wJYM7xYR4AiygYq4c3", "test_prod_1")
390+
await cache.set("did_ledger_id_resolver::Av63wJYM7xYR4AiygYq4c3", "test_prod_2")
389391
self.profile.context.injector.bind_instance(BaseCache, cache)
390392
(ledger_id, ledger_inst,) = await self.manager.lookup_did_in_configured_ledgers(
391393
"Av63wJYM7xYR4AiygYq4c3", cache_did=True
392394
)
393-
assert ledger_id == "test_prod_1"
394-
assert ledger_inst.pool.name == "test_prod_1"
395+
assert ledger_id == "test_prod_2"
396+
assert ledger_inst.pool.name == "test_prod_2"
395397

396398
async def test_lookup_did_in_configured_ledgers_cached_non_prod_ledger(self):
397399
cache = InMemoryCache()

0 commit comments

Comments
 (0)