def test_set_default_identity_by_index(self):
wm = WalletManager()
path = os.path.join(os.getcwd(), 'test.json')
wm.open_wallet(path)
size = 3
for i in range(size):
private_key = util.get_random_str(64)
wm.create_identity_from_pri_key("ide", str(i), private_key)
identities = wm.get_wallet().get_identities()
self.assertEqual(len(identities), size)
self.assertRaises(SDKException,
wm.get_wallet().set_default_identity_by_index, size)
for index in range(size):
wm.get_wallet().set_default_identity_by_index(index)
default_identity = wm.get_default_identity()
self.assertEqual(identities[index], default_identity)
os.remove(path)
def test_create_identity_from_prikey(self):
wm = WalletManager()
path = os.path.join(os.getcwd(), 'test.json')
wm.open_wallet(path)
private_key = '75de8489fcb2dcaf2ef3cd607feffde18789de7da129b5e97c81e001793cb7cf'
ide = wm.create_identity_from_pri_key("ide", "1", private_key)
self.assertEqual(ide.label, 'ide')
self.assertEqual(ide.ont_id,
'did:ont:AazEvfQPcQ2GEFFPLF1ZLwQ7K5jDn81hve')
os.remove(path)
def test_set_default_identity_by_ont_id(self):
wm = WalletManager()
path = os.path.join(os.getcwd(), 'test.json')
wm.open_wallet(path)
password = "******"
size = 3
for i in range(size):
private_key = util.get_random_str(64)
wm.create_identity_from_pri_key("ide", str(i), private_key)
identities = wm.get_wallet().get_identities()
self.assertEqual(len(identities), size)
self.assertRaises(SDKException,
wm.get_wallet().set_default_identity_by_ont_id, '')
ont_id_list = list()
for identity in wm.get_wallet().identities:
ont_id_list.append(identity.ont_id)
for index in range(size * 5):
rand_ont_id = random.choice(ont_id_list)
wm.get_wallet().set_default_identity_by_ont_id(rand_ont_id)
default_identity = wm.get_default_identity()
self.assertEqual(rand_ont_id, default_identity.ont_id)
os.remove(path)
