def _create_test_wallet(self): """ Generate a Wallet with a number of keypairs for testing :rtype: Wallet """ tmpdir = tempfile.mkdtemp() self.tmpdirs.append(tmpdir) wallet = Wallet(directory=tmpdir) wallet.unlock(b'MYPASS') wallet.generate_keys(count=20) wallet.lock() return wallet
def test_locked(self): # generate a new block and check if we increase balance w = Wallet(directory=self.directory) with self.assertRaises(OutOfUnusedAddresses): w.get_unused_address() # now it should work w.unlock(PASSWORD) w.get_unused_address() # lock wallet and fake that there are no more unused keys w.unused_keys = set() w.lock() with self.assertRaises(OutOfUnusedAddresses): w.get_unused_address() with self.assertRaises(WalletLocked): w.generate_keys()