def test_create_write(self): wm = WalletManager() wm.create_wallet_file(path) try: wm.open_wallet(path) random_password = utils.get_random_hex_str(10) label = 'label' wm.create_account(random_password, label) default_account = wm.get_default_account_data() self.assertEqual(label, default_account.label) wm.create_identity(random_password, label) default_identity = wm.get_default_identity() self.assertEqual(default_identity.label, label) wm.write_wallet() finally: wm.del_wallet_file()
def test_get_default_account(self): wm = WalletManager() wm.create_wallet_file(path) try: wm.open_wallet(path) size = 3 for _ in range(size): wm.create_account(password) accounts = wm.get_wallet().get_accounts() self.assertEqual(len(accounts), size) for acct in accounts: wm.get_wallet().set_default_account_by_address( acct.b58_address) default_account = wm.get_default_account_data() self.assertEqual(default_account.b58_address, acct.b58_address) finally: wm.del_wallet_file()