def test_get_default_account(self): wm = WalletManager() path = os.path.join(os.getcwd(), 'test.json') wm.open_wallet(path) password = "******" size = 3 for i 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.address) default_account = wm.get_default_account() self.assertEqual(default_account.address, acct.address)
def test_create_write(self): wm = WalletManager() path = os.path.join(os.getcwd(), 'test.json') wm.open_wallet(path) password = util.get_random_str(10) label = 'label' wm.create_account(label, password) default_account = wm.get_default_account() self.assertEqual(label, default_account.label) wm.create_identity(label, password) default_identity = wm.get_default_identity() self.assertEqual(label, default_identity.label) wm.write_wallet() os.remove(path)