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()