async def test_send_schema_already_exists(
self,
mock_build_schema_req,
mock_create_schema,
mock_check_existing,
mock_close_pool,
mock_open_ledger,
mock_create_config,
mock_set_proto,
):
# mock_did = async_mock.CoroutineMock()
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_wallet.get_public_did.return_value.did = "abc"
mock_create_schema.return_value = (1, 2)
fetch_schema_id = f"{mock_wallet.get_public_did.return_value.did}:{2}:schema_name:schema_version"
mock_check_existing.return_value = fetch_schema_id
ledger = IndyLedger("name", mock_wallet)
async with ledger:
schema_id = await ledger.send_schema("schema_name",
"schema_version", [1, 2, 3])
assert schema_id == fetch_schema_id
async def test_submit_signed(
self,
mock_sign_submit,
mock_close_pool,
mock_open_ledger,
mock_create_config,
mock_set_proto,
):
mock_sign_submit.return_value = '{"op": "REPLY"}'
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
ledger = IndyLedger("name", mock_wallet)
async with ledger:
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_wallet.get_public_did.return_value = None
with self.assertRaises(BadLedgerRequestError):
await ledger._submit("{}", True)
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_did = mock_wallet.get_public_did.return_value
mock_did.did = self.test_did
await ledger._submit("{}", True)
mock_wallet.get_public_did.assert_called_once_with()
mock_sign_submit.assert_called_once_with(ledger.pool_handle,
mock_wallet.handle,
mock_did.did, "{}")
async def test_submit_unsigned(
self,
mock_submit,
mock_close_pool,
mock_open_ledger,
mock_create_config,
mock_set_proto,
):
mock_did = async_mock.MagicMock()
future = asyncio.Future()
future.set_result(mock_did)
mock_submit.return_value = '{"op": "REPLY"}'
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
mock_wallet.get_public_did.return_value = future
ledger = IndyLedger("name", mock_wallet)
async with ledger:
await ledger._submit("{}", False)
mock_wallet.get_public_did.assert_not_called()
mock_submit.assert_called_once_with(ledger.pool_handle, "{}")
async def test_get_credential_definition(
self,
mock_parse_get_cred_def_req,
mock_build_get_cred_def_req,
mock_create_schema,
mock_submit,
mock_close,
mock_open,
):
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_did = mock_wallet.get_public_did.return_value
mock_parse_get_cred_def_req.return_value = (None, "{}")
ledger = IndyLedger("name", mock_wallet)
async with ledger:
response = await ledger.get_credential_definition("cred_def_id")
mock_wallet.get_public_did.assert_called_once_with()
mock_build_get_cred_def_req.assert_called_once_with(
mock_did.did, "cred_def_id")
mock_submit.assert_called_once_with(
mock_build_get_cred_def_req.return_value, sign=True)
mock_parse_get_cred_def_req.assert_called_once_with(
mock_submit.return_value)
assert response == json.loads(
mock_parse_get_cred_def_req.return_value[1])
async def test_get_schema(
self,
mock_parse_get_schema_req,
mock_build_get_schema_req,
mock_submit,
mock_close,
mock_open,
):
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_did = mock_wallet.get_public_did.return_value
mock_did.did = self.test_did
mock_parse_get_schema_req.return_value = (None, "{}")
mock_submit.return_value = "{\"result\":{\"seqNo\":1}}"
ledger = IndyLedger("name", mock_wallet)
async with ledger:
response = await ledger.get_schema("schema_id")
mock_wallet.get_public_did.assert_called_once_with()
mock_build_get_schema_req.assert_called_once_with(
mock_did.did, "schema_id")
mock_submit.assert_called_once_with(
mock_build_get_schema_req.return_value, sign=True)
mock_parse_get_schema_req.assert_called_once_with(
mock_submit.return_value)
assert response == json.loads(
mock_parse_get_schema_req.return_value[1])
async def test_get_schema(
self,
mock_parse_get_schema_req,
mock_build_get_schema_req,
mock_create_schema,
mock_submit,
mock_close,
mock_open,
):
mock_wallet = async_mock.MagicMock()
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_did = mock_wallet.get_public_did.return_value
mock_parse_get_schema_req.return_value = (None, "{}")
ledger = IndyLedger("name", mock_wallet, "genesis_transactions")
async with ledger:
response = await ledger.get_schema("schema_id")
mock_wallet.get_public_did.assert_called_once_with()
mock_build_get_schema_req.assert_called_once_with(
mock_did.did, "schema_id")
mock_submit.assert_called_once_with(
mock_build_get_schema_req.return_value, sign=True)
mock_parse_get_schema_req.assert_called_once_with(
mock_submit.return_value)
assert response == json.loads(
mock_parse_get_schema_req.return_value[1])
async def test_send_schema_already_exists(
self,
mock_build_schema_req,
mock_create_schema,
mock_submit,
mock_close_pool,
mock_open_ledger,
mock_create_config,
mock_set_proto,
):
# mock_did = async_mock.CoroutineMock()
mock_wallet = async_mock.CoroutineMock()
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_wallet.get_public_did.return_value.did = "abc"
mock_create_schema.return_value = (1, 2)
mock_submit.side_effect = DuplicateSchemaError
ledger = IndyLedger("name", mock_wallet, "genesis_transactions")
async with ledger:
schema_id = await ledger.send_schema("schema_name",
"schema_version", [1, 2, 3])
assert (
schema_id ==
f"{mock_wallet.get_public_did.return_value.did}:{2}:schema_name:schema_version"
)
async def test_submit_pool_closed(self, mock_close_pool, mock_open_ledger,
mock_create_config, mock_set_proto):
ledger = IndyLedger("name", "wallet", "genesis_transactions")
with self.assertRaises(ClosedPoolError) as context:
await ledger._submit("{}")
assert "sign and submit request to closed pool" in str(
context.exception)
async def test_submit_pool_closed(
self, mock_close_pool, mock_open_ledger, mock_create_config, mock_set_proto
):
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
ledger = IndyLedger("name", mock_wallet)
with self.assertRaises(ClosedPoolError) as context:
await ledger._submit("{}")
assert "sign and submit request to closed pool" in str(context.exception)
def test_init(self, mock_open):
mock_open.return_value = async_mock.MagicMock()
ledger = IndyLedger("name", "wallet", "genesis_transactions")
assert ledger.name == "name"
assert ledger.wallet == "wallet"
mock_open.assert_called_once_with(GENESIS_TRANSACTION_PATH, "w")
mock_open.return_value.__enter__.return_value.write.assert_called_once_with(
"genesis_transactions")
async def test_submit_rejected(
self,
mock_submit,
mock_close_pool,
mock_open_ledger,
mock_create_config,
mock_set_proto,
):
mock_did = async_mock.MagicMock()
future = asyncio.Future()
future.set_result(mock_did)
mock_submit.return_value = '{"op": "REQNACK", "reason": "a reason"}'
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
mock_wallet.get_public_did.return_value = future
ledger = IndyLedger("name", mock_wallet)
async with ledger:
with self.assertRaises(LedgerTransactionError) as context:
await ledger._submit("{}", False)
assert "Ledger rejected transaction request" in str(
context.exception)
mock_submit.return_value = '{"op": "REJECT", "reason": "another reason"}'
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
mock_wallet.get_public_did.return_value = future
ledger = IndyLedger("name", mock_wallet)
async with ledger:
with self.assertRaises(LedgerTransactionError) as context:
await ledger._submit("{}", False)
assert "Ledger rejected transaction request" in str(
context.exception)
async def test_send_schema(
self,
mock_build_schema_req,
mock_create_schema,
mock_add_record,
mock_fetch_schema_by_seq_no,
mock_fetch_schema_by_id,
mock_submit,
mock_close,
mock_open,
):
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
ledger = IndyLedger("name", mock_wallet)
mock_create_schema.return_value = ("schema_issuer_did:name:1.0", "{}")
mock_fetch_schema_by_id.return_value = None
mock_fetch_schema_by_seq_no.return_value = None
async with ledger:
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_wallet.get_public_did.return_value = None
with self.assertRaises(BadLedgerRequestError):
schema_id = await ledger.send_schema("schema_name",
"schema_version",
[1, 2, 3])
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_did = mock_wallet.get_public_did.return_value
mock_did.did = self.test_did
schema_id = await ledger.send_schema("schema_name",
"schema_version", [1, 2, 3])
mock_wallet.get_public_did.assert_called_once_with()
mock_create_schema.assert_called_once_with(mock_did.did,
"schema_name",
"schema_version",
json.dumps([1, 2, 3]))
mock_build_schema_req.assert_called_once_with(
mock_did.did, mock_create_schema.return_value[1])
mock_submit.assert_called_once_with(
mock_build_schema_req.return_value, public_did=mock_did.did)
assert schema_id == mock_create_schema.return_value[0]
async def test_aenter_aexit(self, mock_close_pool, mock_open_ledger,
mock_set_proto):
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
ledger = IndyLedger("name", mock_wallet)
async with ledger as l:
mock_set_proto.assert_called_once_with(2)
mock_open_ledger.assert_called_once_with("name", "{}")
assert l == ledger
mock_close_pool.assert_not_called()
assert l.pool_handle == mock_open_ledger.return_value
mock_close_pool.assert_called_once()
assert ledger.pool_handle == None
async def test_aenter_aexit(self, mock_close_pool, mock_open_ledger,
mock_create_config, mock_set_proto):
ledger = IndyLedger("name", "wallet", "genesis_transactions")
async with ledger as l:
mock_set_proto.assert_called_once_with(2)
mock_create_config.assert_called_once_with(
"name", json.dumps({"genesis_txn": GENESIS_TRANSACTION_PATH}))
mock_open_ledger.assert_called_once_with("name", "{}")
assert l == ledger
mock_close_pool.assert_not_called()
assert l.pool_handle == mock_open_ledger.return_value
mock_close_pool.assert_called_once()
assert ledger.pool_handle == None
async def test_init(self, mock_open, mock_create_config):
mock_open.return_value = async_mock.MagicMock()
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
ledger = IndyLedger("name", mock_wallet)
assert ledger.pool_name == "name"
assert ledger.wallet is mock_wallet
await ledger.create_pool_config("genesis_transactions")
mock_open.assert_called_once_with(GENESIS_TRANSACTION_PATH, "w")
mock_open.return_value.__enter__.return_value.write.assert_called_once_with(
"genesis_transactions")
mock_create_config.assert_called_once_with(
"name", json.dumps({"genesis_txn": GENESIS_TRANSACTION_PATH}))
async def test_send_credential_definition(
self,
mock_build_cred_def,
mock_create_store_cred_def,
mock_add_record,
mock_submit,
mock_fetch_cred_def,
mock_close,
mock_open,
mock_get_schema,
):
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
mock_get_schema.return_value = "{}"
cred_id = "cred_id"
cred_json = "[]"
mock_create_store_cred_def.return_value = (cred_id, cred_json)
mock_fetch_cred_def.return_value = None
ledger = IndyLedger("name", mock_wallet)
schema_id = "schema_issuer_did:name:1.0"
tag = "tag"
async with ledger:
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_wallet.get_public_did.return_value = None
with self.assertRaises(BadLedgerRequestError):
await ledger.send_credential_definition(schema_id, tag)
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_did = mock_wallet.get_public_did.return_value
result_id = await ledger.send_credential_definition(schema_id, tag)
assert result_id == cred_id
mock_wallet.get_public_did.assert_called_once_with()
mock_get_schema.assert_called_once_with(schema_id)
mock_build_cred_def.assert_called_once_with(
mock_did.did, cred_json)
async def test_credential_definition_id2schema_id(
self,
mock_build_get_txn_req,
mock_submit,
mock_close,
mock_open,
):
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
mock_build_get_txn_req.return_value = json.dumps("dummy")
mock_submit.return_value = json.dumps({
"result": {
"data": {
"txn": {
"type": "101",
"metadata": {
"from": f"{TestIndyLedger.test_did}"
},
"data": {
"data": {
"name": "favourite_drink",
"version": "1.0"
}
}
}
}
}
})
ledger = IndyLedger("name", mock_wallet)
async with ledger:
s_id_short = await ledger.credential_definition_id2schema_id(
f"{TestIndyLedger.test_did}:3:CL:9999:tag")
mock_build_get_txn_req.assert_called_once_with(None,
None,
seq_no=9999)
mock_submit.assert_called_once_with(
mock_build_get_txn_req.return_value)
assert s_id_short == f"{TestIndyLedger.test_did}:2:favourite_drink:1.0"
s_id_long = await ledger.credential_definition_id2schema_id(
f"{TestIndyLedger.test_did}:3:CL:{s_id_short}:tag")
assert s_id_long == s_id_short
async def test_send_schema(
self,
mock_build_schema_req,
mock_create_schema,
mock_submit,
mock_close,
mock_open,
):
mock_wallet = async_mock.MagicMock()
ledger = IndyLedger("name", mock_wallet, "genesis_transactions")
mock_create_schema.return_value = ("schema_id", "{}")
async with ledger:
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_wallet.get_public_did.return_value = None
with self.assertRaises(BadLedgerRequestError):
schema_id = await ledger.send_schema("schema_name",
"schema_version",
[1, 2, 3])
mock_wallet.get_public_did = async_mock.CoroutineMock()
mock_did = mock_wallet.get_public_did.return_value
schema_id = await ledger.send_schema("schema_name",
"schema_version", [1, 2, 3])
mock_wallet.get_public_did.assert_called_once_with()
mock_create_schema.assert_called_once_with(mock_did.did,
"schema_name",
"schema_version",
json.dumps([1, 2, 3]))
mock_build_schema_req.assert_called_once_with(
mock_did.did, mock_create_schema.return_value[1])
mock_submit.assert_called_once_with(
mock_build_schema_req.return_value)
assert schema_id == mock_create_schema.return_value[0]
async def test_credential_definition_id2schema_id(self, mock_get_schema,
mock_close, mock_open):
mock_wallet = async_mock.MagicMock()
mock_wallet.WALLET_TYPE = "indy"
S_ID = f"{TestIndyLedger.test_did}:2:favourite_drink:1.0"
SEQ_NO = "9999"
mock_get_schema.return_value = {"id": S_ID}
ledger = IndyLedger("name", mock_wallet)
async with ledger:
s_id_short = await ledger.credential_definition_id2schema_id(
f"{TestIndyLedger.test_did}:3:CL:{SEQ_NO}:tag")
mock_get_schema.assert_called_once_with(SEQ_NO)
assert s_id_short == S_ID
s_id_long = await ledger.credential_definition_id2schema_id(
f"{TestIndyLedger.test_did}:3:CL:{s_id_short}:tag")
assert s_id_long == s_id_short
